🚧 More work on Steam source

src/utils/steam.py

@@ -0,0 +1,78 @@
+import re
+import logging
+from typing import TypedDict
+
+import requests
+from requests import HTTPError, JSONDecodeError
+
+
+class SteamError(Exception):
+    pass
+
+
+class SteamGameNotFoundError(SteamError):
+    pass
+
+
+class SteamNotAGameError(SteamError):
+    pass
+
+
+class SteamInvalidManifestError(SteamError):
+    pass
+
+
+class SteamManifestData(TypedDict):
+    name: str
+    appid: str
+    StateFlags: str
+
+
+class SteamAPIData(TypedDict):
+    developers: str
+
+
+class SteamHelper:
+    """Helper around the Steam API"""
+
+    base_url = "https://store.steampowered.com/api"
+
+    def get_manifest_data(self, manifest_path) -> SteamManifestData:
+        """Get local data for a game from its manifest"""
+
+        with open(manifest_path) as file:
+            contents = file.read()
+
+        data = {}
+
+        for key in SteamManifestData.__required_keys__:
+            regex = f'"{key}"\s+"(.*)"\n'
+            if (match := re.search(regex, contents)) is None:
+                raise SteamInvalidManifestError()
+            data[key] = match.group(1)
+
+        return SteamManifestData(**data)
+
+    def get_api_data(self, appid) -> SteamAPIData:
+        """Get online data for a game from its appid"""
+
+        try:
+            with requests.get(
+                f"{self.base_url}/appdetails?appids={appid}", timeout=5
+            ) as response:
+                response.raise_for_status()
+                data = response.json()[appid]
+
+        except (HTTPError, JSONDecodeError) as error:
+            logging.warning("Error while querying Steam API for %s", appid)
+            raise error
+
+        if not data["success"]:
+            raise SteamGameNotFoundError()
+
+        if data["data"]["type"] != "game":
+            raise SteamNotAGameError()
+
+        # Return API values we're interested in
+        values = SteamAPIData(developers=", ".join(data["data"]["developers"]))
+        return values