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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
|
#!/usr/bin/env python
import os
import re
import socket
import sys
import time
from typing import List
import django
import pyuwsgi
from django.contrib.auth import get_user_model
from django.core.management import call_command, execute_from_command_line
DEFAULT_ENVS = {
"DJANGO_SETTINGS_MODULE": "pydis_site.settings",
"SUPER_USERNAME": "admin",
"SUPER_PASSWORD": "admin"
}
for key, value in DEFAULT_ENVS.items():
os.environ.setdefault(key, value)
class SiteManager:
"""
Manages the preparation and serving of the website.
Handles both development and production environments.
Usage:
manage.py run [option]...
Options:
--debug Runs a development server with debug mode enabled.
--silent Sets no output in console for preparation commands.
--verbose Sets verbose output for preparation commands.
"""
def __init__(self, args: List[str]):
self.debug = "--debug" in args
self.silent = "--silent" in args
if self.silent:
self.verbosity = 0
else:
self.verbosity = 2 if "--verbose" in args else 1
if self.verbosity and self.debug:
os.environ.setdefault("DEBUG", "true")
print("Starting in debug mode.")
@staticmethod
def create_superuser() -> None:
"""Create a default django admin super user in development environments."""
print("Creating a superuser.")
name = os.environ["SUPER_USERNAME"]
password = os.environ["SUPER_PASSWORD"]
user = get_user_model()
if user.objects.filter(username=name).exists():
return print('Admin superuser already exists')
user.objects.create_superuser(name, '', password)
@staticmethod
def wait_for_postgres() -> None:
"""Wait for the PostgreSQL database specified in DATABASE_URL."""
print("Waiting for PostgreSQL database.")
# Get database URL based on environmental variable passed in compose
database_url = os.environ["DATABASE_URL"]
match = re.search(r"@(\w+):(\d+)/", database_url)
if not match:
raise OSError("Valid DATABASE_URL environmental variable not found.")
domain = match.group(1)
port = int(match.group(2))
# Attempt to connect to the database socket
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
while True:
try:
# Ignore 'incomplete startup packet'
s.connect((domain, port))
s.shutdown(socket.SHUT_RDWR)
print("Database is ready.")
break
except socket.error:
print("Not ready yet, retrying.")
time.sleep(0.5)
def prepare_server(self) -> None:
"""Perform preparation tasks before running the server."""
django.setup()
if self.debug:
self.wait_for_postgres()
self.create_superuser()
print("Applying migrations.")
call_command("migrate", verbosity=self.verbosity)
print("Collecting static files.")
call_command("collectstatic", interactive=False, clear=True, verbosity=self.verbosity)
def run_server(self) -> None:
"""Prepare and run the web server."""
in_reloader = os.environ.get('RUN_MAIN') == 'true'
# Prevent preparing twice when in debug mode due to reloader
if not self.debug or in_reloader:
self.prepare_server()
print("Starting server.")
# Create a superuser and run the development server
if self.debug:
call_command("runserver", "0.0.0.0:8000")
return
# Run uwsgi for production server
pyuwsgi.run(["--ini", "docker/uwsgi.ini"])
def main() -> None:
"""Entry point for Django management script."""
# Use the custom site manager for launching the server
if len(sys.argv) > 1 and sys.argv[1] == "run":
SiteManager(sys.argv).run_server()
# Pass any others directly to standard management commands
else:
execute_from_command_line(sys.argv)
if __name__ == '__main__':
main()
|
