Create salesforce/create_violation.py

salesforce/create_violation.py

@@ -0,0 +1,22 @@
+# salesforce/create_violation.py
+
+import requests
+import json
+
+def create_salesforce_violation_record(violation_data):
+    salesforce_url = f"{SALESFORCE_URL}/services/data/vXX.0/sobjects/Safety_Violation_Log__c/"
+    headers = {
+        'Authorization': f'Bearer {SALESFORCE_TOKEN}',
+        'Content-Type': 'application/json'
+    }
+    violation_obj = {
+        'Site_ID__c': violation_data['site_id'],
+        'Violation_Type__c': violation_data['violation_type'],
+        'Timestamp__c': violation_data['timestamp'],
+        'Snapshot_URL__c': violation_data['snapshot_url'],
+        'Severity__c': violation_data['severity'],
+        'Alert_Sent__c': True,
+        'Resolved__c': False
+    }
+    response = requests.post(salesforce_url, headers=headers, data=json.dumps(violation_obj))
+    return response.json()