Handle both old and new DB schemas.

w9jds · May 20, 2017 · e828eb4 · e828eb4
1 parent 9737631
commit e828eb4
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/eos/db/migrations/upgrade6.py b/eos/db/migrations/upgrade6.py
@@ -6,6 +6,12 @@
 
 
 def upgrade(saveddata_engine):
-    saveddata_engine.execute('DELETE FROM damagePatterns WHERE name LIKE ? OR ID LIKE ?', ("Uniform", "1"))
-    saveddata_engine.execute('INSERT INTO damagePatterns VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)',
-                             (1, "Uniform", 25, 25, 25, 25, None, None, None))
+    try:
+        saveddata_engine.execute('DELETE FROM damagePatterns WHERE name LIKE ? OR ID LIKE ?', ("Uniform", "1"))
+        saveddata_engine.execute('INSERT INTO damagePatterns VALUES (?, ?, ?, ?, ?, ?, ?)',
+                                 (1, "Uniform", 25, 25, 25, 25, None))
+    except:
+        # Most likely using the newer DB schema (migration 22).  Go ahead and account for that.
+        saveddata_engine.execute('DELETE FROM damagePatterns WHERE name LIKE ? OR ID LIKE ?', ("Uniform", "1"))
+        saveddata_engine.execute('INSERT INTO damagePatterns VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)',
+                                 (1, "Uniform", 25, 25, 25, 25, None, None, None))