diff --git a/eos/const.py b/eos/const.py index 4c71d76c5..e1348c5b8 100644 --- a/eos/const.py +++ b/eos/const.py @@ -93,3 +93,11 @@ class SpoolType(IntEnum): SCALE = 0 # [0..1] TIME = 1 # Expressed via time in seconds since spool up started CYCLES = 2 # Expressed in amount of cycles since spool up started + + +@unique +class FitSystemSecurity(IntEnum): + HISEC = 0 + LOWSEC = 1 + NULLSEC = 2 + WSPACE = 3 diff --git a/eos/db/migrations/upgrade31.py b/eos/db/migrations/upgrade31.py new file mode 100644 index 000000000..448801970 --- /dev/null +++ b/eos/db/migrations/upgrade31.py @@ -0,0 +1,15 @@ +""" +Migration 31 + +- added fit system security column +""" + + +import sqlalchemy + + +def upgrade(saveddata_engine): + try: + saveddata_engine.execute("SELECT systemSecurity FROM fits LIMIT 1") + except sqlalchemy.exc.DatabaseError: + saveddata_engine.execute("ALTER TABLE fits ADD COLUMN systemSecurity INT") diff --git a/eos/db/saveddata/fit.py b/eos/db/saveddata/fit.py index efdac1c76..274c50b23 100644 --- a/eos/db/saveddata/fit.py +++ b/eos/db/saveddata/fit.py @@ -44,6 +44,7 @@ from eos.saveddata.module import Module from eos.saveddata.targetResists import TargetResists from eos.saveddata.user import User + fits_table = Table("fits", saveddata_meta, Column("ID", Integer, primary_key=True), Column("ownerID", ForeignKey("users.ID"), nullable=True, index=True), @@ -59,7 +60,8 @@ fits_table = Table("fits", saveddata_meta, Column("notes", String, nullable=True), Column("ignoreRestrictions", Boolean, default=0), Column("created", DateTime, nullable=True, default=datetime.datetime.now), - Column("modified", DateTime, nullable=True, default=datetime.datetime.now, onupdate=datetime.datetime.now) + Column("modified", DateTime, nullable=True, default=datetime.datetime.now, onupdate=datetime.datetime.now), + Column("systemSecurity", Integer, nullable=True) ) projectedFits_table = Table("projectedFits", saveddata_meta,