Added extra field processing to other suppliers

sparkmicro · Apr 21, 2023 · 7e5041e · 7e5041e
1 parent ac2c82d
commit 7e5041e
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 0 deletions.
diff --git a/kintree/search/element14_api.py b/kintree/search/element14_api.py
@@ -208,6 +208,15 @@ def search_timeout(timeout=10):
         except TypeError:
             # Parameter list is empty
             pass
+
+    # Extra search fields
+    if settings.CONFIG_ELEMENT14.get('EXTRA_FIELDS', None):
+        for extra_field in settings.CONFIG_ELEMENT14['EXTRA_FIELDS']:
+            if part.get(extra_field, None):
+                part_info['parameters'][extra_field] = part[extra_field]
+            else:
+                from ..common.tools import cprint
+                cprint(f'[INFO]\tWarning: Extra field "{extra_field}" not found in search results', silent=False)
 
     # Append Store URL
     # Element14 support said "At this time our API is not structured to provide a URL to product pages in the selected storeInfo.id value."

diff --git a/kintree/search/lcsc_api.py b/kintree/search/lcsc_api.py
@@ -99,6 +99,15 @@ def search_timeout(timeout=10):
         # Parameter list is empty
         pass
 
+    # Extra search fields
+    if settings.CONFIG_LCSC.get('EXTRA_FIELDS', None):
+        for extra_field in settings.CONFIG_LCSC['EXTRA_FIELDS']:
+            if part.get(extra_field, None):
+                part_info['parameters'][extra_field] = part[extra_field]
+            else:
+                from ..common.tools import cprint
+                cprint(f'[INFO]\tWarning: Extra field "{extra_field}" not found in search results', silent=False)
+
     return part_info
 
 

diff --git a/kintree/search/mouser_api.py b/kintree/search/mouser_api.py
@@ -109,6 +109,15 @@ def search_timeout():
         # Append to parameters dictionary
         part_info['parameters'][parameter_name] = parameter_value
 
+    # Extra search fields
+    if settings.CONFIG_MOUSER.get('EXTRA_FIELDS', None):
+        for extra_field in settings.CONFIG_MOUSER['EXTRA_FIELDS']:
+            if part.get(extra_field, None):
+                part_info['parameters'][extra_field] = part[extra_field]
+            else:
+                from ..common.tools import cprint
+                cprint(f'[INFO]\tWarning: Extra field "{extra_field}" not found in search results', silent=False)
+
     return part_info