add validation on write and create function

1 files changed, 34 insertions, 0 deletions

diff --git a/indoteknik_custom/models/product_template.py b/indoteknik_custom/models/product_template.py
index 29608297..d4b0abf4 100755
--- a/indoteknik_custom/models/product_template.py
+++ b/indoteknik_custom/models/product_template.py
@@ -67,6 +67,23 @@ 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
+    
+    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 _compute_qr_code(self):
     #     for rec in self.product_variant_ids:
     #         qr = qrcode.QRCode(
@@ -403,6 +420,23 @@ 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
+    
+    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 _compute_qr_code_variant(self):
         for rec in self:
             # Skip inactive variants