Update import restrict model

2 files changed, 12 insertions, 6 deletions

diff --git a/indoteknik_custom/models/base_import_import.py b/indoteknik_custom/models/base_import_import.py
index c277f9a2..7c66a228 100644
--- a/indoteknik_custom/models/base_import_import.py
+++ b/indoteknik_custom/models/base_import_import.py
@@ -7,6 +7,9 @@ class Import(models.TransientModel):
 
     def _get_config_enable_import(self):
         return self.env['ir.config_parameter'].get_param('base_import.import.enable_import', '')
+    
+    def _get_config_restrict_model(self):
+        return self.env['ir.config_parameter'].get_param('base_import.import.restrict_model', '')
 
     def _check_time_range(self, config_string):
         current_time = datetime.now().time()
@@ -24,12 +27,18 @@ class Import(models.TransientModel):
 
     def _check_enable_import(self):
         enable_import = self._get_config_enable_import()
+        restrict_model = self._get_config_restrict_model()
+        
+        is_enable_import = False
         if enable_import.lower() == 'true':
-            return True
+            is_enable_import = True
         elif enable_import.lower() == 'false':
-            return False
+            is_enable_import = False
         elif '-' in enable_import:
-            return self._check_time_range(enable_import)
+            is_enable_import = self._check_time_range(enable_import)
+        
+        if not is_enable_import and self.res_model in restrict_model:
+            return False
         
         return True
     
diff --git a/indoteknik_custom/models/purchase_order.py b/indoteknik_custom/models/purchase_order.py
index a53c6df0..3fb61955 100755
--- a/indoteknik_custom/models/purchase_order.py
+++ b/indoteknik_custom/models/purchase_order.py
@@ -4,7 +4,6 @@ from datetime import datetime, timedelta
 import logging
 from pytz import timezone, utc
 import io
-from openpyxl import Workbook
 import base64
 try:
     from odoo.tools.misc import xlsxwriter
@@ -269,8 +268,6 @@ class PurchaseOrder(models.Model):
         
         format6 = workbook.add_format({'font_size': 12, 'align': 'center', 'bg_color': '#D3D3D3', 'bold': True})
         format1 = workbook.add_format({'font_size': 11, 'align': 'center', 'valign': 'vcenter'})
-        format2 = workbook.add_format({'font_size': 10, 'bold': True})
-        format3 = workbook.add_format({'font_size': 10, 'bold': True})
         
         worksheet.set_column(0, 0, 10)  
         worksheet.set_column(1, 1, 20)