paperless-ngx/src/documents/management/commands/manage_superuser.py

import logging
import os

from django.contrib.auth.models import User
from django.core.management.base import BaseCommand


logger = logging.getLogger("paperless.management.superuser")


class Command(BaseCommand):

    help = """
        Creates a Django superuser based on env variables.
    """.replace(
        "    ",
        "",
    )

    def handle(self, *args, **options):

        username = os.getenv("PAPERLESS_ADMIN_USER")
        if not username:
            return

        mail = os.getenv("PAPERLESS_ADMIN_MAIL", "root@localhost")
        password = os.getenv("PAPERLESS_ADMIN_PASSWORD")

        # Check if user exists already, leave as is if it does
        if User.objects.filter(username=username).exists():
            user: User = User.objects.get_by_natural_key(username)
            user.set_password(password)
            user.save()
            self.stdout.write(f"Changed password of user {username}.")
        elif password:
            # Create superuser based on env variables
            User.objects.create_superuser(username, mail, password)
            self.stdout.write(f'Created superuser "{username}" with provided password.')
        else:
            self.stdout.write(f'Did not create superuser "{username}".')
            self.stdout.write(
                'Make sure you specified "PAPERLESS_ADMIN_PASSWORD" in your '
                '"docker-compose.env" file.',
            )