takahe/manage.py

39 lines
1.1 KiB
Python
Raw Normal View History

2022-11-05 20:17:27 +00:00
#!/usr/bin/env python
"""Django's command-line utility for administrative tasks."""
import os
import sys
2022-11-13 20:57:27 +00:00
# List of settings files that should guard against running certain commands
GUARDED_ENVIRONMENTS = [
"production",
]
2022-11-20 18:57:23 +00:00
GUARDED_COMMANDS = [
2022-11-13 20:57:27 +00:00
"test",
]
2022-11-05 20:17:27 +00:00
def main():
"""Run administrative tasks."""
2022-11-13 05:10:06 +00:00
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "takahe.settings.production")
2022-11-13 20:57:27 +00:00
# Guard against running tests in arbitrary environments
env_name = os.environ["DJANGO_SETTINGS_MODULE"].rsplit(".", 1)[-1]
if env_name in GUARDED_ENVIRONMENTS:
for cmd in sys.argv:
if cmd in GUARDED_COMMANDS:
raise Exception(f"Cannot run {cmd} in {env_name}")
2022-11-05 20:17:27 +00:00
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()