From 0baca19e02fc48ceb4d6a1102c837dfd3648635d Mon Sep 17 00:00:00 2001
From: Lars Noack <lars.noack@outlook.de>
Date: Fri, 28 Oct 2022 12:35:26 +0200
Subject: [PATCH] added couple functions to fetch data from database

---
 src/metadata/database.py            | 37 ++++++++++++++++++++++-------
 src/metadata/database_structure.sql |  7 ++++--
 2 files changed, 34 insertions(+), 10 deletions(-)

diff --git a/src/metadata/database.py b/src/metadata/database.py
index e3b03b5..c4ab77c 100644
--- a/src/metadata/database.py
+++ b/src/metadata/database.py
@@ -13,7 +13,8 @@ def get_temp_dir():
     return temp_dir
 
 
-DATABASE_STRUCTURE_FILE = "database_structure.sql"
+# DATABASE_STRUCTURE_FILE = "database_structure.sql"
+DATABASE_STRUCTURE_FILE = "src/metadata/database_structure.sql"
 TEMP_DIR = get_temp_dir()
 DATABASE_FILE = "metadata.db"
 db_path = os.path.join(TEMP_DIR, DATABASE_FILE)
@@ -124,9 +125,14 @@ def add_track(
     cursor.execute(query, values)
     connection.commit()
 
+def get_custom_track_querry(custom_where: list) -> str:
+    where_args = [
+        "track.release_id == release_.id",
+        "release_group.id == release_.release_group_id"
+    ]
+    where_args.extend(custom_where)
 
-def get_track_metadata(musicbrainz_releasetrackid: str):
-    # this would be vulnerable if musicbrainz_releasetrackid would be user input
+    where_arg = " AND ".join(where_args)
     query = f"""
 SELECT DISTINCT 
     track.id as musicbrainz_releasetrackid,
@@ -149,21 +155,36 @@ SELECT DISTINCT
     release_group.album_artist_id as album_artist_id
 FROM track, release_, release_group, artist
 WHERE
-    track.id == "{musicbrainz_releasetrackid}" AND
-    track.release_id == release_.id AND
-    release_group.id == release_.release_group_id;
+    {where_arg};
     """
+    return query
 
-    resulting_tracks = list(cursor.execute(query))
+def get_custom_track(custom_where: list):
+    query = get_custom_track_querry(custom_where=custom_where)
+    return [dict(i) for i in cursor.execute(query)]
+
+def get_track_metadata(musicbrainz_releasetrackid: str):
+    # this would be vulnerable if musicbrainz_releasetrackid would be user input
+    resulting_tracks = get_custom_track([f'track.id == "{musicbrainz_releasetrackid}"'])
     if len(resulting_tracks) != 1:
         return -1
 
     return dict(resulting_tracks[0])
 
+def get_tracks_to_download():
+    return get_custom_track(["track.downloaded == 0"])
+
+def get_tracks_without_isrc():
+    return get_custom_track(["track.isrc IS NULL"])
 
 init_db(cursor=cursor, connection=connection, reset_anyways=False)
 
 if __name__ == "__main__":
     logging.basicConfig(level=logging.DEBUG)
 
-    print(get_track_metadata("a85d5ed5-20e5-4f95-8034-d204d81a36dd"))
+    # get_track(["track.downloaded == 0", "track.isrc IS NOT NULL"])
+    #
+    for track in get_tracks_without_isrc():
+        print(track['track'])
+
+    #print(get_track_metadata("a85d5ed5-20e5-4f95-8034-d204d81a36dd"))
diff --git a/src/metadata/database_structure.sql b/src/metadata/database_structure.sql
index 0d86ee2..9f7fc29 100644
--- a/src/metadata/database_structure.sql
+++ b/src/metadata/database_structure.sql
@@ -43,8 +43,11 @@ CREATE TABLE release_ (
 DROP TABLE IF EXISTS track;
 CREATE TABLE track (
     id TEXT PRIMARY KEY NOT NULL,
+    downloaded BOOLEAN NOT NULL DEFAULT 0,
     release_id TEXT NOT NULL,
     track TEXT,
-    isrc TEXT
+    isrc TEXT,
+    filepath TEXT,
+    url TEXT,
+    src TEXT
 );
-