mirror of
https://github.com/clinton-hall/nzbToMedia.git
synced 2025-07-14 17:22:53 -07:00
parent
617bf37878
commit
c80a678ded
54 changed files with 567 additions and 567 deletions
63
core/databases/mainDB.py
Normal file
63
core/databases/mainDB.py
Normal file
|
@ -0,0 +1,63 @@
|
|||
import core
|
||||
from core import logger, nzbToMediaDB
|
||||
from core.nzbToMediaUtil import backupVersionedFile
|
||||
|
||||
MIN_DB_VERSION = 1 # oldest db version we support migrating from
|
||||
MAX_DB_VERSION = 2
|
||||
|
||||
def backupDatabase(version):
|
||||
logger.info("Backing up database before upgrade")
|
||||
if not backupVersionedFile(nzbToMediaDB.dbFilename(), version):
|
||||
logger.log_error_and_exit("Database backup failed, abort upgrading database")
|
||||
else:
|
||||
logger.info("Proceeding with upgrade")
|
||||
|
||||
# ======================
|
||||
# = Main DB Migrations =
|
||||
# ======================
|
||||
# Add new migrations at the bottom of the list; subclass the previous migration.
|
||||
|
||||
class InitialSchema(nzbToMediaDB.SchemaUpgrade):
|
||||
def test(self):
|
||||
no_update = False
|
||||
if self.hasTable("db_version"):
|
||||
cur_db_version = self.checkDBVersion()
|
||||
no_update = not cur_db_version < MAX_DB_VERSION
|
||||
return no_update
|
||||
|
||||
def execute(self):
|
||||
if not self.hasTable("downloads") and not self.hasTable("db_version"):
|
||||
queries = [
|
||||
"CREATE TABLE db_version (db_version INTEGER);",
|
||||
"CREATE TABLE downloads (input_directory TEXT, input_name TEXT, input_hash TEXT, input_id TEXT, client_agent TEXT, status INTEGER, last_update NUMERIC, CONSTRAINT pk_downloadID PRIMARY KEY (input_directory, input_name));",
|
||||
"INSERT INTO db_version (db_version) VALUES (2);"
|
||||
]
|
||||
for query in queries:
|
||||
self.connection.action(query)
|
||||
|
||||
else:
|
||||
cur_db_version = self.checkDBVersion()
|
||||
|
||||
if cur_db_version < MIN_DB_VERSION:
|
||||
logger.log_error_and_exit("Your database version (" + str(
|
||||
cur_db_version) + ") is too old to migrate from what this version of nzbToMedia supports (" + \
|
||||
str(MIN_DB_VERSION) + ").\n" + \
|
||||
"Please remove nzbtomedia.db file to begin fresh."
|
||||
)
|
||||
|
||||
if cur_db_version > MAX_DB_VERSION:
|
||||
logger.log_error_and_exit("Your database version (" + str(
|
||||
cur_db_version) + ") has been incremented past what this version of nzbToMedia supports (" + \
|
||||
str(MAX_DB_VERSION) + ").\n" + \
|
||||
"If you have used other forks of nzbToMedia, your database may be unusable due to their modifications."
|
||||
)
|
||||
if cur_db_version < MAX_DB_VERSION: # We need to upgrade.
|
||||
queries = [
|
||||
"CREATE TABLE downloads2 (input_directory TEXT, input_name TEXT, input_hash TEXT, input_id TEXT, client_agent TEXT, status INTEGER, last_update NUMERIC, CONSTRAINT pk_downloadID PRIMARY KEY (input_directory, input_name));",
|
||||
"INSERT INTO downloads2 SELECT * FROM downloads;",
|
||||
"DROP TABLE IF EXISTS downloads;",
|
||||
"ALTER TABLE downloads2 RENAME TO downloads;",
|
||||
"INSERT INTO db_version (db_version) VALUES (2);"
|
||||
]
|
||||
for query in queries:
|
||||
self.connection.action(query)
|
Loading…
Add table
Add a link
Reference in a new issue