fdsafafdaf

src/main.py

@@ -15,6 +15,7 @@ logging.basicConfig(level=logging.INFO)
 TEMP_FOLDER = "music-downloader"
 DATABASE_FILE = "metadata.db"
 DATABASE_STRUCTURE_FILE = "database_structure.sql"
+DATABASE_STRUCTURE_FALLBACK = "https://raw.githubusercontent.com/HeIIow2/music-downloader/new_metadata/assets/database_structure.sql"
 
 DATABASE_LOGGER = logging.getLogger("database")
 METADATA_DOWNLOAD_LOGGER = logging.getLogger("metadata-download")
@@ -34,7 +35,9 @@ if not os.path.exists(temp_dir):
     os.mkdir(temp_dir)
 
 database = Database(os.path.join(temp_dir, DATABASE_FILE),
-                    os.path.join(temp_dir, DATABASE_STRUCTURE_FILE), DATABASE_LOGGER,
+                    os.path.join(temp_dir, DATABASE_STRUCTURE_FILE),
+                    DATABASE_STRUCTURE_FALLBACK, 
+                    DATABASE_LOGGER,
                     reset_anyways=True)
 
 

src/metadata/database.py

@@ -2,10 +2,11 @@ import sqlite3
 import os
 import logging
 import json
+import requests
 
 
 class Database:
-    def __init__(self, path_to_db: str, db_structure: str, logger: logging.Logger, reset_anyways: bool = False):
+    def __init__(self, path_to_db: str, db_structure: str, db_structure_fallback: str, logger: logging.Logger, reset_anyways: bool = False):
         self.logger = logger
         self.path_to_db = path_to_db
 
@@ -13,9 +14,9 @@ class Database:
         self.cursor = self.connection.cursor()
 
         # init database
-        self.init_db(database_structure=db_structure, reset_anyways=reset_anyways)
+        self.init_db(database_structure=db_structure, database_structure_fallback=db_structure_fallback, reset_anyways=reset_anyways)
 
-    def init_db(self, database_structure: str, reset_anyways: bool = False):
+    def init_db(self, database_structure: str, database_structure_fallback: str, reset_anyways: bool = False):
         # check if db exists
         exists = True
         try:
@@ -32,6 +33,13 @@ class Database:
             # reset the database if reset_anyways is true or if an error has been thrown previously.
             self.logger.info("Creating/Reseting Database.")
 
+            if not os.path.exists(database_structure):
+                self.logger.info("database structure file doesn't exist yet, fetching from github")
+                r = requests.get(database_structure_fallback)
+                
+                with open(database_structure, "w") as f:
+                    f.write(r.text)
+
             # read the file
             with open(database_structure, "r") as database_structure_file:
                 query = database_structure_file.read()