feat: add script demo & timing decorator

Author: NightMarcher

examples/service/main.py examples/service/__main__.pyexamples/service/main.py renamed to examples/service/__main__.py

@@ -1,6 +1,6 @@
 """
 working directory: fastapi-efficient-sql/
-command: python -m examples.service.main
+command: python -m examples.service
 """
 import uvicorn
 

examples/service/routers/account.py

@@ -1,4 +1,4 @@
-import logging
+from logging import getLogger
 from typing import List
 
 from faker import Faker
@@ -9,7 +9,7 @@
 from examples.service.managers.demo.account import AccountMgr
 
 router = APIRouter()
-logger = logging.getLogger(__name__)
+logger = getLogger(__name__)
 
 
 @router.post("/create")

examples/service/routers/basic.py

@@ -1,11 +1,11 @@
 from asyncio import sleep
-import logging
+from logging import getLogger
 from time import perf_counter
 
 from fastapi import APIRouter, BackgroundTasks
 
 router = APIRouter()
-logger = logging.getLogger(__name__)
+logger = getLogger(__name__)
 
 
 @router.get("/ping")

examples/service/scripts/demo.py

@@ -0,0 +1,26 @@
+"""
+python -m examples.service.scripts.demo
+"""
+from asyncio import gather, run, sleep
+
+from fastapi_esql import timing
+
+from .. import app
+
+
+async def work(sec):
+    print(f"starting sleep {sec}")
+    await sleep(sec)
+    print(f"slept {sec}")
+
+
+@timing
+async def main():
+    for func in app.router.on_startup:
+        await func()
+
+    await gather(work(1), work(2), work(3))
+
+
+if __name__ == "__main__":
+    run(main())

fastapi_esql/__init__.py

@@ -1,16 +1,24 @@
 from logging.config import dictConfig
 
+from .const.error import WrongParamsError
 from .orm.base_app import AppMetaclass
 from .orm.base_manager import BaseManager
 from .orm.base_model import BaseModel
+from .utils.cursor_handler import CursorHandler
+from .utils.decorator import timing
 from .utils.sqlizer import RawSQL, SQLizer
 
+__version__ = "0.1.0"
+
 __all__ = [
     "AppMetaclass",
     "BaseManager",
     "BaseModel",
+    "CursorHandler",
     "RawSQL",
     "SQLizer",
+    "WrongParamsError",
+    "timing",
 ]
 
 dictConfig({

fastapi_esql/orm/base_manager.py

@@ -1,4 +1,4 @@
-import logging
+from logging import getLogger
 from typing import Any, Dict, List, Optional
 
 from tortoise.backends.base.client import BaseDBAsyncClient
@@ -9,7 +9,7 @@
 from ..utils.sqlizer import SQLizer
 from ..utils.cursor_handler import CursorHandler
 
-logger = logging.getLogger(__name__)
+logger = getLogger(__name__)
 
 
 class BaseManager(metaclass=AppMetaclass):

fastapi_esql/utils/decorator.py

@@ -0,0 +1,16 @@
+from functools import wraps
+from logging import getLogger
+from time import perf_counter
+
+logger = getLogger(__name__)
+
+
+def timing(func):
+    @wraps(func)
+    async def wrapper(*args, **kwargs):
+        st = perf_counter()
+        result = await func(*args, **kwargs)
+        et = perf_counter()
+        logger.info(f"{func.__module__}/{func.__name__} Cost {1000 * (et - st):.2f} ms")
+        return result
+    return wrapper

fastapi_esql/utils/sqlizer.py

@@ -1,10 +1,10 @@
-import logging
+from logging import getLogger
 from json import dumps
 from typing import Any, Dict, List, Optional
 
 from ..const.error import WrongParamsError
 
-logger = logging.getLogger(__name__)
+logger = getLogger(__name__)
 # ensure the functionality of the RawSQL
 try:
     from tortoise.expressions import RawSQL