fix bug create payment link

1 files changed, 16 insertions, 16 deletions

diff --git a/indoteknik_custom/models/sale_order.py b/indoteknik_custom/models/sale_order.py
index 2f3871ad..5c223beb 100755
--- a/indoteknik_custom/models/sale_order.py
+++ b/indoteknik_custom/models/sale_order.py
@@ -549,22 +549,22 @@ class SaleOrder(models.Model):
         redirect_url = json.loads(lookup_json)['redirect_url']
         self.payment_link_midtrans = str(redirect_url)
 
-        # Generate QR code
-        qr = qrcode.QRCode(
-            version=1,
-            error_correction=qrcode.constants.ERROR_CORRECT_L,
-            box_size=10,
-            border=4,
-        )
-        qr.add_data(redirect_url)
-        qr.make(fit=True)
-        img = qr.make_image(fill_color="black", back_color="white")
-
-        buffer = BytesIO()
-        img.save(buffer, format="PNG")
-        qr_code_img = base64.b64encode(buffer.getvalue()).decode()
-
-        self.payment_qr_code = qr_code_img
+        if 'redirect_url' in response:
+            qr = qrcode.QRCode(
+                version=1,
+                error_correction=qrcode.constants.ERROR_CORRECT_L,
+                box_size=10,
+                border=4,
+            )
+            qr.add_data(redirect_url)
+            qr.make(fit=True)
+            img = qr.make_image(fill_color="black", back_color="white")
+
+            buffer = BytesIO()
+            img.save(buffer, format="PNG")
+            qr_code_img = base64.b64encode(buffer.getvalue()).decode()
+
+            self.payment_qr_code = qr_code_img
 
     @api.model
     def _generate_so_access_token(self, limit=50):