From 45137b75b60f104f8414817d9167ad5946037127 Mon Sep 17 00:00:00 2001 From: suchmememanyskill <38142618+suchmememanyskill@users.noreply.github.com> Date: Sun, 7 Apr 2024 23:27:21 +0200 Subject: [PATCH] Add backend fetch_class_mappings function --- main.py | 17 +++++++++++++++-- 1 file changed, 15 insertions(+), 2 deletions(-) diff --git a/main.py b/main.py index 72bd0fc..f09f58a 100644 --- a/main.py +++ b/main.py @@ -180,6 +180,20 @@ async def get_last_load_errors(self): async def upload_theme(self, name : str, base_url : str, bearer_token : str) -> dict: return (await self.loader.upload_theme(name, base_url, bearer_token)).to_dict() + async def fetch_class_mappings(self): + await self._fetch_class_mappings(self) + return Result(True).to_dict() + + async def _fetch_class_mappings(self, run_in_bg : bool = False): + global SUCCESSFUL_FETCH_THIS_RUN + + SUCCESSFUL_FETCH_THIS_RUN = False + css_translations_path = os.path.join(get_theme_path(), "css_translations.json") + if run_in_bg: + asyncio.get_event_loop().create_task(every(60, fetch_class_mappings, css_translations_path, self.loader)) + else: + await fetch_class_mappings(css_translations_path, self.loader) + async def _main(self): global Initialized if Initialized: @@ -208,8 +222,7 @@ async def _main(self): if (ALWAYS_RUN_SERVER or store_or_file_config("server")): await self.enable_server(self) - css_translations_path = os.path.join(get_theme_path(), "css_translations.json") - asyncio.get_event_loop().create_task(every(60, fetch_class_mappings, css_translations_path, self.loader)) + await self._fetch_class_mappings(self, True) await initialize() if __name__ == '__main__':