73 lines
2.5 KiB
Python
73 lines
2.5 KiB
Python
import json
|
|
from typing import Literal
|
|
|
|
from fastapi import FastAPI
|
|
from starlette.responses import JSONResponse
|
|
|
|
from app.general.ModuleConfigModule import ModuleConfigModule
|
|
from app.general.general_utils.sql_service import sql_service
|
|
from app.utils.db_utils import AsyncSessionDep
|
|
|
|
|
|
def add_general_route(app: FastAPI):
|
|
types = ['list', 'item', 'insert', 'update', 'delete', 'batch_insert', 'batch_update']
|
|
for type in types:
|
|
@app.post("/general/{module}/" + type)
|
|
async def func(module: str, query_param: dict, session: AsyncSessionDep, type=type):
|
|
try:
|
|
result = await perform_general_operation(session=session, module=module, data=query_param, debug_data=[], type=type)
|
|
if "error" in result:
|
|
return JSONResponse(content=result, status_code=500)
|
|
else:
|
|
return result
|
|
finally:
|
|
await session.close()
|
|
|
|
|
|
async def perform_general_operation(
|
|
session: AsyncSessionDep,
|
|
module: str,
|
|
data: dict,
|
|
debug_data=None,
|
|
type: Literal['list', 'item', 'insert', 'update', 'delete', 'batch_insert', 'batch_update'] = 'query'
|
|
):
|
|
if debug_data is None:
|
|
debug_data = []
|
|
|
|
module_query_result = await get_module_config(session, module, debug_data)
|
|
|
|
if 'error' in module_query_result:
|
|
return module_query_result
|
|
|
|
module_config = module_query_result.get('result', {"": ""}).get('moduleConfig', None)
|
|
|
|
if module_config is not None:
|
|
module_config = json.loads(module_config)
|
|
|
|
if module_config is None:
|
|
return {
|
|
"error": f"Can't find module config match: {module}",
|
|
"body": module_query_result,
|
|
}
|
|
|
|
if type == 'list':
|
|
return await sql_service.list(session, data, module_config, debug_data)
|
|
elif type == 'item':
|
|
return await sql_service.item(session, data, module_config, debug_data)
|
|
elif type == 'insert':
|
|
return await sql_service.insert(session, data, module_config, debug_data)
|
|
elif type == 'update':
|
|
return await sql_service.update(session, data, module_config, debug_data)
|
|
elif type == 'batch_insert':
|
|
return await sql_service.batch_insert(session, data, module_config, debug_data)
|
|
elif type == 'batch_update':
|
|
return await sql_service.batch_update(session, data, module_config, debug_data)
|
|
elif type == 'delete':
|
|
return await sql_service.delete(session, data, module_config, debug_data)
|
|
else:
|
|
return {"error": f"Can't handle operation type: {type}"}
|
|
|
|
|
|
async def get_module_config(session: AsyncSessionDep, module: str, debug_data=None):
|
|
return await sql_service.item(session, {"code": module}, ModuleConfigModule, debug_data)
|