-vsi pln post

src/agratek/api/merchant/views/vsi/pln_post.py

@@ -68,8 +68,12 @@ class Vendor(VendorClass):
         power = "segmenpower" in data and data["segmenpower"].split("/") or [""]
         power = len(power)>1 and power[-1:][0] or power[0]
         product_id = self.invoice_det.produk.id
-        partner_id = self.invoice_det.h2h_ar_invoice.customer_id
-        discount = PartnerProduk.get_discount(partner_id, product_id)
+        if hasattr(self.invoice_det,'customer_id'):
+            customer_id = self.invoice_det.customer_id
+        else:
+            customer_id = self.invoice_det.h2h_ar_invoice.customer_id
+
+        discount = PartnerProduk.get_discount(customer_id, product_id)
         admin = self.vendor_produk.produk.harga
         harga = "nominal" in data and data["nominal"] or 0
         subtotal = harga + admin