| 84 | |
| 85 | |
| 86 | def upgrade(saveddata_engine): |
| 87 | # Update fits schema to include target resists attribute |
| 88 | try: |
| 89 | saveddata_engine.execute("SELECT targetResistsID FROM fits LIMIT 1") |
| 90 | except sqlalchemy.exc.DatabaseError: |
| 91 | saveddata_engine.execute("ALTER TABLE fits ADD COLUMN targetResistsID INTEGER;") |
| 92 | |
| 93 | # Convert modules |
| 94 | for replacement_item, list in CONVERSIONS.items(): |
| 95 | for retired_item in list: |
| 96 | saveddata_engine.execute('UPDATE "modules" SET "itemID" = ? WHERE "itemID" = ?', |
| 97 | (replacement_item, retired_item)) |
| 98 | saveddata_engine.execute('UPDATE "cargo" SET "itemID" = ? WHERE "itemID" = ?', |
| 99 | (replacement_item, retired_item)) |