advice

src/agratek/api/merchant/views/merchan.py

@@ -34,11 +34,13 @@ def build_request(typ, vendor_produk, partner_log=None):
         # dict(data=)
     return result
 
+
 def get_vendor_produk():
     return PartnerProduk.query() \
         .join(Partner, Partner.id == PartnerProduk.partner_id) \
         .join(Produk, Produk.id == PartnerProduk.produk_id)
 
+
 @jsonrpc_method(method='inquiry', endpoint='api-merchant')
 def inquiry(request, data, **kwargs):
     """
@@ -285,7 +287,10 @@ def advice(request, data):
     i = 0
     is_list = type(data) == list
     data = is_list and data or [data]
-    customer = Partner.query_user(user)
+    customer = Partner.query_user(user).first()
+    if not customer:
+        raise JsonRpcCustomerNotFoundError
+
     r_data = []
     for dat in data:
         if "invoice_no" not in dat:
@@ -293,9 +298,9 @@ def advice(request, data):
             dat["message"] = "Parameter tidak lengkap"
         else:
             inv_no = dat["invoice_no"]
-            invoice = H2hArInvoice.query()\
-                .filter_by(cust_inv_no = inv_no,
-                           customer_id = customer.id).first()
+            invoice = H2hArInvoice.query() \
+                .filter_by(cust_inv_no=inv_no,
+                           customer_id=customer.id).first()
             if not invoice:
                 dat["status"] = "FAILED"
                 dat["message"] = "Invoice %s Tidak Ditemukan" % inv_no