From 44c28fa1043a03669fd71ca4ce709b210f4bfddd Mon Sep 17 00:00:00 2001
From: "rx.wu" <rx.wu@trip.com>
Date: Tue, 14 Nov 2023 15:47:20 +0800
Subject: [PATCH] feat: add custom router

---
 app/routers/__init__.py | 131 ++++++++++++++++++++++++----------------
 1 file changed, 78 insertions(+), 53 deletions(-)

diff --git a/app/routers/__init__.py b/app/routers/__init__.py
index 153d2b71..04dc54cd 100755
--- a/app/routers/__init__.py
+++ b/app/routers/__init__.py
@@ -1,6 +1,11 @@
-from fastapi import Header
+from collections import namedtuple
+from typing import Dict, List, Type, TypeVar
+
+from fastapi import Header, APIRouter, Depends
+from pydantic import BaseModel
 from starlette import status
 
+from app.crud import Mapper
 from app.crud.auth.UserDao import UserDao
 from app.exception.request import AuthException, PermissionException
 from app.handler.fatcory import PityResponse
@@ -41,55 +46,75 @@ async def get_session():
     async with async_session() as session:
         yield session
 
-# _router: Dict[str, APIRouter] = {}
-#
-#
-# class PityRouter(object):
-#     """
-#     pity路由基类，支持单标crud
-#     """
-#
-#     def __init__(self, group: str, schema: PityModel = None, path: str = "", read: int = Config.MEMBER,
-#                  write: int = Config.MANAGER, operation_log: bool = True):
-#         """
-#         初始化路由相关数据，包括schema，path，权限等
-#         :param group:
-#         :param schema:
-#         :param path:
-#         :param read:
-#         :param write:
-#         :param operation_log:
-#         """
-#         self.group = group
-#         self.schema = schema
-#         self.path = path
-#         self.read = read
-#         self.write = write
-#         self.operation_log = operation_log
-#
-#     def _get_router(self):
-#         router = _router.get(self.group)
-#         if router is None:
-#             router = APIRouter(prefix=self.group)
-#             _router[self.group] = router
-#         return router
-#
-#     def generate_router(self):
-#         router = self._get_router()
-#         router.add_api_route(f"{self.path}/list", self._list(), methods=['GET'], response_model=None)
-#         return router
-#
-#     def _list(self):
-#         schema = self.schema
-#
-#         def route(query: Query):
-#             print(query)
-#             return PityResponse.success()
-#
-#         return route
-#
-#     def _query(self):
-#         pass
-#
-#     def _update(self):
-#         pass
+
+_router: Dict[str, APIRouter] = {}
+T = TypeVar("T", bound=BaseModel)
+DAO = TypeVar("DAO", bound=Mapper)
+CurdParams = namedtuple("parameters", ["name", "type_", "default", "like"])
+
+
+class PityRouter(object):
+    """
+    pity路由基类，支持单标crud
+    """
+
+    def __init__(self, router: APIRouter, schema: Type[T], dao: Type[DAO], business, path, prefix="",
+                 tags: List[str] = None, query_list: List[CurdParams] = None):
+        self.router = router
+        self.schema = schema
+        self.dao = dao
+        self.tags = tags
+        self.business = business
+        self.prefix = prefix
+        self.path = path
+        self.query_list = query_list
+
+    @staticmethod
+    def get_query_parameters(query: List[CurdParams]):
+        params = ",".join(
+            [f"{x.name}: {x.type_}" + (f" = '{x.default}'" if x.default is not None else "") for x in query])
+        return params
+
+    @staticmethod
+    def get_query_sentence(query: List[CurdParams]):
+        return ",".join([f"""{x.name}={x.name if not x.like else f"f'%{{{x.name}}}%'"}""" for x in query])
+
+    def add_all(self):
+        query = [] if self.query_list is None else self.query_list
+
+        async def create(data: self.schema, user_info=Depends(AuthUser())):  # type: ignore
+            model = data.dict()
+            model.pop('id', None)
+            result = await self.dao.insert(self.dao.model(**model, user=user_info.name))
+            return PityResponse.success(result)
+
+        params = ",".join([f"{x.name}: {x.type_}" for x in query])
+        if params:
+            params += ', '
+        loc = dict(Permission=Permission, Depends=Depends, Response=PityResponse, self=self)
+        list_func = f"""
+    async def list_data({PityRouter.get_query_parameters(query)}):
+        result = await self.dao.list_record({PityRouter.get_query_sentence(query)})
+        return Response.ok(Response.model_to_list(result))
+            """
+        exec(list_func, loc)
+
+        async def update(data: self.schema, user_info=Depends(AuthUser())):  # type: ignore
+            result = await self.dao.update_by_id(data, user_info.name)
+            return PityResponse.success(PityResponse.model_to_dict(result))
+
+        async def delete(id: int, user_info=Depends(Permission())):
+            await self.dao.delete_by_id(id)
+            return PityResponse.success()
+
+        self.router.add_api_route(f"{self.prefix}/{self.path}", create, tags=self.tags,
+                                  summary=f"添加{self.business}", methods=['PUT'])
+
+        self.router.add_api_route(f"{self.prefix}/{self.path}", update, tags=self.tags,
+                                  summary=f"编辑{self.business}", methods=['POST'])
+
+        self.router.add_api_route(f"{self.prefix}/{self.path}", delete, tags=self.tags,
+                                  summary=f"删除{self.business}", methods=['DELETE'])
+
+        self.router.add_api_route(f"{self.prefix}/{self.path}", loc.get("list_data"), tags=self.tags,
+                                  summary=f"获取{self.business}", methods=['GET'])