python/fastapi-admin
1
0
Fork 0
mirror of https://github.com/fastapi-admin/fastapi-admin.git synced 2025-08-15 03:04:37 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix cli error

Browse Source
This commit is contained in:
long2ice
2020-05-22 13:33:56 +08:00
parent 4f48d74970
commit ae4eb02145
4 changed files with 12 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
fastapi_admin/cli.py
View File

@ -28,8 +28,10 @@ class Logger:
def import_obj(path):
module_name, class_name = path.rsplit('.', 1)
return getattr(importlib.import_module(module_name), class_name)
splits = path.split(".")
module = ".".join(splits[:-1])
class_name = splits[-1]
return getattr(importlib.import_module(module), class_name)
async def init_tortoise(args):
@ -63,7 +65,7 @@ async def register_permissions(args):
async def createsuperuser(args):
await init_tortoise(args)
user_model = import_obj(args.user_model)
user_model = Tortoise.apps.get('models').get(args.user_model)
prompt = PromptSession()
while True:
try:
@ -104,6 +106,6 @@ def cli():
run_async(parse_args.func(parse_args))
if __name__ == '__main__':
def main():
sys.path.insert(0, ".")
cli()