50 lines
1.7 KiB
Python
50 lines
1.7 KiB
Python
import os
|
|
from typing import Any, Dict
|
|
|
|
from django.conf import settings
|
|
from django.contrib.staticfiles.management.commands.collectstatic import \
|
|
Command as CollectStatic
|
|
from django.core.management.base import CommandError
|
|
from django.db.utils import OperationalError
|
|
|
|
from ...views import WebclientJavaScriptView
|
|
|
|
|
|
class Command(CollectStatic):
|
|
"""
|
|
Custom collectstatic command.
|
|
"""
|
|
realms = ['site', 'projector']
|
|
js_filename = 'webclient-{}.js'
|
|
|
|
def handle(self, **options: Any) -> str:
|
|
try:
|
|
self.view = WebclientJavaScriptView()
|
|
except OperationalError:
|
|
raise CommandError('You have to run OpenSlides first to create a ' +
|
|
'database before collecting staticfiles.')
|
|
return super().handle(**options)
|
|
|
|
def collect(self) -> Dict[str, Any]:
|
|
try:
|
|
destination_dir = os.path.join(settings.STATICFILES_DIRS[0], 'js')
|
|
except IndexError:
|
|
# If the user does not want do have staticfiles, he should not get
|
|
# the webclient files either.
|
|
pass
|
|
else:
|
|
if not os.path.exists(destination_dir):
|
|
os.makedirs(destination_dir)
|
|
|
|
for realm in self.realms:
|
|
filename = self.js_filename.format(realm)
|
|
content = self.view.get(realm=realm).content
|
|
path = os.path.join(destination_dir, filename)
|
|
with open(path, 'wb+') as f:
|
|
f.write(content)
|
|
self.stdout.write("Written WebclientJavaScriptView for realm {} to '{}'".format(
|
|
realm,
|
|
path))
|
|
|
|
return super().collect()
|