paradizelost/paperless-ngx
1
0
Fork 0
mirror of https://github.com/paperless-ngx/paperless-ngx.git synced 2025-10-30 03:56:23 -05:00
Code Issues Packages Projects Releases Wiki Activity

Fix code guidelines

Browse Source
This commit is contained in:
Chris Nagy
2021-04-10 16:10:45 +02:00
parent 79ca66fa14
commit 38dc806b43
1 changed files with 18 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
src/documents/management/commands/manage_superuser.py
View File

@@ -3,6 +3,7 @@ import os
from django.contrib.auth.models import User
from django.core.management.base import BaseCommand, CommandError
class Command(BaseCommand):
help = """
@@ -11,23 +12,29 @@ class Command(BaseCommand):
def handle(self, *args, **options):
# Get user details from env variables
PAPERLESS_ADMIN_USER=os.getenv('PAPERLESS_ADMIN_USER')
PAPERLESS_ADMIN_MAIL=os.getenv('PAPERLESS_ADMIN_MAIL', 'root@localhost')
PAPERLESS_ADMIN_PASSWORD=os.getenv('PAPERLESS_ADMIN_PASSWORD')
PAPERLESS_ADMIN_USER = os.getenv('PAPERLESS_ADMIN_USER')
PAPERLESS_ADMIN_MAIL = os.getenv('PAPERLESS_ADMIN_MAIL', 'root@localhost')
PAPERLESS_ADMIN_PASSWORD = os.getenv('PAPERLESS_ADMIN_PASSWORD')
# If PAPERLESS_ADMIN_USER env variable is set
if PAPERLESS_ADMIN_USER:
try:
# Check if user exists already, leave as is if it does
if User.objects.filter(username=PAPERLESS_ADMIN_USER).exists():
self.stdout.write(f'The user "{PAPERLESS_ADMIN_USER}" already exists! Leaving user as is.')
if User.objects.filter(
username=PAPERLESS_ADMIN_USER).exists():
self.stdout.write(
f'The user "{PAPERLESS_ADMIN_USER}" already exists! Leaving user as is.')
elif PAPERLESS_ADMIN_PASSWORD:
# Create superuser based on env variables
User.objects.create_superuser(PAPERLESS_ADMIN_USER, PAPERLESS_ADMIN_MAIL, PAPERLESS_ADMIN_PASSWORD)
self.stdout.write(f'Created superuser "{PAPERLESS_ADMIN_USER}" with provided password.')
User.objects.create_superuser(
PAPERLESS_ADMIN_USER, PAPERLESS_ADMIN_MAIL, PAPERLESS_ADMIN_PASSWORD)
self.stdout.write(
f'Created superuser "{PAPERLESS_ADMIN_USER}" with provided password.')
else:
self.stdout.write(f'Did not create superuser "{PAPERLESS_ADMIN_USER}".')
self.stdout.write('Make sure you specified "PAPERLESS_ADMIN_PASSWORD" in your "docker-compose.env" file.')
self.stdout.write(
f'Did not create superuser "{PAPERLESS_ADMIN_USER}".')
self.stdout.write(
'Make sure you specified "PAPERLESS_ADMIN_PASSWORD" in your "docker-compose.env" file.')
except Exception as error:
self.stdout.write(f'Exception occured while creating superuser: {error}')
self.stdout.write(
f'Exception occured while creating superuser: {error}')