ogn-python/app/model/device.py

62 wiersze
2.1 KiB
Python

import datetime
from sqlalchemy.ext.hybrid import hybrid_property
from app import db
from .device_info import DeviceInfo
class Device(db.Model):
__tablename__ = "devices"
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String, index=True)
# address = db.Column(db.String(6), index=True)
address = db.Column(db.String, index=True)
firstseen = db.Column(db.DateTime, index=True)
lastseen = db.Column(db.DateTime, index=True)
aircraft_type = db.Column(db.SmallInteger, index=True)
stealth = db.Column(db.Boolean)
software_version = db.Column(db.Float(precision=2))
hardware_version = db.Column(db.SmallInteger)
real_address = db.Column(db.String(6))
def __repr__(self):
return "<Device: %s,%s,%s,%s,%s,%s>" % (self.address, self.aircraft_type, self.stealth, self.software_version, self.hardware_version, self.real_address)
@hybrid_property
def info(self):
query = db.session.query(DeviceInfo).filter(DeviceInfo.address == self.address).order_by(DeviceInfo.address_origin)
return query.first()
def get_infos(self):
query = db.session.query(DeviceInfo).filter(DeviceInfo.address == self.address).order_by(DeviceInfo.address_origin)
return [info for info in query.all()]
EXPIRY_DATES = {
6.67: datetime.date(2020, 10, 31),
6.63: datetime.date(2020, 5, 31),
6.62: datetime.date(2020, 5, 31),
6.6: datetime.date(2020, 1, 31),
6.42: datetime.date(2019, 10, 31),
6.41: datetime.date(2019, 1, 31),
6.4: datetime.date(2019, 1, 31),
6.09: datetime.date(2018, 9, 30),
6.08: datetime.date(2018, 9, 30),
6.07: datetime.date(2018, 3, 31),
6.06: datetime.date(2017, 9, 30),
6.05: datetime.date(2017, 3, 31),
}
def expiry_date(self):
if self.name.startswith("FLR"):
if self.software_version in self.EXPIRY_DATES:
return self.EXPIRY_DATES[self.software_version]
else:
return datetime.date(2000, 1, 1)
else:
return None