ohmyapi 0.1.12__py3-none-any.whl → 0.1.13__py3-none-any.whl

ohmyapi/cli.py

@@ -1,4 +1,5 @@
 import asyncio
+import atexit
 import importlib
 import sys
 import typer
@@ -45,6 +46,28 @@ def shell(root: str = "."):
     project_path = Path(root).resolve()
     project = runtime.Project(project_path)
 
+    # Ensure the ORM is shutdown
+    async def close_project():
+        try:
+            await project.close_orm()
+            print("Tortoise ORM closed successfully.")
+        except Exception as e:
+            print(f"Error closing ORM: {e}")
+
+    def cleanup():
+        loop = None
+        try:
+            loop = asyncio.get_running_loop()
+        except RuntimeError:
+            pass
+        if loop and loop.is_running():
+            asyncio.create_task(close_project())
+        else:
+            asyncio.run(close_project())
+
+    # Ensure the ORM is initialized
+    asyncio.run(project.init_orm())
+
     try:
         from IPython import start_ipython
         shell_vars = {
@@ -56,11 +79,13 @@ def shell(root: str = "."):
         c.TerminalInteractiveShell.banner2 = banner.format(**{
             "project_name": f"{f'{project.settings.PROJECT_NAME} ' if getattr(project.settings, 'PROJECT_NAME', '') else ''}[{Path(project_path).resolve()}]",
         })
+        atexit.register(cleanup)
         start_ipython(argv=[], user_ns=shell_vars, config=c)
     except ImportError:
         typer.echo("IPython is not installed. Falling back to built-in Python shell.")
         import code
-        code.interact(local={"settings": project.settings})
+        atexit.register(cleanup)
+        code.interact(local={"p": project})
 
 
 @app.command()

{ohmyapi-0.1.12.dist-info → ohmyapi-0.1.13.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ohmyapi
-Version: 0.1.12
+Version: 0.1.13
 Summary: A Django-like but async web-framework based on FastAPI and TortoiseORM.
 License-Expression: MIT
 Keywords: fastapi,tortoise,orm,async,web-framework
@@ -353,12 +353,17 @@ Out[2]:
  Routes:
   - APIRoute(path='/auth/login', name='login', methods=['POST'])
   - APIRoute(path='/auth/refresh', name='refresh_token', methods=['POST'])
-  - APIRoute(path='/auth/me', name='me', methods=['GET'])
-  - APIRoute(path='/auth/introspect', name='introspect', methods=['GET'])}
+  - APIRoute(path='/auth/introspect', name='introspect', methods=['GET'])
+  - APIRoute(path='/auth/me', name='me', methods=['GET']),
+ 'tournament': App: tournament
+ Models:
+  - Tournament
+  - Event
+  - Team
+ Routes:
+  - APIRoute(path='/tournament/', name='list', methods=['GET'])}
 
 In [3]: from tournament.models import Tournament
-Out[3]:
-
 ```
 
 

{ohmyapi-0.1.12.dist-info → ohmyapi-0.1.13.dist-info}/RECORD

@@ -4,7 +4,7 @@ ohmyapi/builtin/auth/__init__.py,sha256=TY1RKgwWmJ6FKz_v4J3m0Ang69qSmtVDLe4rqjLk
 ohmyapi/builtin/auth/models.py,sha256=Xsxn9m5RTgY2a0PPfW3wTj77ocuuISytdl4ec_TR_kw,1524
 ohmyapi/builtin/auth/permissions.py,sha256=jf-I2b9rIOw2EF4Kga-_Bz1ZPPHU0vHNaXGrDQSwhSI,145
 ohmyapi/builtin/auth/routes.py,sha256=iMxmgP6TSpMrtSKf-g7mc60NcOiw8FRkngUroalGCyk,5519
-ohmyapi/cli.py,sha256=qCGi9GtgybCh_jb_NVwu9dlG9XTbilg6qnXop2IIgC8,3768
+ohmyapi/cli.py,sha256=Mip1eBKYyu1hmUHbNW-1kSMc05EJOTVXqdI0c0r_B4g,4440
 ohmyapi/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ohmyapi/core/runtime.py,sha256=l5zffc4VTwQOo7G9mfiYHsPGapMONRX_jtO_z9QaKHU,8577
 ohmyapi/core/scaffolding.py,sha256=iMymscokJ-YqzB0ZTC-gcc2T71o73577j9tnb2x8lH8,2572
@@ -19,7 +19,7 @@ ohmyapi/db/exceptions.py,sha256=I7AubrdqQF_UvAvzKqz2ve08-BkXHzEWXnwG300StHE,35
 ohmyapi/db/model/__init__.py,sha256=k3StTNuKatpwZo_Z5JBFa-927eJrzibFE8U4SA82asc,32
 ohmyapi/db/model/model.py,sha256=BajFtLlQ1s0mZ2hj-_JNQhLQmxuVe-Lw2LuW5t2C7Rw,1579
 ohmyapi/router.py,sha256=hutccsrP9RT8W5O6uBDhOJehwqrkRoPzaUI5zoHPh9A,55
-ohmyapi-0.1.12.dist-info/METADATA,sha256=i8wVhc1RwYUKSXXQmffQCMcuawbKmD07yu48rxK3zOY,8673
-ohmyapi-0.1.12.dist-info/WHEEL,sha256=M5asmiAlL6HEcOq52Yi5mmk9KmTVjY2RDPtO4p9DMrc,88
-ohmyapi-0.1.12.dist-info/entry_points.txt,sha256=wb3lw8-meAlpiv1mqcQ3m25ukL7djagU_w89GkrC37k,43
-ohmyapi-0.1.12.dist-info/RECORD,,
+ohmyapi-0.1.13.dist-info/METADATA,sha256=UcPMZIqwNTN5LrJ29RYgyLbhMUmCHsVMGm5UFCOHdU0,8812
+ohmyapi-0.1.13.dist-info/WHEEL,sha256=M5asmiAlL6HEcOq52Yi5mmk9KmTVjY2RDPtO4p9DMrc,88
+ohmyapi-0.1.13.dist-info/entry_points.txt,sha256=wb3lw8-meAlpiv1mqcQ3m25ukL7djagU_w89GkrC37k,43
+ohmyapi-0.1.13.dist-info/RECORD,,