mirror of
https://github.com/jointakahe/takahe.git
synced 2024-11-14 03:11:46 +00:00
38 lines
1.1 KiB
Python
Executable file
38 lines
1.1 KiB
Python
Executable file
#!/usr/bin/env python
|
|
"""Django's command-line utility for administrative tasks."""
|
|
import os
|
|
import sys
|
|
|
|
# List of settings files that should guard against running certain commands
|
|
GUARDED_ENVIRONMENTS = [
|
|
"production",
|
|
]
|
|
|
|
GUARDED_COMMANDS = [
|
|
"test",
|
|
]
|
|
|
|
|
|
def main():
|
|
"""Run administrative tasks."""
|
|
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "takahe.settings.production")
|
|
|
|
# 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}")
|
|
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()
|