Update return signatures

Author: Garulf

plugin/library.py

@@ -12,6 +12,14 @@
 
 log = logging.getLogger(__name__)
 
+ASSETS = [
+    'header',
+    'library_600x900',
+    'library_hero',
+    'library_hero_blur',
+    'logo'
+]
+
 
 @dataclass
 class Library:
@@ -94,6 +102,12 @@ def __init__(self, name: str, path: Union[str, Path], image_dir: LibraryImageDir
         self._id = id
         self._image_id = self.id
 
+    def _get_hashed_icon(self) -> Optional[Path]:
+        asset_path = Path(self.image_dir).joinpath(self.id)
+        for asset in asset_path.iterdir():
+            if asset.stem not in ASSETS:
+                return asset
+
     @property
     def id(self):
         """
@@ -124,11 +138,11 @@ def get_image(self, type: str, sep='_') -> Path:
         return self.image_dir.get_image(id, type, sep)
 
     @cached_property
-    def icon(self) -> Path:
+    def icon(self) -> Optional[Path]:
         """
         Return the icon for this library item.
         """
-        return self.get_image('icon') or Path(self.unquoted_path())
+        return self._get_hashed_icon()
 
     @cached_property
     def hero(self) -> Path:

plugin/steam.py

@@ -1,7 +1,7 @@
 from pathlib import Path
 import logging
-import winreg as reg
-from winreg import HKEY_CURRENT_USER, HKEY_LOCAL_MACHINE
+#import winreg as reg
+#from winreg import HKEY_CURRENT_USER, HKEY_LOCAL_MACHINE
 from typing import Union
 
 from .vdfs import VDF