interface: switch to asyncio TaskGroup

Signed-off-by: Álvaro Fernández Rojas <[email protected]>
Noltari · Jul 23, 2024 · af42881 · af42881
1 parent d7510d4
commit af42881
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/aemet_opendata/interface.py b/aemet_opendata/interface.py
@@ -417,12 +417,10 @@ async def update_station(self) -> None:
 
     async def update(self) -> None:
         """Update all AEMET OpenData data."""
-        tasks = [
-            self.update_daily(),
-            self.update_hourly(),
-            self.update_station(),
-        ]
-        await asyncio.gather(*tasks)
+        async with asyncio.TaskGroup() as tg:
+            tg.create_task(self.update_daily())
+            tg.create_task(self.update_hourly())
+            tg.create_task(self.update_station())
 
     def weather(self) -> dict[str, Any] | None:
         """Update AEMET OpenData town daily forecast."""