OpenSlides/openslides/core/management/commands/collectstatic.py

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()