first commit

modules/production_work_timesheet/work.py

@@ -0,0 +1,112 @@
+# This file is part of Tryton.  The COPYRIGHT file at the top level of
+# this repository contains the full copyright notices and license terms.
+from collections import defaultdict
+
+from trytond.model import fields
+from trytond.pool import Pool, PoolMeta
+from trytond.pyson import Eval
+from trytond.transaction import Transaction
+
+
+class Work(metaclass=PoolMeta):
+    __name__ = 'production.work'
+
+    timesheet_works = fields.One2Many(
+        'timesheet.work', 'origin', 'Timesheet Works', readonly=True, size=1)
+    timesheet_work = fields.Function(
+        fields.Many2One('timesheet.work', "Timesheet Work"),
+        'get_timesheet_work')
+    timesheet_available = fields.Function(
+        fields.Boolean('Available on timesheets'),
+        'on_change_with_timesheet_available')
+    timesheet_lines = fields.Function(
+        fields.One2Many('timesheet.line', None, 'Timesheet Lines',
+            states={
+                'invisible': ~Eval('timesheet_works'),
+                },
+            domain=[
+                ('company', '=', Eval('company', -1)),
+                ('work', '=', Eval('timesheet_work', -1)),
+                ]),
+        'get_timesheet_lines', setter='set_timesheet_lines')
+
+    def get_timesheet_work(self, name):
+        if self.timesheet_works:
+            timesheet_work, = self.timesheet_works
+            return timesheet_work.id
+
+    @fields.depends('operation')
+    def on_change_with_timesheet_available(self, name=None):
+        if self.operation:
+            return self.operation.timesheet_available
+
+    def get_timesheet_lines(self, name):
+        if self.timesheet_works:
+            return [l.id for w in self.timesheet_works
+                for l in w.timesheet_lines]
+        else:
+            return []
+
+    @classmethod
+    def set_timesheet_lines(cls, works, name, value):
+        pool = Pool()
+        TimesheetWork = pool.get('timesheet.work')
+        if value:
+            timesheet_works = [tw for w in works for tw in w.timesheet_works]
+            TimesheetWork.write(timesheet_works, {
+                    'timesheet_lines': value,
+                    })
+
+    @classmethod
+    def on_modification(cls, mode, works, field_names=None):
+        super().on_modification(mode, works, field_names=field_names)
+        if mode in {'create', 'write'}:
+            cls._set_timesheet_work(works)
+
+    @classmethod
+    def on_delete(cls, works):
+        pool = Pool()
+        TimesheetWork = pool.get('timesheet.work')
+        callback = super().on_delete(works)
+        timesheet_works = [w for pw in works for w in pw.timesheet_works]
+        callback.append(lambda: TimesheetWork.delete(timesheet_works))
+        return callback
+
+    @classmethod
+    def _set_timesheet_work(cls, works):
+        pool = Pool()
+        Timesheet = pool.get('timesheet.work')
+        Date = pool.get('ir.date')
+
+        to_create = []
+        to_delete = []
+        to_write = defaultdict(list)
+        for work in works:
+            with Transaction().set_context(company=work.company.id):
+                today = Date.today()
+            if work.timesheet_available:
+                ended = work.state in {'done', 'cancelled'}
+                if not work.timesheet_works:
+                    to_create.append({
+                            'origin': str(work),
+                            'company': work.company.id,
+                            'timesheet_end_date': today if ended else None,
+                            })
+                elif ended:
+                    for timesheet in work.timesheet_works:
+                        date = max([today]
+                            + [l.date for l in timesheet.timesheet_lines])
+                        to_write[date].append(timesheet)
+            if (not work.timesheet_available
+                    and work.timesheet_works):
+                if all(not w.timesheet_lines
+                        for w in work.timesheet_works):
+                    to_delete.extend(work.timesheet_works)
+        if to_create:
+            Timesheet.create(to_create)
+        if to_delete:
+            Timesheet.delete(to_delete)
+        for date, timesheets in list(to_write.items()):
+            Timesheet.write(timesheets, {
+                    'timesheet_end_date': date,
+                    })