From f38664721ed117f445b27af839fe41898ed90a47 Mon Sep 17 00:00:00 2001
From: prathamesh
Date: Wed, 24 Mar 2021 12:56:33 +0530
Subject: Modify QR code functionality, courtesy @adityacp

---
 yaksh/models.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

(limited to 'yaksh/models.py')

diff --git a/yaksh/models.py b/yaksh/models.py
index d792205..efe772c 100644
--- a/yaksh/models.py
+++ b/yaksh/models.py
@@ -3360,8 +3360,10 @@ class QRcode(models.Model):
 
     def generate_image(self, content):
         img = qrcode.make(content)
-        path = os.path.join(settings.MEDIA_ROOT, 'qrcode',
-                            '{0}.png'.format(self.short_key))
+        qr_dir = os.path.join(settings.MEDIA_ROOT, 'qrcode')
+        if not os.path.exists(qr_dir):
+            os.makedirs(qr_dir)
+        path = os.path.join(qr_dir, f'{self.short_key}.png')
         img.save(path)
         self.image = os.path.join('qrcode', '{0}.png'.format(self.short_key))
         self.activate()
-- 
cgit