Add filter by status on get stock picking API

1 files changed, 23 insertions, 10 deletions

diff --git a/indoteknik_api/controllers/api_v1/stock_picking.py b/indoteknik_api/controllers/api_v1/stock_picking.py
index e0a60c98..7e246a9f 100644
--- a/indoteknik_api/controllers/api_v1/stock_picking.py
+++ b/indoteknik_api/controllers/api_v1/stock_picking.py
@@ -13,6 +13,7 @@ class StockPicking(controller.Controller):
         get_params = self.get_request_params(kw, {
             'partner_id': ['number'],
             'q': [],
+            'status': [],
             'limit': ['default:0', 'number'],
             'offset': ['default:0', 'number']
         })
@@ -27,13 +28,29 @@ class StockPicking(controller.Controller):
         
         child_ids = request.env['res.partner'].browse(partner_id).get_child_ids()
 
+        pending_domain = [('driver_departure_date', '=', False), ('driver_arrival_date', '=', False)]
+        shipment_domain = [('driver_departure_date', '!=', False), ('driver_arrival_date', '=', False)]
+        completed_domain = [('driver_departure_date', '!=', False), ('driver_arrival_date', '!=', False)]
+
         picking_model = request.env['stock.picking']
-        default_domain = [('partner_id', 'in', child_ids), ('sale_id', '!=', False), ('origin', 'ilike', 'SO%'), ('state', '!=', 'cancel')]
+        default_domain = [
+            ('partner_id', 'in', child_ids), 
+            ('sale_id', '!=', False), 
+            ('origin', 'ilike', 'SO%'), 
+            ('state', '!=', 'cancel')
+        ]
         
-        domain = default_domain
+        domain = default_domain.copy()
         if params['q']:
             query_like = '%' + params['q'].replace(' ', '%') + '%'
             domain += ['|', '|', ('name', 'ilike', query_like), ('sale_id.client_order_ref', 'ilike', query_like), ('delivery_tracking_no', 'ilike', query_like)]
+        
+        if params['status'] == 'pending':
+            domain += pending_domain
+        elif params['status'] == 'shipment':
+            domain += shipment_domain
+        elif params['status'] == 'completed':
+            domain += completed_domain
 
         stock_pickings = picking_model.search(domain, offset=offset, limit=limit, order='create_date desc')
         res_pickings = []
@@ -54,17 +71,13 @@ class StockPicking(controller.Controller):
                 'last_manifest': next(iter(manifests), None)
             })
 
-        pending_count = picking_model.search_count(default_domain + [('driver_departure_date', '=', False), ('driver_arrival_date', '=', False)])
-        shipment_count = picking_model.search_count(default_domain + [('driver_departure_date', '!=', False), ('driver_arrival_date', '=', False)])
-        completed_count = picking_model.search_count(default_domain + [('driver_departure_date', '!=', False), ('driver_arrival_date', '!=', False)])
-
         return self.response({
             'summary': {
-                'pending_count': pending_count,
-                'shipment_count': shipment_count,
-                'completed_count': completed_count
+                'pending_count': picking_model.search_count(default_domain + pending_domain),
+                'shipment_count': picking_model.search_count(default_domain + shipment_domain),
+                'completed_count': picking_model.search_count(default_domain + completed_domain)
             },
-            'picking_total': picking_model.search_count(default_domain),
+            'picking_total': picking_model.search_count(domain),
             'pickings': res_pickings
         })