fix cli error

CHANGELOG.rst

@@ -4,6 +4,10 @@ ChangeLog
 
 0.2
 ===
+0.2.6
+-----
+- Fix createsuperuser error.
+
 0.2.5
 -----
 - Now there has builtin menus.

fastapi_admin/cli.py

@@ -28,8 +28,10 @@ class Logger:
 
 
 def import_obj(path):
-    module_name, class_name = path.rsplit('.', 1)
+    splits = path.split(".")
-    return getattr(importlib.import_module(module_name), class_name)
+    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()

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "fastapi-admin"
-version = "0.1.0"
+version = "0.2.6"
 description = "Fast Admin Dashboard based on fastapi and tortoise-orm and rest-admin."
 authors = ["long2ice <long2ice@prismslight.com>"]
 

setup.py

@@ -34,7 +34,7 @@ setup(
     include_package_data=True,
     zip_safe=True,
     entry_points={
-        'console_scripts': ['fastapi-admin = fastapi_admin.cli:cli'],
+        'console_scripts': ['fastapi-admin = fastapi_admin.cli:main'],
     },
     platforms='any',
     keywords=(