mirror of
https://github.com/fastapi-admin/fastapi-admin.git
synced 2025-08-15 03:04:37 +08:00
106 lines
3.3 KiB
Python
106 lines
3.3 KiB
Python
import argparse
|
|
import importlib
|
|
|
|
from colorama import Fore, init
|
|
from prompt_toolkit import PromptSession
|
|
from tortoise import Tortoise, run_async
|
|
|
|
from fastapi_admin import enums
|
|
from fastapi_admin.common import pwd_context
|
|
from fastapi_admin.models import Permission
|
|
|
|
init(autoreset=True)
|
|
|
|
|
|
class Logger:
|
|
@classmethod
|
|
def success(cls, text):
|
|
print(Fore.GREEN + text)
|
|
|
|
@classmethod
|
|
def waring(cls, text):
|
|
print(Fore.YELLOW + text)
|
|
|
|
@classmethod
|
|
def error(cls, text):
|
|
print(Fore.RED + text)
|
|
|
|
|
|
def import_obj(path):
|
|
module_name, class_name = path.rsplit('.', 1)
|
|
return getattr(importlib.import_module(module_name), class_name)
|
|
|
|
|
|
async def init_tortoise(args):
|
|
await Tortoise.init(config=import_obj(args.config))
|
|
|
|
|
|
async def register_permissions(args):
|
|
await init_tortoise(args)
|
|
if args.clean:
|
|
await Permission.all().delete()
|
|
Logger.waring('Cleaned all permissions success.')
|
|
models = Tortoise.apps.get('models').keys()
|
|
models = list(models)
|
|
for model in models:
|
|
for action in enums.PermissionAction:
|
|
label = f'{enums.PermissionAction.choices().get(action)} {model}'
|
|
defaults = dict(
|
|
label=label,
|
|
model=model,
|
|
action=action,
|
|
)
|
|
_, created = await Permission.get_or_create(
|
|
**defaults,
|
|
)
|
|
if created:
|
|
Logger.success(f'Create permission {label} success.')
|
|
|
|
|
|
async def createsuperuser(args):
|
|
await init_tortoise(args)
|
|
|
|
user_model = import_obj(args.user_model)
|
|
prompt = PromptSession()
|
|
while True:
|
|
try:
|
|
username = await prompt.prompt_async('Username: ')
|
|
password = await prompt.prompt_async('Password: ', is_password=True)
|
|
try:
|
|
await user_model.create(
|
|
username=username,
|
|
password=pwd_context.hash(password),
|
|
is_superuser=True
|
|
)
|
|
Logger.success(f'Create superuser {username} success.')
|
|
return
|
|
except Exception as e:
|
|
Logger.error(f'Create superuser {username} error,{e}')
|
|
except (EOFError, KeyboardInterrupt):
|
|
Logger.success(f'Exit success!')
|
|
return
|
|
|
|
|
|
def cli():
|
|
parser = argparse.ArgumentParser()
|
|
subparsers = parser.add_subparsers(title='subcommands')
|
|
parser.add_argument('-c', '--config', required=True,
|
|
help='Tortoise-orm config dict import path,like settings.TORTOISE_ORM.')
|
|
|
|
parser_register_permissions = subparsers.add_parser('register_permissions')
|
|
parser_register_permissions.add_argument('--clean', required=False, action='store_true',
|
|
help='Clean up old permissions then renew.')
|
|
parser_register_permissions.set_defaults(func=register_permissions)
|
|
|
|
parser_createsuperuser = subparsers.add_parser('createsuperuser')
|
|
parser_createsuperuser.add_argument('--user-model', required=True,
|
|
help='User model import path,like examples.models.User.')
|
|
parser_createsuperuser.set_defaults(func=createsuperuser)
|
|
|
|
parse_args = parser.parse_args()
|
|
run_async(parse_args.func(parse_args))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
cli()
|