cnc-service/src/models.py

#!/usr/bin/env python3
import enum
from sqlalchemy.dialects.postgresql import UUID
from sqlalchemy.sql import func
from db import db

"""
SQLAlchemy models
"""

__author__ = '@tormakris'
__copyright__ = "Copyright 2020, Birbnetes Team"
__module_name__ = "models"
__version__text__ = "1"


class DeviceStatusEnum(enum.Enum):
    error = "error"
    online = "online"
    offline = "offline"


class SensorStatusEnum(enum.Enum):
    unknown = "unknown"
    online = "online"
    offline = "offline"


class Device(db.Model):
    __tablename__ = 'device'
    id = db.Column(db.Integer, primary_key=True)
    sensorid = db.Column(UUID(as_uuid=True), nullable=True)
    status = db.Column(db.Enum(DeviceStatusEnum), nullable=False)
    url = db.Column(db.String, nullable=False)
    lastupdate = db.Column(db.TIMESTAMP, nullable=False, server_default=func.now(), onupdate=func.current_timestamp())
    sensors = db.relationship("Sensor")


class Sensor(db.Model):
    __tablename__ = 'sensor'
    id = db.Column(db.UUID, primary_key=True)
    status = db.Column(db.Enum(SensorStatusEnum), nullable=False)
    device_id = db.Column(db.UUID, db.ForeignKey('device.id'))