diff --git a/eos/db/migrations/upgrade1.py b/eos/db/migrations/upgrade1.py index 1d8417dfc..2eb065345 100644 --- a/eos/db/migrations/upgrade1.py +++ b/eos/db/migrations/upgrade1.py @@ -83,7 +83,21 @@ CONVERSIONS = { } def upgrade(saveddata_engine): - # Update fits schema + + # Update characters schema to include default chars (pre-1.5.0 migration) + try: + saveddata_engine.execute("SELECT defaultChar, chars FROM characters LIMIT 1") + except sqlalchemy.exc.DatabaseError: + saveddata_engine.execute("ALTER TABLE characters ADD COLUMN defaultChar INTEGER;") + saveddata_engine.execute("ALTER TABLE characters ADD COLUMN chars VARCHAR;") + + # Update fits schema to include booster attribute (pre-1.5.0 migration) + try: + saveddata_engine.execute("SELECT booster FROM fits LIMIT 1") + except sqlalchemy.exc.DatabaseError: + saveddata_engine.execute("ALTER TABLE fits ADD COLUMN booster BOOLEAN;") + + # Update fits schema to include target resists attribute try: saveddata_engine.execute("SELECT targetResistsID FROM fits LIMIT 1") except sqlalchemy.exc.DatabaseError: