mirror of
https://github.com/etesync/server
synced 2025-07-05 14:52:33 +00:00
112 lines
5.3 KiB
Python
Executable File
112 lines
5.3 KiB
Python
Executable File
from django.core.management.base import BaseCommand
|
|
from ._utils import argbool, argdate
|
|
from myauth.models import User
|
|
from django.contrib.auth.models import Group, Permission
|
|
from django.db.utils import IntegrityError
|
|
|
|
class Command(BaseCommand):
|
|
|
|
def add_arguments(self, parser):
|
|
parser.add_argument( 'username'
|
|
, type=str
|
|
, help="New user's login username." )
|
|
parser.add_argument( '-p'
|
|
, '--password'
|
|
, type=str
|
|
, help="New user's plain text login password." )
|
|
parser.add_argument( '-f'
|
|
, '--first_name'
|
|
, '--first'
|
|
, type=str
|
|
, default=''
|
|
, help="New user's first name." )
|
|
parser.add_argument( '-l'
|
|
, '--last_name'
|
|
, '--last'
|
|
, type=str
|
|
, default=''
|
|
, help="New user's last name." )
|
|
parser.add_argument( '-e'
|
|
, '--email'
|
|
, type=str
|
|
, default=''
|
|
, help="New user's email address." )
|
|
parser.add_argument( '-a'
|
|
, '--is_active'
|
|
, '--active'
|
|
, nargs='?'
|
|
, type=argbool
|
|
, const=True
|
|
, default=False
|
|
, help="Enable login. [YES]" )
|
|
parser.add_argument( '-s'
|
|
, '--is_staff'
|
|
, '--staff'
|
|
, nargs='?'
|
|
, type=argbool
|
|
, const=True
|
|
, default=False
|
|
, help="Mark user as staff. [NO]" )
|
|
parser.add_argument( '-S'
|
|
, '--is_superuser'
|
|
, '--superuser'
|
|
, nargs='?'
|
|
, type=argbool
|
|
, const=True
|
|
, default=False
|
|
, help="Mark user as superuser. [NO]" )
|
|
parser.add_argument( '-g'
|
|
, '--groups'
|
|
, type=str
|
|
, nargs='*'
|
|
, default=[]
|
|
, help="New user's groups." )
|
|
parser.add_argument( '--user_permissions'
|
|
, '--user-permissions'
|
|
, '--permissions'
|
|
, type=str
|
|
, nargs='*'
|
|
, default=[]
|
|
, help="New user's user permissions." )
|
|
parser.add_argument( '-j'
|
|
, '--date_joined'
|
|
, '--date-joined'
|
|
, type=str
|
|
, default=None
|
|
, help="New user's date joined, formated as '%Y-%m-%d %H:%M:%S.%f'." )
|
|
parser.add_argument( '--last_login'
|
|
, '--last-login'
|
|
, type=str
|
|
, default=None
|
|
, help="New user's last login date, formated as '%Y-%m-%d %H:%M:%S.%f'." )
|
|
|
|
def handle(self, *args, **options):
|
|
try:
|
|
for index,group in enumerate(options["groups"]):
|
|
options["groups"][index] = Group.objects.get(name=group)
|
|
for index,permission in enumerate(options["user_permissions"]):
|
|
options["user_permissions"][index] = Permission.objects.get(name=permission)
|
|
options["date_joined"] = argdate(options["date_joined"])
|
|
options["last_login" ] = argdate(options["last_login" ])
|
|
|
|
user = User.objects.create_user( username = options["username" ]
|
|
, password = options["password" ]
|
|
, email = options["email" ]
|
|
, first_name = options["first_name" ]
|
|
, last_name = options["last_name" ]
|
|
, is_superuser = options["is_superuser" ]
|
|
, is_staff = options["is_staff" ]
|
|
, is_active = options["is_active" ]
|
|
, last_login = options["last_login" ] )
|
|
user.groups.set(options["groups"])
|
|
user.user_permissions.set(options["user_permissions"])
|
|
if options["date_joined"] != None:
|
|
user.date_joined = options["date_joined"]
|
|
user.save()
|
|
except (IntegrityError,Group.DoesNotExist,Permission.DoesNotExist) as exception:
|
|
self.stdout.write(self.style.ERROR(f'Unable to create user "{options["username"]}": ' + str(exception)))
|
|
exit(1)
|
|
|
|
self.stdout.write(self.style.SUCCESS(f'User "{options["username"]}" has been created.'))
|
|
exit(0)
|