lots of typing fixes
This commit is contained in:
parent
51f580a304
commit
3665c34961
43 changed files with 260 additions and 178 deletions
|
@ -1,5 +1,6 @@
|
|||
import enum
|
||||
from datetime import datetime
|
||||
from typing import List, Any
|
||||
|
||||
from app import db
|
||||
|
||||
|
@ -11,7 +12,7 @@ class AlarmState(enum.Enum):
|
|||
CRITICAL = 3
|
||||
|
||||
|
||||
class Alarm(db.Model):
|
||||
class Alarm(db.Model): # type: ignore
|
||||
id = db.Column(db.Integer, primary_key=True)
|
||||
target = db.Column(db.String(60), nullable=False)
|
||||
group_id = db.Column(db.Integer, db.ForeignKey("group.id"))
|
||||
|
@ -30,18 +31,18 @@ class Alarm(db.Model):
|
|||
bridge = db.relationship("Bridge", back_populates="alarms")
|
||||
|
||||
@classmethod
|
||||
def csv_header(cls):
|
||||
def csv_header(cls) -> List[str]:
|
||||
return [
|
||||
"id", "target", "group_id", "origin_id", "proxy_id", "bridge_id", "alarm_type",
|
||||
"alarm_state", "state_changed", "last_updated", "text"
|
||||
]
|
||||
|
||||
def csv_row(self):
|
||||
def csv_row(self) -> List[Any]:
|
||||
return [
|
||||
getattr(self, x) for x in self.csv_header()
|
||||
]
|
||||
|
||||
def update_state(self, state: AlarmState, text: str):
|
||||
def update_state(self, state: AlarmState, text: str) -> None:
|
||||
if self.alarm_state != state or self.state_changed is None:
|
||||
self.state_changed = datetime.utcnow()
|
||||
self.alarm_state = state
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue