reyserver 1.1.38__py3-none-any.whl → 1.1.39__py3-none-any.whl

reyserver/rbase.py

@@ -9,12 +9,17 @@
 """
 
 
-from reykit.rbase import Base
+from typing import Sequence
+from inspect import iscoroutinefunction
+from contextlib import asynccontextmanager
+from fastapi import FastAPI
+from reykit.rbase import CoroutineFunctionSimple, Base, is_iterable
 
 
 __all__ = (
     'ServerBase',
-    'ServerAPI'
+    'ServerAPI',
+    'generate_lifespan'
 )
 
 
@@ -28,3 +33,50 @@ class ServerAPI(ServerBase):
     """
     Server API type.
     """
+
+
+def generate_lifespan(
+    before: CoroutineFunctionSimple | Sequence[CoroutineFunctionSimple] | None = None,
+    after: CoroutineFunctionSimple | Sequence[CoroutineFunctionSimple] | None = None,
+):
+    """
+    Generate function of lifespan manager.
+
+    Parameters
+    ----------
+    before : Execute before server start.
+    after : Execute after server end.
+    """
+
+    # Parameter.
+    if before is None:
+        before = ()
+    elif iscoroutinefunction(before):
+        before = (before,)
+    if after is None:
+        after = ()
+    elif iscoroutinefunction(after):
+        after = (after,)
+
+    # Define.
+    @asynccontextmanager
+    async def lifespan(app: FastAPI):
+        """
+        Server lifespan manager.
+
+        Parameters
+        ----------
+        app : Server APP.
+        """
+
+        # Before.
+        for task in before:
+            await task()
+        yield
+
+        # After.
+        for task in after:
+            await after()
+
+
+    return lifespan

reyserver/rdb.py

@@ -0,0 +1,11 @@
+# !/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+"""
+@Time    : 2025-10-07 18:03:06
+@Author  : Rey
+@Contact : reyxbo@163.com
+@Explain : Database methods.
+"""
+
+

reyserver/rserver.py

@@ -10,14 +10,16 @@
 
 
 from typing import Literal
+from inspect import iscoroutinefunction
 from collections.abc import Sequence, Callable, Coroutine
-from fastapi import FastAPI
+from fastapi import FastAPI, Depends
 from fastapi.middleware.gzip import GZipMiddleware
 from fastapi.staticfiles import StaticFiles
 from uvicorn import run as uvicorn_run
+from contextlib import asynccontextmanager
 from reykit.rbase import CoroutineFunctionSimple
 
-from .rbase import ServerBase
+from .rbase import ServerBase, generate_lifespan
 
 
 __all__ = (
@@ -48,31 +50,43 @@ class Server(ServerBase):
         ----------
         public : Public directory.
         depend : Global api dependencies.
-        before : Before 
+        before : Execute before server start.
+        after : Execute after server end.
         """
 
         # Parameter.
         if type(ssl_cert) != type(ssl_key):
             raise
+        lifespan = generate_lifespan(before, after)
+        if iscoroutinefunction(depend):
+            depend = (depend,)
+        dependencies = [
+            Depends(depend)
+            for task in depend
+        ]
 
         # Build.
-        self.app = FastAPI()
-        # self.index = Folder(public) + 'index.html'
         self.ssl_cert = ssl_cert
         self.ssl_key = ssl_key
 
-        ## Middleware.
-        self.app.add_middleware(GZipMiddleware)
-        # self.app.add_middleware(TrustedHostMiddleware)
-        # self.app.add_middleware(HTTPSRedirectMiddleware)
+        ## App.
+        self.app = FastAPI(
+            dependencies=dependencies,
+            lifespan=lifespan
+        )
 
         ## Static.
         if public is not None:
             subapp = StaticFiles(directory=public, html=True)
             self.app.mount('/', subapp)
 
+        ## Middleware.
+        self.app.add_middleware(GZipMiddleware)
+        # self.app.add_middleware(TrustedHostMiddleware)
+        # self.app.add_middleware(HTTPSRedirectMiddleware)
+
 
