api category tree

1 files changed, 16 insertions, 1 deletions

diff --git a/indoteknik_api/models/sale_order.py b/indoteknik_api/models/sale_order.py
index aa20ccdb..826f2a14 100644
--- a/indoteknik_api/models/sale_order.py
+++ b/indoteknik_api/models/sale_order.py
@@ -11,8 +11,23 @@ class SaleOrder(models.Model):
             'sales': sale_order.user_id.name,
             'amount_total': sale_order.amount_total,
             'purchase_order_name': sale_order.partner_purchase_order_name,
-            'invoice_count': sale_order.invoice_count
+            'invoice_count': sale_order.invoice_count,
+            'status': 'draft',
         }
+        if sale_order.state == 'cancel':
+            data['status'] = 'cancel'
+        if sale_order.state in ['draft', 'sent']:
+            data['status'] = 'draft'
+            if sale_order.approval_status in ['pengajuan1', 'pengajuan2']:
+                data['status'] = 'waiting'
+        if sale_order.state == 'sale': 
+            data['status'] = 'sale'
+            for picking in sale_order.picking_ids:
+                if picking.state == 'assigned':
+                    data['status'] = 'shipping'
+        if sale_order.state == 'done': 
+            data['status'] = 'done'
+        
         if context:
             if context == 'with_detail':
                 res_users = self.env['res.users']