59 lines
2.7 KiB
Python
59 lines
2.7 KiB
Python
from typing import List
|
|
from fastapi import APIRouter, FastAPI
|
|
|
|
|
|
class RouteRegisterController:
|
|
|
|
def __init__(self, app: FastAPI, router_list: List[APIRouter]):
|
|
self.router_list = router_list
|
|
self.app = app
|
|
|
|
@staticmethod
|
|
def add_router_with_event_to_database(router: APIRouter):
|
|
from schemas import EndpointRestriction
|
|
|
|
# Endpoint operation_id is static now if record exits update() record else create()
|
|
with EndpointRestriction.new_session() as db_session:
|
|
for route in router.routes:
|
|
route_path = str(getattr(route, "path"))
|
|
route_summary = str(getattr(route, "name"))
|
|
operation_id = getattr(route, "operation_id", None)
|
|
if not operation_id:
|
|
raise ValueError(f"Route {route_path} operation_id is not found")
|
|
|
|
for route_method in [
|
|
method.lower() for method in getattr(route, "methods")
|
|
]:
|
|
methods = [method.lower() for method in getattr(route, "methods")]
|
|
print('methods count : ', len(methods))
|
|
print(dict(
|
|
route_method=route_method,
|
|
operation_uu_id=operation_id,
|
|
route_path=route_path,
|
|
route_summary=route_summary,
|
|
))
|
|
# add_or_update_dict = dict(
|
|
# endpoint_method=route_method,
|
|
# endpoint_name=route_path,
|
|
# endpoint_desc=route_summary.replace("_", " "),
|
|
# endpoint_function=route_summary,
|
|
# operation_uu_id=operation_id,
|
|
# is_confirmed=True,
|
|
# )
|
|
# endpoint_restriction_found = EndpointRestriction.filter_one_system(
|
|
# EndpointRestriction.operation_uu_id == operation_id, db=db_session,
|
|
# ).data
|
|
# if endpoint_restriction_found:
|
|
# endpoint_restriction_found.update(**add_or_update_dict, db=db_session)
|
|
# endpoint_restriction_found.save(db=db_session)
|
|
# else:
|
|
# restriction = EndpointRestriction.find_or_create(**add_or_update_dict, db=db_session)
|
|
# if restriction.meta_data.created:
|
|
# restriction.save(db=db_session)
|
|
|
|
def register_routes(self):
|
|
for router in self.router_list:
|
|
self.app.include_router(router)
|
|
self.add_router_with_event_to_database(router)
|
|
return self.app
|