Remove unnecessary/broken features

pull/99/head
Martin Michalec 3 years ago
parent 8803d749db
commit b72d2121bd

@ -1,14 +1,5 @@
from distutils.util import strtobool
from datetime import datetime
import pytz
def argbool(arg):
if arg == None: return None
return bool(strtobool(arg))
def argdate(arg):
if arg == None: return None
try:
return pytz.utc.localize(datetime.strptime(arg, '%Y-%m-%d %H:%M:%S'))
except ValueError:
return pytz.utc.localize(datetime.strptime(arg, '%Y-%m-%d %H:%M:%S.%f'))

@ -1,30 +0,0 @@
from django.core.management.base import BaseCommand
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( 'name'
, type=str
, help="New group's name." )
parser.add_argument( '--permissions'
, type=str
, nargs='*'
, default=[]
, help="New group's permissions." )
def handle(self, *args, **options):
try:
for index,permission in enumerate(options["permissions"]):
options["permissions"][index] = Permission.objects.get(name=permission)
group = Group.objects.create(name=options["name"])
group.permissions.set(options["permissions"])
group.save()
except (IntegrityError,Permission.DoesNotExist) as exception:
self.stdout.write(self.style.ERROR(f'Unable to create group "{options["name"]}": ' + str(exception)))
exit(1)
self.stdout.write(self.style.SUCCESS(f'Group "{options["name"]}" has been created.'))
exit(0)

@ -1,16 +0,0 @@
from django.core.management.base import BaseCommand
from django.contrib.auth.models import Group
class Command(BaseCommand):
def add_arguments(self, parser):
parser.add_argument( 'name'
, type=str
, help="Name of the group to be deleted." )
def handle(self, *args, **options):
try:
Group.objects.get(name = options["name"]).delete()
self.stdout.write(self.style.SUCCESS(f'Grop "{options["name"]}" has been deleted.'))
except Group.DoesNotExist as exception:
self.stdout.write(self.style.ERROR(f'Unable to delete group "{options["name"]}": ' + str(exception)))

@ -1,48 +0,0 @@
from django.core.management.base import BaseCommand
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( 'name'
, type=str
, help="Group's name." )
parser.add_argument( '-n'
, '--new_name'
, '--new-name'
, type=str
, default=None
, help="Group's new name." )
parser.add_argument( '-m'
, '--mode'
, type=str
, choices=['set', 'add', 'remove']
, default='set'
, help="Set modification mode. Affects --permissions." )
parser.add_argument( '--permissions'
, type=str
, nargs='*'
, default=None
, help="Group's new permissions." )
def handle(self, *args, **options):
try:
if options["permissions"] != None:
for index,permission in enumerate(options["permissions"]):
options["permissions"][index] = Permission.objects.get(name=permission)
group = Group.objects.get(name=options["name"])
if options["new_name"] != None:
group.name = options["new_name"]
if options["permissions"] != None:
if options["mode"] == "set" : group.permissions.set ( options["permissions"])
if options["mode"] == "add" : group.permissions.add (*options["permissions"])
if options["mode"] == "remove" : group.permissions.remove(*options["permissions"])
group.save()
self.stdout.write(self.style.SUCCESS(f'Group "{options["name"]}" has been modified.'))
except (User.DoesNotExist, ValueError) as exception:
self.stdout.write(self.style.ERROR(f'Unable to modify group "{options["name"]}": ' + str(exception)))

@ -1,27 +0,0 @@
from django.core.management.base import BaseCommand
from django.contrib.auth.models import Group
class Command(BaseCommand):
def add_arguments(self, parser):
parser.add_argument( '-y'
, '--yes'
, action='store_true'
, default=False
, help="Allow deletion of all groups!" )
def handle(self, *args, **options):
if options["yes"] != True:
print('Do you really want to delete all groups? [y/N]: ', end='')
if input() not in ('y', 'Y', 'yes', 'YES', 'Yes'):
self.stdout.write(self.style.SUCCESS(f'No groups have been deleted.'))
exit(0)
try:
for group in Group.objects.all():
group.delete()
self.stdout.write(self.style.SUCCESS(f'All groups have been deleted.'))
exit(0)
except Group.DoesNotExist as exception:
self.stdout.write(self.style.ERROR(f'Unable to delete all groups: ' + str(exception)))
exit(1)

@ -1,8 +0,0 @@
from django.core.management.base import BaseCommand
from django.contrib.auth.models import Group
class Command(BaseCommand):
def handle(self, *args, **options):
for group in Group.objects.all():
print(group.name)

@ -1,8 +0,0 @@
from django.core.management.base import BaseCommand
from django.contrib.auth.models import Permission
class Command(BaseCommand):
def handle(self, *args, **options):
for permission in Permission.objects.all():
print(permission.name)

@ -1,7 +1,6 @@
from django.core.management.base import BaseCommand
from ._utils import argbool, argdate
from ._utils import argbool
from myauth.models import User
from django.contrib.auth.models import Group, Permission
from django.db.utils import IntegrityError
class Command(BaseCommand):
@ -55,40 +54,9 @@ class Command(BaseCommand):
, 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" ]
@ -96,12 +64,7 @@ class Command(BaseCommand):
, 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"]
, is_active = options["is_active" ] )
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)))

