Allow initconfig to execute on first run

Signed-off-by: Sergio Oliveira <sergio@tracy.com.br>
colab · Aug 31, 2015 · caea32a · caea32a
1 parent 8e394ec
commit caea32a
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/colab/utils/runner.py b/colab/utils/runner.py
@@ -1,14 +1,20 @@
 
 import os
+import sys
 
 from django.core.management import ManagementUtility
-
+from colab.management.commands import initconfig
 
 def execute_from_command_line(argv=None):
     """
     A simple method that runs a ManagementUtility.
     """
     os.environ.setdefault("DJANGO_SETTINGS_MODULE", "colab.settings")
+    from django.conf import settings
 
-    utility = ManagementUtility(argv)
-    utility.execute()
+    if not hasattr(settings, 'SECRET_KEY') and 'initconfig' in sys.argv:
+        command = initconfig.Command()
+        command.handle()
+    else:
+        utility = ManagementUtility(argv)
+        utility.execute()