87 lines
2.5 KiB
Python
87 lines
2.5 KiB
Python
from sqlite3 import connect
|
|
from pathlib import Path
|
|
from time import time
|
|
from typing import Optional, Generator
|
|
|
|
from src import shared
|
|
from src.utils.decorators import replaced_by_env_path, replaced_by_path
|
|
from src.game import Game
|
|
from src.importer.sources.source import (
|
|
Source,
|
|
SourceIterator,
|
|
LinuxSource,
|
|
WindowsSource,
|
|
)
|
|
|
|
|
|
class ItchSourceIterator(SourceIterator):
|
|
source: "ItchSource"
|
|
|
|
def generator_builder(self) -> Generator[Optional[Game], None, None]:
|
|
"""Generator method producing games"""
|
|
|
|
# Query the database
|
|
db_request = """
|
|
SELECT
|
|
games.id,
|
|
games.title,
|
|
games.cover_url,
|
|
games.still_cover_url,
|
|
caves.id
|
|
FROM
|
|
'caves'
|
|
INNER JOIN
|
|
'games'
|
|
ON
|
|
caves.game_id = games.id
|
|
;
|
|
"""
|
|
connection = connect(self.source.location / "db" / "butler.db")
|
|
cursor = connection.execute(db_request)
|
|
|
|
# Create games from the db results
|
|
for row in cursor:
|
|
values = {
|
|
"version": shared.SPEC_VERSION,
|
|
"added": int(time()),
|
|
"source": self.source.id,
|
|
"game_id": self.source.game_id_format.format(game_id=row[0]),
|
|
"executable": self.source.executable_format.format(cave_id=row[4]),
|
|
}
|
|
yield Game(values, allow_side_effects=False)
|
|
|
|
# TODO pass image URIs to the pipeline somehow
|
|
# - Add a reserved field to the Game object
|
|
# - Reconstruct those from the pipeline (we already have them)
|
|
# - Pass game and additional data to the pipeline separately (requires deep changes)
|
|
|
|
|
|
class ItchSource(Source):
|
|
name = "Itch"
|
|
location_key = "itch-location"
|
|
|
|
def __iter__(self) -> SourceIterator:
|
|
return ItchSourceIterator(self)
|
|
|
|
|
|
class ItchLinuxSource(ItchSource, LinuxSource):
|
|
variant = "linux"
|
|
executable_format = "xdg-open itch://caves/{cave_id}/launch"
|
|
|
|
@ItchSource.replaced_by_schema_key()
|
|
@replaced_by_path("~/.var/app/io.itch.itch/config/itch/")
|
|
@replaced_by_env_path("XDG_DATA_HOME", "itch/")
|
|
@replaced_by_path("~/.config/itch")
|
|
def location(self) -> Path:
|
|
raise FileNotFoundError()
|
|
|
|
|
|
class ItchWindowsSource(ItchSource, WindowsSource):
|
|
variant = "windows"
|
|
executable_format = "start itch://caves/{cave_id}/launch"
|
|
|
|
@ItchSource.replaced_by_schema_key()
|
|
@replaced_by_env_path("appdata", "itch/")
|
|
def location(self) -> Path:
|
|
raise FileNotFoundError()
|