@ -1,16 +0,0 @@
from django.core.management.base import BaseCommand
from myauth.models import User
class Command(BaseCommand):
def add_arguments(self, parser):
parser.add_argument( 'username'
, type=str
, help="Login username of the user to be deleted." )
def handle(self, *args, **options):
try:
User.objects.get(username = options["username"]).delete()
self.stdout.write(self.style.SUCCESS(f'User "{options["username"]}" has been deleted.'))
except User.DoesNotExist as exception:
self.stdout.write(self.style.ERROR(f'Unable to delete user "{options["username"]}": ' + str(exception)))

@ -1,7 +1,6 @@
from django.core.management.base import BaseCommand
from ._utils import argbool, argdate
from ._utils import argbool
from myauth.models import User
from django.contrib.auth.models import Group, Permission
from django.db.utils import IntegrityError
class Command(BaseCommand):
@ -16,10 +15,6 @@ class Command(BaseCommand):
, type=str
, default=None
, help="User's new login username." )
parser.add_argument( '-p'
, '--password'
, type=str
, help="User's new plain text login password." )
parser.add_argument( '-f'
, '--first_name'
, '--first-name'
@ -66,72 +61,21 @@ class Command(BaseCommand):
, const=True
, default=None
, help="Mark/Unmark user as superuser." )
parser.add_argument( '-m'
, '--mode'
, type=str
, choices=['set', 'add', 'remove']
, default='set'
, help="Set modification mode. Affects --groups and --user_permissions." )
parser.add_argument( '-g'
, '--groups'
, type=str
, nargs='*'
, default=None
, help="User's new groups." )
parser.add_argument( '--user_permissions'
, '--user-permissions'
, '--permissions'
, type=str
, nargs='*'
, default=None
, help="User's new user permissions." )
parser.add_argument( '-j'
, '--date_joined'
, '--date-joined'
, type=str
, default=None
, help="User's new date joined, formated as '%Y-%m-%d %H:%M:%S.%f'." )
parser.add_argument( '--last_login'
, '--last-login'
, type=str
, default=None
, help="User's new last login date, formated as '%Y-%m-%d %H:%M:%S.%f'." )
def handle(self, *args, **options):
try:
if options["groups"] != None:
for index,group in enumerate(options["groups"]):
options["groups"][index] = Group.objects.get(name=group)
if options["user_permissions"] != None:
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.get(username = options["username"])
if options["new_username"] != None: user.username = options["new_username"]
if options["password" ] != None: user.password = options["password" ]
if options["email" ] != None: user.email = options["email" ]
if options["first_name" ] != None: user.first_name = options["first_name" ]
if options["last_name" ] != None: user.last_name = options["last_name" ]
if options["is_active" ] != None: user.is_active = options["is_active" ]
if options["is_staff" ] != None: user.is_staff = options["is_staff" ]
if options["is_superuser"] != None: user.is_superuser = options["is_superuser"]
if options["date_joined" ] != None: user.date_joined = options["date_joined" ]
if options["last_login" ] != None: user.last_login = options["last_login" ]
if options["groups"] != None:
if options["mode"] == "set" : user.groups.set ( options["groups"])
if options["mode"] == "add" : user.groups.add (*options["groups"])
if options["mode"] == "remove" : user.groups.remove(*options["groups"])
if options["user_permissions"] != None:
if options["mode"] == "set" : user.user_permissions.set ( options["user_permissions"])
if options["mode"] == "add" : user.user_permissions.add (*options["user_permissions"])
if options["mode"] == "remove" : user.user_permissions.remove(*options["user_permissions"])
user.save()
self.stdout.write(self.style.SUCCESS(f'User "{options["username"]}" has been modified.'))
except (User.DoesNotExist, ValueError) as exception:
self.stdout.write(self.style.ERROR(f'Unable to modify user "{options["username"]}": ' + str(exception)))
exit(1)
self.stdout.write(self.style.SUCCESS(f'User "{options["username"]}" has been modified.'))
exit(0)

@ -1,30 +0,0 @@
from django.core.management.base import BaseCommand
from myauth.models import User
class Command(BaseCommand):
def add_arguments(self, parser):
parser.add_argument( 'usernames'
, default=False
, type=str
, nargs='*'
, default=[]
, help="Delete ALL users!" )
parser.add_argument( '-a'
, '--all'
, action='store_true'
, default=False
, help="Delete ALL users!" )
def handle(self, *args, **options):
try:
if options["all"]:
for user in User.objects.all():
user.delete()
self.stdout.write(self.style.SUCCESS(f'All users have been deleted.'))
else:
for username in options["usernames"]:
User.objects.get(username=username).delete()
self.stdout.write(self.style.SUCCESS(f'Users have been deleted.'))
except User.DoesNotExist as exception:
self.stdout.write(self.style.ERROR(f'Unable to delete users: ' + str(exception)))
Loading…
Cancel
Save