blob: 8ff2e5ff67958bfd5d4faf95381a99f2da35aac4 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
|
#!/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()
|
