Merge pull request #482 from michaeltoop/sms_logger_fix

Fixed logger to acomodate submit_sm_resp_bill and submit_sm_bill
jookies · Oct 6, 2016 · 30e1b21 · 30e1b21
2 parents a13c729 + 492a3d5
commit 30e1b21
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/misc/scripts/sms_logger.py b/misc/scripts/sms_logger.py
@@ -84,7 +84,11 @@ def gotConnection(conn, username, password):
             pdu = pickle.loads(msg.content.body)
             pdu_count = 1
             short_message = pdu.params['short_message']
-            submit_sm_bill = pickle.loads(props['headers']['submit_sm_resp_bill'])
+            billing = props['headers']
+            billing_pickle = billing.get('submit_sm_resp_bill')
+            if not billing_pickle:
+                billing_pickle = billing.get('submit_sm_bill')
+            submit_sm_bill = pickle.loads(billing_pickle)
             source_connector = props['headers']['source_connector']
             routed_cid = msg.routing_key[10:]