Commit c8c0479c authored by ibuler's avatar ibuler

[Update] 修改清理用户组脚本

parent 30afcecf
...@@ -17,9 +17,14 @@ os.environ.setdefault("DJANGO_SETTINGS_MODULE", "jumpserver.settings") ...@@ -17,9 +17,14 @@ os.environ.setdefault("DJANGO_SETTINGS_MODULE", "jumpserver.settings")
django.setup() django.setup()
from users.models import UserGroup from users.models import UserGroup
from django.core.exceptions import FieldError
def clean_group(interactive=True): def clean_group(interactive=True):
try:
UserGroup.objects.all().filter(is_discard=True).delete()
except FieldError:
pass
groups = UserGroup.objects.all() groups = UserGroup.objects.all()
groups_name_list = groups.values_list('name', flat=True) groups_name_list = groups.values_list('name', flat=True)
groups_with_info = groups.annotate(Count('users'))\ groups_with_info = groups.annotate(Count('users'))\
...@@ -50,7 +55,7 @@ def clean_group(interactive=True): ...@@ -50,7 +55,7 @@ def clean_group(interactive=True):
"Delete user group <{}>, create at {}? ([y]/n)".format( "Delete user group <{}>, create at {}? ([y]/n)".format(
name, group.date_created) name, group.date_created)
) )
if confirm.lower() == "y": if confirm.lower() in ["y", ""]:
confirm = True confirm = True
break break
elif confirm.lower() == "n": elif confirm.lower() == "n":
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment