cli/db: handle alarm states

This commit is contained in:
Iain Learmonth 2022-04-22 14:45:47 +01:00
parent 6f9cd1eaaa
commit 120be8eaa5

View file

@ -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()