<Miqdad> Done

1 files changed, 9 insertions, 5 deletions

diff --git a/indoteknik_api/models/sale_order.py b/indoteknik_api/models/sale_order.py
index 9be03927..c59dead9 100644
--- a/indoteknik_api/models/sale_order.py
+++ b/indoteknik_api/models/sale_order.py
@@ -60,13 +60,17 @@ class SaleOrder(models.Model):
                 # 'tracking_number': picking.delivery_tracking_no or '',
                 # 'delivered': picking.waybill_id.delivered or picking.driver_arrival_date != False or picking.sj_return_date != False,
             })
+
         if sale_order.state == 'cancel':
             data['status'] = 'cancel'
-        if sale_order.state == 'draft' and sale_order.approval_status == False:
-            data['status'] = 'draft'
-        if sale_order.state == 'draft' and sale_order.approval_status in ['pengajuan1', 'pengajuan2']:
-            data['status'] = 'waiting'
-
+        elif sale_order.state == 'draft':
+            if not sale_order.approval_status:
+                data['status'] = 'draft'
+            elif sale_order.approval_status in ('pengajuan1', 'pengajuan2'):
+                if sale_order.payment_status in ('', 'pending', False, None, 'expire'):
+                    data['status'] = 'belum_bayar'
+                elif sale_order.payment_status not in ['', 'pending', False, None, 'expire']:
+                    data['status'] = 'waiting'
 
         if sale_order.state == 'sale':
             bu_pickings = [