mirror of
https://github.com/paperless-ngx/paperless-ngx.git
synced 2025-04-02 13:45:10 -05:00
Add superuser management script
This commit is contained in:
parent
e4909c3133
commit
e3ba968fef
@ -49,6 +49,19 @@ wait_for_postgres() {
|
||||
|
||||
}
|
||||
|
||||
superuser() {
|
||||
|
||||
if [[ -n "${PAPERLESS_DBHOST}" ]]
|
||||
then
|
||||
wait_for_postgres
|
||||
fi
|
||||
|
||||
if [[ ! -z "${PAPERLESS_ADMIN_PASSWORD}" ]]
|
||||
then
|
||||
sudo -HEu paperless python3 manage.py manage_superuser
|
||||
fi
|
||||
|
||||
}
|
||||
|
||||
migrations() {
|
||||
|
||||
@ -86,6 +99,7 @@ initialize() {
|
||||
chown -R paperless:paperless /tmp/paperless
|
||||
|
||||
migrations
|
||||
superuser
|
||||
|
||||
}
|
||||
|
||||
|
@ -1,4 +1,4 @@
|
||||
for command in document_archiver document_exporter document_importer mail_fetcher document_create_classifier document_index document_renamer document_retagger document_thumbnails document_sanity_checker;
|
||||
for command in document_archiver document_exporter document_importer mail_fetcher document_create_classifier document_index document_renamer document_retagger document_thumbnails document_sanity_checker manage_superuser;
|
||||
do
|
||||
echo "installing $command..."
|
||||
sed "s/management_command/$command/g" management_script.sh > /usr/local/bin/$command
|
||||
|
30
src/documents/management/commands/manage_superuser.py
Normal file
30
src/documents/management/commands/manage_superuser.py
Normal file
@ -0,0 +1,30 @@
|
||||
import os
|
||||
|
||||
from django.contrib.auth.models import User
|
||||
from django.core.management.base import BaseCommand, CommandError
|
||||
|
||||
class Command(BaseCommand):
|
||||
|
||||
help = """
|
||||
Creates a Django superuser based on env variables.
|
||||
""".replace(" ", "")
|
||||
|
||||
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')
|
||||
|
||||
# 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.')
|
||||
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.')
|
||||
except Exception as error:
|
||||
self.stdout.write(f'Exception occured while managing superuser: {error}')
|
Loading…
x
Reference in New Issue
Block a user