diff --git a/app/cli/db.py b/app/cli/db.py index e6e0a4a..cd45361 100644 --- a/app/cli/db.py +++ b/app/cli/db.py @@ -9,7 +9,7 @@ from app.extensions import db from app.models.base import Group, MirrorList from app.models.bridges import BridgeConf from app.models.mirrors import Mirror, Origin, Proxy -from app.models.alarms import Alarm +from app.models.alarms import Alarm, AlarmState models = { "bridge": BridgeConf, @@ -54,6 +54,9 @@ def impot(model: db.Model): elif header[i].endswith("_id") and line[i] == "": # integer foreign keys line[i] = None + elif header[i] in ["alarm_state"]: + # alarm states + line[i] = getattr(AlarmState, line[i][len("AlarmState."):]) setattr(x, header[i], line[i]) db.session.add(x) db.session.commit()