From 6dd8e3965f8ba88fc09450cce1005caf0f2a8216 Mon Sep 17 00:00:00 2001
From: GeoffreyCoulaud <geoffrey.coulaud@gmail.com>
Date: Tue, 13 Jun 2023 10:32:07 +0200
Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20Ported=20sqlite=20fix=20from=20m?=
 =?UTF-8?q?ain?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/importer/sources/itch_source.py   | 13 ++++++++-----
 src/importer/sources/lutris_source.py |  8 +++++++-
 src/utils/sqlite.py                   | 17 +++++++++++++++++
 3 files changed, 32 insertions(+), 6 deletions(-)
 create mode 100644 src/utils/sqlite.py

diff --git a/src/importer/sources/itch_source.py b/src/importer/sources/itch_source.py
index 6d8a157..e93ad17 100644
--- a/src/importer/sources/itch_source.py
+++ b/src/importer/sources/itch_source.py
@@ -1,4 +1,5 @@
 from pathlib import Path
+from shutil import rmtree
 from sqlite3 import connect
 from time import time
 
@@ -9,10 +10,8 @@ from src.importer.sources.source import (
     SourceIterator,
     URLExecutableSource,
 )
-from src.utils.decorators import (
-    replaced_by_path,
-    replaced_by_schema_key,
-)
+from src.utils.decorators import replaced_by_path, replaced_by_schema_key
+from src.utils.sqlite import copy_db
 
 
 class ItchSourceIterator(SourceIterator):
@@ -37,7 +36,8 @@ class ItchSourceIterator(SourceIterator):
             caves.game_id = games.id
         ;
         """
-        connection = connect(self.source.location / "db" / "butler.db")
+        db_path = copy_db(self.source.location / "db" / "butler.db")
+        connection = connect(db_path)
         cursor = connection.execute(db_request)
 
         # Create games from the db results
@@ -54,6 +54,9 @@ class ItchSourceIterator(SourceIterator):
             game = Game(values, allow_side_effects=False)
             yield (game, additional_data)
 
+        # Cleanup
+        rmtree(str(db_path.parent))
+
 
 class ItchSource(URLExecutableSource):
     name = "Itch"
diff --git a/src/importer/sources/lutris_source.py b/src/importer/sources/lutris_source.py
index ddf6778..0da9b87 100644
--- a/src/importer/sources/lutris_source.py
+++ b/src/importer/sources/lutris_source.py
@@ -1,3 +1,4 @@
+from shutil import rmtree
 from sqlite3 import connect
 from time import time
 
@@ -9,6 +10,7 @@ from src.importer.sources.source import (
     URLExecutableSource,
 )
 from src.utils.decorators import replaced_by_path, replaced_by_schema_key
+from src.utils.sqlite import copy_db
 
 
 class LutrisSourceIterator(SourceIterator):
@@ -30,7 +32,8 @@ class LutrisSourceIterator(SourceIterator):
             ;
         """
         params = {"import_steam": shared.schema.get_boolean("lutris-import-steam")}
-        connection = connect(self.source.location / "pga.db")
+        db_path = copy_db(self.source.location / "pga.db")
+        connection = connect(db_path)
         cursor = connection.execute(request, params)
 
         # Create games from the DB results
@@ -56,6 +59,9 @@ class LutrisSourceIterator(SourceIterator):
             # Produce game
             yield (game, additional_data)
 
+        # Cleanup
+        rmtree(str(db_path.parent))
+
 
 class LutrisSource(URLExecutableSource):
     """Generic lutris source"""
diff --git a/src/utils/sqlite.py b/src/utils/sqlite.py
new file mode 100644
index 0000000..9661c2a
--- /dev/null
+++ b/src/utils/sqlite.py
@@ -0,0 +1,17 @@
+from glob import escape
+from pathlib import Path
+from shutil import copyfile
+
+from gi.repository import GLib
+
+
+def copy_db(original_path: Path) -> Path:
+    """
+    Copy a sqlite database to a cache dir and return its new path.
+    The caller in in charge of deleting the returned path's parent dir.
+    """
+    tmp = Path(GLib.Dir.make_tmp())
+    for file in original_path.parent.glob(f"{escape(original_path.name)}*"):
+        copy = tmp / file.name
+        copyfile(str(file), str(copy))
+    return tmp / original_path.name