ogn-python/app/model/device.py

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