-    def run(self):
+    def run(self) -> None:
         """
         Run.
         """
@@ -85,23 +99,26 @@ class Server(ServerBase):
         )
 
 
-    def add_api_all(self):
+    # def add_api_all(self) -> None:
+    #     """
+    #     Add all API.
+    #     """
 
-        self.add_api_all()
+    #     self.add_api_all()
 
 
-    def add_api_base(self):
+    # def add_api_base(self):
 
-        # @self.app.get('/')
-        # def index():
-        #     file_bytes = File(self.index).bytes
-        #     response = HTMLResponse(file_bytes)
-        #     return response
+    #     # @self.app.get('/')
+    #     # def index():
+    #     #     file_bytes = File(self.index).bytes
+    #     #     response = HTMLResponse(file_bytes)
+    #     #     return response
 
 
-        @self.app.get('/test')
-        def test():
-            return {'message': 'test'}
+    #     @self.app.get('/test')
+    #     def test():
+    #         return {'message': 'test'}
 
 
     def add_api_file(self): ...

{reyserver-1.1.38.dist-info → reyserver-1.1.39.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: reyserver
-Version: 1.1.38
+Version: 1.1.39
 Summary: Backend server method set.
 Project-URL: homepage, https://github.com/reyxbo/reyserver/
 Author-email: Rey <reyxbo@163.com>

reyserver-1.1.39.dist-info/RECORD

@@ -0,0 +1,10 @@
+reyserver/__init__.py,sha256=ZNzM6wBvBSXe1LgV40ZJ1WIAbejWNYGl8x22dmt8C60,289
+reyserver/rall.py,sha256=F-vUJIf5DgP0FXI0pdTrEeEiMtN-g81_yTo4ibUL9xk,233
+reyserver/rbase.py,sha256=EVLCkX2xdPNdPwkRnZ2ZSjM7kctnMoQ_FSlSNHSZt2E,1630
+reyserver/rdb.py,sha256=dfM3alvIII_eDgSCYhGfmQH7kh6eYsDmz4DaprYy9c8,170
+reyserver/rfile.py,sha256=P3i1wa2JYsgNFrm0pDCsDN_zFo_Xpv4XigePotVzn8s,8213
+reyserver/rserver.py,sha256=e-AFB75511HTOf7XjWerCNq86-QEm_2C81psMrldxls,3085
+reyserver-1.1.39.dist-info/METADATA,sha256=EgXGwkne3cVZIovlCdQEjTngnnpWIjWVY35AO89_TTA,1549
+reyserver-1.1.39.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+reyserver-1.1.39.dist-info/licenses/LICENSE,sha256=UYLPqp7BvPiH8yEZduJqmmyEl6hlM3lKrFIefiD4rvk,1059
+reyserver-1.1.39.dist-info/RECORD,,

reyserver-1.1.38.dist-info/RECORD

@@ -1,9 +0,0 @@
-reyserver/__init__.py,sha256=ZNzM6wBvBSXe1LgV40ZJ1WIAbejWNYGl8x22dmt8C60,289
-reyserver/rall.py,sha256=F-vUJIf5DgP0FXI0pdTrEeEiMtN-g81_yTo4ibUL9xk,233
-reyserver/rbase.py,sha256=TtCmqjofwKkE0vdOIc5WhWxbgAUd8pFgx26-VON11yA,397
-reyserver/rfile.py,sha256=P3i1wa2JYsgNFrm0pDCsDN_zFo_Xpv4XigePotVzn8s,8213
-reyserver/rserver.py,sha256=NyI9_evkZ2uGXajI75X8Paj18N1S2f9xUGgLGeQ_2ag,2546
-reyserver-1.1.38.dist-info/METADATA,sha256=CuZu-kkGUgw0qSvNpzMYVlhIhvF2QL7uG36GZuEn9qE,1549
-reyserver-1.1.38.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-reyserver-1.1.38.dist-info/licenses/LICENSE,sha256=UYLPqp7BvPiH8yEZduJqmmyEl6hlM3lKrFIefiD4rvk,1059
-reyserver-1.1.38.dist-info/RECORD,,