From 5a1512ac501bc16a0c5b3bd6e0ab78c34fe2db41 Mon Sep 17 00:00:00 2001
From: Facenapalm <Antik125@ya.ru>
Date: Thu, 6 Jun 2024 12:26:42 +0300
Subject: [PATCH] Cleanup

---
 seek_lutris_id.py      | 41 +++++++++++++++++++++--------------------
 seek_lutris_id_fast.py |  7 +++----
 seek_playground_id.py  |  2 +-
 3 files changed, 25 insertions(+), 25 deletions(-)

diff --git a/seek_lutris_id.py b/seek_lutris_id.py
index f054733..733941a 100644
--- a/seek_lutris_id.py
+++ b/seek_lutris_id.py
@@ -34,7 +34,7 @@
 from urllib.parse import unquote
 from common.seek_basis import SearchIDSeekerBot
 
-class LutrisSeekerBot(SearchIDSeekerBot):
+class LutrisBot():
     ids_data = {
         "igdb": {
             "property": "P5794",
@@ -64,25 +64,6 @@ class LutrisSeekerBot(SearchIDSeekerBot):
         # TODO: GOG DB (for example: https://lutris.net/games/the-chaos-engine/ ) ?
     }
 
-    def __init__(self):
-        super().__init__(
-            database_property="P7597",
-            default_matching_property="P1733",
-            allowed_matching_properties=[entry["property"] for entry in self.ids_data.values()],
-        )
-
-    def search(self, query, max_results=None):
-        params = {
-            "q": query,
-            "unpublished-filter": "on"
-        }
-        response = requests.get("https://lutris.net/games", params=params, headers=self.headers)
-        time.sleep(1)
-        if not response:
-            raise RuntimeError(f"can't get search results for query `{query}`. Status code: {response.status_code}")
-
-        return re.findall(r"<div class=[\"']game-preview[\"']>\s+<a href=[\"']/games/([^\"']+)/\"", response.text)
-
     def parse_entry(self, entry_id):
         response = requests.get(f"https://lutris.net/games/{entry_id}", headers=self.headers)
         time.sleep(1)
@@ -105,5 +86,25 @@ def parse_entry(self, entry_id):
                     print(f"WARNING: {data['property']} found, but `{href}` doesn't match a mask")
         return result
 
+class LutrisSeekerBot(LutrisBot, SearchIDSeekerBot):
+    def __init__(self):
+        super().__init__(
+            database_property="P7597",
+            default_matching_property="P1733",
+            allowed_matching_properties=[entry["property"] for entry in self.ids_data.values()],
+        )
+
+    def search(self, query, max_results=None):
+        params = {
+            "q": query,
+            "unpublished-filter": "on"
+        }
+        response = requests.get("https://lutris.net/games", params=params, headers=self.headers)
+        time.sleep(1)
+        if not response:
+            raise RuntimeError(f"can't get search results for query `{query}`. Status code: {response.status_code}")
+
+        return re.findall(r"<div class=[\"']game-preview[\"']>\s+<a href=[\"']/games/([^\"']+)/\"", response.text)
+
 if __name__ == "__main__":
     LutrisSeekerBot().run()
diff --git a/seek_lutris_id_fast.py b/seek_lutris_id_fast.py
index 8588f94..aade407 100644
--- a/seek_lutris_id_fast.py
+++ b/seek_lutris_id_fast.py
@@ -32,18 +32,17 @@
 """
 
 from common.seek_basis import DirectIDSeekerBot
-from seek_lutris_id import LutrisSeekerBot
+from seek_lutris_id import LutrisBot
 
-class LutrisSeekerBotLite(DirectIDSeekerBot):
+class LutrisSeekerBotLite(LutrisBot, DirectIDSeekerBot):
     def __init__(self):
         super().__init__(
             database_property='P7597',
             default_matching_property='P5794',
         )
-        self.seeker = LutrisSeekerBot()
 
     def seek_database_entry(self):
-        parsed_entry = self.seeker.parse_entry(self.matching_value)
+        parsed_entry = self.parse_entry(self.matching_value)
         if 'P5794' not in parsed_entry:
             raise RuntimeError(f'Lutris entry `{self.matching_value}` have no backlink to IGDB')
         if parsed_entry['P5794'] != self.matching_value:
diff --git a/seek_playground_id.py b/seek_playground_id.py
index 601a854..e3e5e66 100644
--- a/seek_playground_id.py
+++ b/seek_playground_id.py
@@ -56,7 +56,7 @@ def __init__(self):
         super().__init__(
             database_property='P10354',
             default_matching_property='P1733',
-            allowed_matching_properties=['P1733', 'P6278'],
+            allowed_matching_properties=[entry["property"] for entry in self.stores_data.values() if entry],
         )
 
     def search(self, query, max_results=None):