Skip to content

Commit fda99fd

Browse files
author
Wojciech Kiwerski
committed
picoapi/api:
- remove on_init from __init()__ args - change line 85 - add line 86
1 parent 8ccea7b commit fda99fd

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

picoapi/api.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,6 @@ def __init__(
5353
allow_headers: List[str] = [
5454
x for x in os.getenv("API_CORS_ALLOW_HEADERS", "*").split()
5555
],
56-
on_startup = None,
5756
*args,
5857
**kwargs
5958
) -> None:
@@ -77,13 +76,15 @@ def __init__(
7776

7877
if self.is_master:
7978
# add service registration
80-
self.add_api_route("/register", self.add_service)
79+
self.add_api_route("/register", self.add_service, , methods=["PUT"])
8180
self.add_api_route("/services/status", self.get_services_status)
8281
self.add_api_route("/services/definition", self.get_services_openapi)
8382

8483
else:
8584
# add the service registration event
86-
self.on_startup = [ register_uservice() ]
85+
kwargs["on_startup"] = [register_uservice, *[x for x in kwargs.get("on_startup")]] if kwargs.get("on_startup") else [register_uservice,]
86+
super().__init__(*args, **kwargs)
87+
8788

8889
async def get_services_status(self):
8990
return JSONResponse(

0 commit comments

Comments
 (0)