Django에서 createsuperuser를 자동화하는 방법

2024-07-27

django-createsuperuserwithpassword 패키지 사용

설치

pip install django-createsuperuserwithpassword

설정

settings.py 파일에 다음 코드를 추가합니다.

INSTALLED_APPS += (
    'django_createsuperuserwithpassword',
)

사용

python manage.py createsuperuserwithpassword \
--username admin \
--password admin \
--email [email protected] \
--preserve

설명

--username: 슈퍼유저 이름
--password: 슈퍼유저 비밀번호
--email: 슈퍼유저 이메일 주소
--preserve: 기존 슈퍼유저가 있는 경우 그대로 유지

사용자 정의 관리 명령 사용

단계

management 폴더에 __init__.py 파일을 만들어 폴더를 패키지로 만듭니다.
commands라는 이름의 하위 폴더를 만듭니다.
commands 폴더에 __init__.py 파일을 추가하고 슈퍼유저를 만드는 코드를 포함하는 Python 파일을 만듭니다.

예제

# commands/createsuperuser.py

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

class Command(BaseCommand):
    help = 'Creates a superuser'

    def handle(self, *args, **options):
        username = input('Username: ')
        email = input('Email: ')
        password = input('Password: ')

        User.objects.create_superuser(username, email, password)
        self.stdout.write('Superuser created successfully.')

management/commands/createsuperuser.py 파일에 대한 실행 권한을 부여합니다.

chmod +x management/commands/createsuperuser.py

python manage.py createsuperuser

이 명령은 사용자에게 사용자 이름, 이메일 주소 및 비밀번호를 입력하도록 요청합니다.
입력된 정보를 사용하여 슈퍼유저를 만듭니다.

post_migrate 신호 사용

from django.dispatch import receiver
from django.contrib.auth.models import User
from django.db.models.signals import post_migrate

@receiver(post_migrate)
def create_superuser(sender, **kwargs):
    if not User.objects.filter(is_superuser=True).exists():
        User.objects.create_superuser('admin', '[email protected]', 'admin')

INSTALLED_SIGS = (
    'myapp.signals',
)

post_migrate 신호는 데이터베이스 마이그레이션이 완료된 후에 발생합니다.
create_superuser 함수는 이 신호를 수신하고 슈퍼유저가 아직 없는 경우 슈퍼유저를 만듭니다.

주의 사항

슈퍼유저는 강력한 권한을 가지고 있으므로 슈퍼유저를 자동으로 만드는 경우 주의해야 합니다.
프로덕션 환경에서 슈퍼유저를 자동으로 만드는 경우 강력한 비밀번호를 사용하고 다른 보안 조치를 취해야 합니다.

결론

예제 코드

django-createsuperuserwithpassword 패키지 사용

# settings.py

INSTALLED_APPS += (
    'django_createsuperuserwithpassword',
)

python manage.py createsuperuserwithpassword \
--username admin \
--password admin \
--email [email protected] \
--preserve

사용자 정의 관리 명령 사용

# management/__init__.py

# management/commands

# management/commands/createsuperuser.py

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

class Command(BaseCommand):
    help = 'Creates a superuser'

    def handle(self, *args, **options):
        username = input('Username: ')
        email = input('Email: ')
        password = input('Password: ')

        User.objects.create_superuser(username, email, password)
        self.stdout.write('Superuser created successfully.')

chmod +x management/commands/createsuperuser.py

createsuperuser 명령을 사용하여 슈퍼유저를 만듭니다.

python manage.py createsuperuser

post_migrate 신호 사용

# signals.py

from django.dispatch import receiver
from django.contrib.auth.models import User
from django.db.models.signals import post_migrate

@receiver(post_migrate)
def create_superuser(sender, **kwargs):
    if not User.objects.filter(is_superuser=True).exists():
        User.objects.create_superuser('admin', '[email protected]', 'admin')

# settings.py

INSTALLED_SIGS = (
    'myapp.signals',
)

참고

위 코드는 예시이며, 실제 환경에 맞게 수정해야 할 수도 있습니다.
슈퍼유저를 자동으로 만드는 경우에는 보안에 유의해야 합니다.

Django에서 createsuperuser를 자동화하는 대체 방법

Celery 또는 RQ와 같은 작업 큐를 사용합니다.

작업 큐를 사용하면 슈퍼유저 생성 작업을 비동기적으로 수행할 수 있습니다. 이는 프로덕션 환경에서 슈퍼유저를 자동으로 만드는 경우 특히 유용할 수 있습니다.

예시

from celery import Celery
from django.contrib.auth.models import User

app = Celery('tasks')

@app.task
def create_superuser():
    username = 'admin'
    email = '[email protected]'
    password = 'admin'

    User.objects.create_superuser(username, email, password)

# 슈퍼유저 생성 작업을 실행합니다.
create_superuser.delay()

Fabric 또는 Ansible과 같은 구성 관리 도구를 사용합니다.

구성 관리 도구를 사용하면 서버에 슈퍼유저를 만드는 스크립트를 만들 수 있습니다. 이는 여러 서버에 슈퍼유저를 자동으로 만들어야 하는 경우 특히 유용할 수 있습니다.

# Fabric 스크립트

from fabric.api import task

@task
def create_superuser():
    run('python manage.py createsuperuser')

사용자 정의 Django 관리 명령을 사용합니다.

앞서 설명한 방법과 달리 사용자 정의 관리 명령을 사용하면 슈퍼유저 생성 프로세스를 더욱 제어할 수 있습니다. 예를 들어 명령줄 인수를 통해 슈퍼유저 이름, 이메일 주소 및 비밀번호를 지정할 수 있습니다.

# management/commands/createsuperuser.py

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

class Command(BaseCommand):
    help = 'Creates a superuser'

    def add_arguments(self, parser):
        parser.add_argument('--username', dest='username', required=True)
        parser.add_argument('--email', dest='email', required=True)
        parser.add_argument('--password', dest='password', required=True)

    def handle(self, *args, **options):
        username = options['username']
        email = options['email']
        password = options['password']

        User.objects.create_superuser(username, email, password)
        self.stdout.write('Superuser created successfully.')