from sqlalchemy import Column, Integer, String, Unicode, Boolean, SmallInteger
from sqlalchemy.orm import relationship

from .base import Base


class Device(Base):
    __tablename__ = 'device'

    id = Column(Integer, primary_key=True)
    address_type = None
    address = Column(String(6), index=True)
    name = Column(Unicode)
    airport = Column(String)
    aircraft = Column(String)
    registration = Column(String(7), index=True)
    competition = Column(String(3))
    frequency = Column(String)
    tracked = Column(Boolean)
    identified = Column(Boolean)
    aircraft_type = Column(SmallInteger, index=True)

    address_origin = Column(SmallInteger)

    # Relations
    aircraft_beacons = relationship('AircraftBeacon')

    def __repr__(self):
        return "<Device: %s,%s,%s,%s,%s,%s,%s,%s,%s,%s>" % (
            self.address_type,
            self.address,
            self.name,
            self.airport,
            self.aircraft,
            self.registration,
            self.competition,
            self.frequency,
            self.tracked,
            self.identified)