Create public_api_inventory.py

public_api/public_api_inventory.py

@@ -0,0 +1,55 @@
+import requests
+from public_api.public_api_auth import login
+
+def get_inventory_holds(session_data, warehouse_id, **kwargs):
+    url = "http://euwdsw202em01:4660/api/inventory/v1/inventoryHolds"
+    session = session_data["wms_auth"]
+    
+    # Build dynamic params from kwargs
+    params = {}
+    if kwargs.get("lodnum"): params["lodnum"] = kwargs.get("lodnum")
+    if kwargs.get("subnum"): params["subnum"] = kwargs.get("subnum")
+    if kwargs.get("dtlnum"): params["dtlnum"] = kwargs.get("dtlnum")
+        
+    headers = {
+        "accept": "application/json",
+        "Content-Type": "application/json"
+    }
+
+    try:
+        response = session.get(url, params=params, headers=headers, verify=False)
+        
+        # 1. Logic for first attempt 401 and retry
+        if response.status_code == 401:
+            print("Session expired. Re-logging in...")
+            new_session, new_wh, success = login()
+            if success:
+                session_data["wms_auth"] = new_session
+                session_data["user_warehouse_id"] = new_wh
+                # Retry immediately with new session
+                response = new_session.get(url, params=params, headers=headers, verify=False)
+
+        # 2. Handle Success (200)
+        if response.status_code == 200:
+            try:
+                res_json = response.json()
+                all_data = res_json.get("data", [])
+                
+                # Dynamic Filtering: Only include records matching the session warehouse
+                # Using 'wh_id' as per your existing logic
+                filtered_data = [
+                    attr for attr in all_data 
+                    if attr.get("wh_id") == warehouse_id
+                ]
+                
+                return {"data": filtered_data, "error": None}
+            except ValueError:
+                return {"data": [], "error": "Invalid JSON format from Holds API."}
+        
+        # 3. Handle all other error codes (404, 500, etc.)
+        print(f"Holds API response code: {response.status_code}\n{response.text}")
+        return {"data": [], "error": f"Holds API Error: {response.text}"}
+
+    except Exception as e:
+        print(f"Holds API exception: {e}")
+        return {"data": [], "error": f"Holds API Connection failed: {str(e)}"}