join tested auth service login/select completed

api_services/api_initializer/create_route.py

@@ -12,44 +12,27 @@ class RouteRegisterController:
     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:
+            EndpointRestriction.set_session(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")
+                if not getattr(route, "methods") and isinstance(getattr(route, "methods")):
+                    raise ValueError(f"Route {route_path} methods 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)
+                route_method = [method.lower() for method in getattr(route, "methods")][0]
+                add_or_update_dict = dict(
+                    endpoint_method=route_method, endpoint_name=route_path, endpoint_desc=route_summary.replace("_", " "), endpoint_function=route_summary, is_confirmed=True
+                )
+                if to_save_endpoint := EndpointRestriction.query.filter(EndpointRestriction.operation_uu_id == operation_id).first():
+                    to_save_endpoint.update(**add_or_update_dict)
+                    to_save_endpoint.save()
+                else:
+                    created_endpoint = EndpointRestriction.create(**add_or_update_dict, operation_uu_id=operation_id)
+                    created_endpoint.save()
 
     def register_routes(self):
         for router in self.router_list: