From a6033664275d852a53be6272bb973f5323ceeddb Mon Sep 17 00:00:00 2001 From: codey Date: Tue, 13 Aug 2024 10:36:53 -0400 Subject: [PATCH] cleaner --- endpoints/lyric_search.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/endpoints/lyric_search.py b/endpoints/lyric_search.py index 2f18f5d..c1c0616 100644 --- a/endpoints/lyric_search.py +++ b/endpoints/lyric_search.py @@ -45,6 +45,12 @@ class LyricSearch(FastAPI): self.endpoint_name = "lyric_search" self.endpoint2_name = "lyric_cache_list" + + self.endpoints = { + "lyric_search": self.lyric_search_handler, + "lyric_cache_list": self.lyric_cache_list_handler, + } + self.acceptable_request_sources = [ "WEB", "IRC-MS", @@ -55,8 +61,8 @@ class LyricSearch(FastAPI): "LIMNORIA-SHARED" ] - app.add_api_route(f"/{self.endpoint_name}/",self.lyric_search_handler, methods=["POST"]) - app.add_api_route(f"/{self.endpoint2_name}/",self.lyric_cache_list_handler, methods=["POST"]) # pylint: disable=trailing-whitespace + for endpoint, handler in self.endpoints.items(): + app.add_api_route(f"/{endpoint}/", handler, methods=["POST"]) async def lyric_cache_list_handler(self): """