github/nzbToMedia
1
0
Fork 0
mirror of https://github.com/clinton-hall/nzbToMedia.git synced 2025-07-30 03:28:26 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix migrate issues.

Browse source
This commit is contained in:
clinton-hall 2014-05-28 12:41:28 +09:30
parent 8fdaa6a674
commit a00c52fd69
1 changed files with 9 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

11
nzbtomedia/databases/mainDB.py
View file

@ -19,7 +19,11 @@ def backupDatabase(version):
class InitialSchema(nzbToMediaDB.SchemaUpgrade): class InitialSchema(nzbToMediaDB.SchemaUpgrade):
def test(self): def test(self):
return self.hasTable("db_version") no_update = True
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): def execute(self):
if not self.hasTable("downloads") and not self.hasTable("db_version"): if not self.hasTable("downloads") and not self.hasTable("db_version"):
@ -49,7 +53,10 @@ class InitialSchema(nzbToMediaDB.SchemaUpgrade):
) )
if cur_db_version < MAX_DB_VERSION: # We need to upgrade. if cur_db_version < MAX_DB_VERSION: # We need to upgrade.
queries = [ queries = [
"ALTER TABLE downloads ADD CONSTRAINT pk_downloadID PRIMARY KEY (input_directory, input_name);", "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);" "INSERT INTO db_version (db_version) VALUES (2);"
] ]
for query in queries: for query in queries: