Basic SteamGridDB support

src/utils/steamgriddb.py

@@ -0,0 +1,69 @@
+from pathlib import Path
+
+import requests
+from gi.repository import Gio
+from steamgrid import SteamGridDB, http
+
+from .save_cover import save_cover
+
+
+class SGDBSave:
+    def __init__(self, importer, games):
+        self.importer = importer
+        self.sgdb = SteamGridDB(importer.parent_widget.schema.get_string("sgdb-key"))
+
+        # Wrap the function in another one as Gio.Task.run_in_thread does not allow for passing args
+        def create_func(game):
+            def wrapper(task, *_unused):
+                self.update_cover(
+                    task,
+                    game,
+                )
+
+            return wrapper
+
+        for game in games:
+            Gio.Task.new(None, None, self.task_done).run_in_thread(create_func(game))
+
+    def update_cover(self, task, game):
+        if self.importer.parent_widget.schema.get_boolean("sgdb-prefer") or (
+            self.importer.parent_widget.schema.get_boolean("sgdb-import")
+            and self.importer.parent_widget.games[game[0]].pixbuf
+            == self.importer.parent_widget.placeholder_pixbuf
+        ):
+            try:
+                search_result = self.sgdb.search_game(game[1])
+            except requests.exceptions.RequestException:
+                task.return_value(game[0])
+                return
+            except http.HTTPException as exception:
+                self.importer.sgdb_exception = str(exception)
+                task.return_value(game[0])
+                return
+
+            try:
+                grid = self.sgdb.get_grids_by_gameid(
+                    [search_result[0].id], is_nsfw=False
+                )[0]
+            except (TypeError, IndexError):
+                task.return_value(game[0])
+                return
+
+            tmp_file = Gio.File.new_tmp(None)[0]
+
+            try:
+                response = requests.get(str(grid), timeout=5)
+            except requests.exceptions.RequestException:
+                task.return_value(game[0])
+                return
+
+            Path(tmp_file.get_path()).write_bytes(response.content)
+            save_cover(self.importer.parent_widget, game[0], tmp_file.get_path())
+
+        task.return_value(game[0])
+
+    def task_done(self, _task, result):
+        game_id = result.propagate_value()[1]
+        self.importer.parent_widget.update_games([game_id])
+        self.importer.queue -= 1
+        self.importer.done()