From e5ede625b3fc7e0b71736b42eb4f52a2f5d63378 Mon Sep 17 00:00:00 2001 From: Parker Date: Mon, 20 Jan 2025 20:44:23 -0600 Subject: Put database within data dir --- code/utils/config.py | 4 +++- code/utils/content_view.py | 2 +- code/utils/jellyfin_create.py | 2 +- code/utils/jellyfin_delete.py | 2 +- 4 files changed, 6 insertions(+), 4 deletions(-) (limited to 'code/utils') diff --git a/code/utils/config.py b/code/utils/config.py index f3fc971..9e1b687 100644 --- a/code/utils/config.py +++ b/code/utils/config.py @@ -153,7 +153,9 @@ def database_setup() -> None: """ Create the database if it does not exist """ - db = sqlite3.connect("cordarr.db") + if not os.path.exists("data"): + os.makedirs("data") + db = sqlite3.connect("data/cordarr.db") cursor = db.cursor() cursor.execute( "CREATE TABLE IF NOT EXISTS requests (title TEXT, release_year TEXT," diff --git a/code/utils/content_view.py b/code/utils/content_view.py index e8df29c..668954f 100644 --- a/code/utils/content_view.py +++ b/code/utils/content_view.py @@ -161,7 +161,7 @@ class RequestButtonView(discord.ui.View): return await interaction.response.send_message(embed=embed) # Keep track of the requests for the `/status` command - db = sqlite3.connect("cordarr.db") + db = sqlite3.connect("data/cordarr.db") cursor = db.cursor() cursor.execute( "INSERT INTO requests (title, release_year, local_id, tmdbid," diff --git a/code/utils/jellyfin_create.py b/code/utils/jellyfin_create.py index 08c5230..e860c2b 100644 --- a/code/utils/jellyfin_create.py +++ b/code/utils/jellyfin_create.py @@ -67,7 +67,7 @@ def create_jellyfin_account(user_id): return False # Add the information to the database - db = sqlite3.connect("cordarr.db") + db = sqlite3.connect("data/cordarr.db") cursor = db.cursor() cursor.execute( "INSERT INTO jellyfin_accounts (user_id, jellyfin_user_id," diff --git a/code/utils/jellyfin_delete.py b/code/utils/jellyfin_delete.py index 6164e40..66af00b 100644 --- a/code/utils/jellyfin_delete.py +++ b/code/utils/jellyfin_delete.py @@ -10,7 +10,7 @@ def delete_accounts(): Delete Jellyfin accounts that have passed their deletion time """ # Get all expired Jellyfin accounts - db = sqlite3.connect("cordarr.db") + db = sqlite3.connect("data/cordarr.db") cursor = db.cursor() cursor.execute( "SELECT jellyfin_user_id FROM jellyfin_accounts WHERE" -- cgit v1.2.3-70-g09d2