fix bug validation create and write

1 files changed, 34 insertions, 30 deletions

diff --git a/indoteknik_custom/models/product_template.py b/indoteknik_custom/models/product_template.py
index 151513e8..99a89d74 100755
--- a/indoteknik_custom/models/product_template.py
+++ b/indoteknik_custom/models/product_template.py
@@ -67,22 +67,24 @@ class ProductTemplate(models.Model):
     print_barcode = fields.Boolean(string='Print Barcode', default=True)
     # qr_code = fields.Binary("QR Code", compute='_compute_qr_code')
 
-    # @api.model
-    # def create(self, vals):
-    #     group_id = self.env.ref('indoteknik_custom.group_role_merchandiser').id
-    #     users_in_group = self.env['res.users'].search([('groups_id', 'in', [group_id])])
-    #     if self.env.user.id not in users_in_group.mapped('id'):
-    #         raise UserError('Hanya MD yang bisa membuat Product')
-    #     result = super(ProductTemplate, self).create(vals)
-    #     return result
+    @api.model
+    def create(self, vals):
+        group_id = self.env.ref('indoteknik_custom.group_role_merchandiser').id
+        users_in_group = self.env['res.users'].search([('groups_id', 'in', [group_id])])
+        active_model = self.env.context.get('active_model')
+        if self.env.user.id not in users_in_group.mapped('id') and active_model == None:
+            raise UserError('Hanya MD yang bisa membuat Product')
+        result = super(ProductTemplate, self).create(vals)
+        return result
     
-    # def write(self, values):
-    #     group_id = self.env.ref('indoteknik_custom.group_role_merchandiser').id
-    #     users_in_group = self.env['res.users'].search([('groups_id', 'in', [group_id])])
-    #     if self.env.user.id not in users_in_group.mapped('id'):
-    #         raise UserError('Hanya MD yang bisa mengedit Product')
-    #     result = super(ProductTemplate, self).write(values)
-    #     return result
+    def write(self, values):
+        group_id = self.env.ref('indoteknik_custom.group_role_merchandiser').id
+        users_in_group = self.env['res.users'].search([('groups_id', 'in', [group_id])])
+        active_model = self.env.context.get('active_model')
+        if self.env.user.id not in users_in_group.mapped('id') and active_model == None:
+            raise UserError('Hanya MD yang bisa mengedit Product')
+        result = super(ProductTemplate, self).write(values)
+        return result
 
     # def _compute_qr_code(self):
     #     for rec in self.product_variant_ids:
@@ -420,22 +422,24 @@ class ProductProduct(models.Model):
     merchandise_ok = fields.Boolean(string='Product Promotion')
     qr_code_variant = fields.Binary("QR Code Variant", compute='_compute_qr_code_variant')
 
-    # @api.model
-    # def create(self, vals):
-    #     group_id = self.env.ref('indoteknik_custom.group_role_merchandiser').id
-    #     users_in_group = self.env['res.users'].search([('groups_id', 'in', [group_id])])
-    #     if self.env.user.id not in users_in_group.mapped('id'):
-    #         raise UserError('Hanya MD yang bisa membuat Product')
-    #     result = super(ProductProduct, self).create(vals)
-    #     return result
+    @api.model
+    def create(self, vals):
+        group_id = self.env.ref('indoteknik_custom.group_role_merchandiser').id
+        active_model = self.env.context.get('active_model')
+        users_in_group = self.env['res.users'].search([('groups_id', 'in', [group_id])])
+        if self.env.user.id not in users_in_group.mapped('id') and active_model == None:
+            raise UserError('Hanya MD yang bisa membuat Product')
+        result = super(ProductProduct, self).create(vals)
+        return result
     
-    # def write(self, values):
-    #     group_id = self.env.ref('indoteknik_custom.group_role_merchandiser').id
-    #     users_in_group = self.env['res.users'].search([('groups_id', 'in', [group_id])])
-    #     if self.env.user.id not in users_in_group.mapped('id'):
-    #         raise UserError('Hanya MD yang bisa mengedit Product')
-    #     result = super(ProductProduct, self).write(values)
-    #     return result
+    def write(self, values):
+        group_id = self.env.ref('indoteknik_custom.group_role_merchandiser').id
+        active_model = self.env.context.get('active_model')
+        users_in_group = self.env['res.users'].search([('groups_id', 'in', [group_id])])
+        if self.env.user.id not in users_in_group.mapped('id') and active_model == None:
+            raise UserError('Hanya MD yang bisa mengedit Product')
+        result = super(ProductProduct, self).write(values)
+        return result
 
     def _compute_qr_code_variant(self):
         for rec in self: