7x31/cmdb
1
0
Fork 0
mirror of https://github.com/veops/cmdb.git synced 2025-08-25 20:09:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(api): click command: add-user #675

Browse Source
This commit is contained in:
pycook
2025-08-14 20:46:57 +08:00
parent a809933a5f
commit f46214aaf8
1 changed files with 15 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
cmdb-api/api/commands/click_acl.py
View File

@@ -1,6 +1,8 @@
import click import click
from flask import current_app
from flask.cli import with_appcontext from flask.cli import with_appcontext
from api.lib.exception import AbortException
from api.lib.perm.acl.user import UserCRUD from api.lib.perm.acl.user import UserCRUD
@@ -46,11 +48,18 @@ def add_user():
email = click.prompt('Enter email ', confirmation_prompt=False) email = click.prompt('Enter email ', confirmation_prompt=False)
is_admin = click.prompt('Admin (Y/N) ', confirmation_prompt=False, type=bool, default=False) is_admin = click.prompt('Admin (Y/N) ', confirmation_prompt=False, type=bool, default=False)
UserCRUD.add(username=username, password=password, email=email) current_app.test_request_context().push()
if is_admin: try:
app = AppCache.get('acl') or App.create(name='acl')
acl_admin = RoleCache.get_by_name(app.id, 'acl_admin') or RoleCRUD.add_role('acl_admin', app.id, True)
rid = RoleCache.get_by_name(None, username).id
RoleRelationCRUD.add(acl_admin, acl_admin.id, [rid], app.id) UserCRUD.add(username=username, password=password, email=email)
if is_admin:
app = AppCache.get('acl') or App.create(name='acl')
acl_admin = RoleCache.get_by_name(None, 'acl_admin') or RoleCRUD.add_role('acl_admin', app.id, True)
rid = RoleCache.get_by_name(None, username).id
RoleRelationCRUD.add(acl_admin, acl_admin.id, [rid], app.id)
except AbortException as e:
print(f"Failed: {e}")