first commit

modules/production_work_timesheet/timesheet.py

@@ -0,0 +1,19 @@
+# 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 trytond.pool import Pool, PoolMeta
+
+
+class Work(metaclass=PoolMeta):
+    __name__ = 'timesheet.work'
+
+    @classmethod
+    def _get_origin(cls):
+        return super()._get_origin() + ['production.work']
+
+    def _validate_company(self):
+        pool = Pool()
+        ProductionWork = pool.get('production.work')
+        result = super()._validate_company()
+        if isinstance(self.origin, ProductionWork):
+            result &= self.company == self.origin.company
+        return result