Merged in production (pull request #80)

Production

1 files changed, 7 insertions, 3 deletions

diff --git a/indoteknik_custom/models/airway_bill.py b/indoteknik_custom/models/airway_bill.py
index 463272b4..37d2b671 100644
--- a/indoteknik_custom/models/airway_bill.py
+++ b/indoteknik_custom/models/airway_bill.py
@@ -41,11 +41,11 @@ class AirwayBill(models.Model):
         self.ensure_one()
         return self._json_decode(self.response)
 
-    def _fetch(self):
+    def _fetch(self, days_before=30):
         # jne, pos, tiki, wahana, jnt, rpx, sap, sicepat, jet, dse, dan first
         carrier_ids = [51, 53, 54, 7, 57, 55, 59, 59, 27, 60, 62, 64]
 
-        delta_time = datetime.now() - timedelta(days=30) # Last 30 days
+        delta_time = datetime.now() - timedelta(days=days_before) # Last 30 days
         delta_time = delta_time.strftime('%Y-%m-%d %H:%M:%S') 
         query = [
             '|',
@@ -61,7 +61,10 @@ class AirwayBill(models.Model):
             history = self._get_waybill_history(out.delivery_tracking_no, rajaongkir.name)
             if not history:
                 continue
-            delivered = history['rajaongkir']['result']['delivered']
+            try:
+                delivered = history['rajaongkir']['result']['delivered']
+            except:
+                delivered = False
             values = {
                 'do_id': out.id,
                 'so_id': out.sale_id.id,
@@ -108,6 +111,7 @@ class AirwayBill(models.Model):
         elif response.get('rajaongkir', {}):
             raja_ongkir = response.get('rajaongkir', {})
         result = raja_ongkir.get('result', {})
+        result = result or {} # Change to empty dict when result is None
         return result.get(key)
 
     def _compute_way_bill(self, airway, key, attribute):