import os import sys def main(): settings_module = "core.settings.production" if "--develop" in sys.argv or '-d' in sys.argv: settings_module = "core.settings.development" dev_flag = '--develop' if "--develop" in sys.argv else '-d' sys.argv.remove(dev_flag) os.environ.setdefault("DJANGO_SETTINGS_MODULE", settings_module) try: from django.core.management import execute_from_command_line except ImportError as exc: raise ImportError( "Couldn't import Django. Are you sure it's installed and " "available on your PYTHONPATH environment variable? Did you " "forget to activate a virtual environment?" ) from exc execute_from_command_line(sys.argv) if __name__ == "__main__": main()