jasima/majuna
3
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

cli/db: handle alarm states

Browse source
This commit is contained in:
Iain Learmonth 2022-04-22 14:45:47 +01:00
parent 6f9cd1eaaa
commit 120be8eaa5
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
app/cli/db.py
View file

@ -9,7 +9,7 @@ from app.extensions import db
from app.models.base import Group, MirrorList from app.models.base import Group, MirrorList
from app.models.bridges import BridgeConf from app.models.bridges import BridgeConf
from app.models.mirrors import Mirror, Origin, Proxy from app.models.mirrors import Mirror, Origin, Proxy
from app.models.alarms import Alarm from app.models.alarms import Alarm, AlarmState
models = { models = {
"bridge": BridgeConf, "bridge": BridgeConf,
@ -54,6 +54,9 @@ def impot(model: db.Model):
elif header[i].endswith("_id") and line[i] == "": elif header[i].endswith("_id") and line[i] == "":
# integer foreign keys # integer foreign keys
line[i] = None 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]) setattr(x, header[i], line[i])
db.session.add(x) db.session.add(x)
db.session.commit() db.session.commit()