Disable part parameter enforce units setting

sparkmicro · Jan 16, 2024 · 3632d03 · 3632d03
1 parent d1f9eb3
commit 3632d03
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/kintree/database/inventree_api.py b/kintree/database/inventree_api.py
@@ -50,6 +50,13 @@ def get_inventree_api_timeout():
     return False
 
 
+def set_inventree_db_test_mode():
+    ''' InvenTree test database setup '''
+    global inventree_api
+
+    inventree_api.patch('settings/global/PART_PARAMETER_ENFORCE_UNITS', {'value': False})
+
+
 def get_inventree_category_id(category_tree: list) -> int:
     ''' Get InvenTree category ID from name, specificy parent if subcategory '''
     global inventree_api

diff --git a/kintree/setup_inventree.py b/kintree/setup_inventree.py
@@ -32,6 +32,9 @@ def create_categories(parent, name, categories):
         if not inventree_connect:
             sys.exit(-1)
 
+        # Setup database for test
+        inventree_api.set_inventree_db_test_mode()
+
     if SETUP_CATEGORIES:
         for category in categories.keys():
             cprint(f'\n[MAIN]\tCreating categories in {category.upper()}')