Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
37 changes: 21 additions & 16 deletions picoapi/api.py
Original file line number Diff line number Diff line change
Expand Up @@ -57,32 +57,37 @@ def __init__(
**kwargs
) -> None:

# call super class __init__
super().__init__(*args, **kwargs)

# add the cors middleware
self.add_middleware(
CORSMiddleware,
allow_origins=allow_origins,
allow_credentials=allow_credentials,
allow_methods=allow_methods,
allow_headers=allow_headers,
)

self.is_master = is_master
self.services = []
self.healthchecks = {}
self.add_api_route(api_health_path, healthcheck)

if self.is_master:
# call super class __init__
super().__init__(*args, **kwargs)

# add service registration
self.add_api_route("/register", self.add_service)
self.add_api_route("/register", self.add_service, methods=["PUT"])
self.add_api_route("/services/status", self.get_services_status)
self.add_api_route("/services/definition", self.get_services_status)
self.add_api_route("/services/definition", self.get_services_openapi)

else:
# add the service registration event
self.router.on_event("startup", register_uservice)
kwargs["on_startup"] = [register_uservice, *[x for x in kwargs.get("on_startup")]] if kwargs.get("on_startup") else [register_uservice]

# call super class __init__
super().__init__(*args, **kwargs)

# add the cors middleware
self.add_middleware(
CORSMiddleware,
allow_origins=allow_origins,
allow_credentials=allow_credentials,
allow_methods=allow_methods,
allow_headers=allow_headers,
)

# add healthcheck route
self.add_api_route(api_health_path, healthcheck)

async def get_services_status(self):
return JSONResponse(
Expand Down