check if keys directory exists before making secret_key (#700)

cvat-ai · Sep 9, 2019 · eaede02 · eaede02
1 parent 8f25bac
commit eaede02
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/cvat/settings/base.py b/cvat/settings/base.py
@@ -33,7 +33,10 @@
     from keys.secret_key import SECRET_KEY
 except ImportError:
     from django.utils.crypto import get_random_string
-    with open(os.path.join(BASE_DIR, 'keys', 'secret_key.py'), 'w') as f:
+    keys_dir = os.path.join(BASE_DIR, 'keys')
+    if not os.path.isdir(keys_dir):
+        os.mkdir(keys_dir)
+    with open(os.path.join(keys_dir, 'secret_key.py'), 'w') as f:
         chars = 'abcdefghijklmnopqrstuvwxyz0123456789!@#$%^&*(-_=+)'
         f.write("SECRET_KEY = '{}'\n".format(get_random_string(50, chars)))
     from keys.secret_key import SECRET_KEY