Added database model

1 files changed, 81 insertions, 0 deletions

diff --git a/mqtt_client/models.py b/mqtt_client/models.py
new file mode 100644
index 0000000..d6bd2ba
--- /dev/null
+++ b/mqtt_client/models.py
@@ -0,0 +1,81 @@
+from sqlalchemy import Column, Integer, String, Float, DateTime, ForeignKey, Boolean
+from sqlalchemy.orm import relationship
+from sqlalchemy.sql import func
+from database import Base
+
+# ----------------------------
+# Location
+# ----------------------------
+class Location(Base):
+    __tablename__ = "locations"
+
+    id = Column(Integer, primary_key=True, index=True)
+    street = Column(String(200), nullable=True, index=True)
+    avenue = Column(String(200), nullable=True, index=True)
+    zip_code = Column(String(20), nullable=True, index=True)
+
+    # Relationship: One location to many stations
+    stations = relationship("Station", back_populates="location")
+
+
+# ----------------------------
+# Station
+# ----------------------------
+class Station(Base):
+    __tablename__ = "stations"
+    
+    id = Column(Integer, primary_key=True, index=True)
+    name = Column(String(100), nullable=False, index=True)
+    location_id = Column(Integer, ForeignKey("locations.id"), nullable=True, index=True)
+    installation_date = Column(DateTime, server_default=func.now())
+    is_active = Column(Boolean, default=True)
+
+    # Relationships
+    location = relationship("Location", back_populates="stations")
+    sensor_devices = relationship("SensorDevice", back_populates="station")
+
+
+# ----------------------------
+# SensorDevice
+# ----------------------------
+class SensorDevice(Base):
+    __tablename__ = "sensor_devices"
+    
+    id = Column(Integer, primary_key=True, index=True)
+    name = Column(String(100), nullable=True, index=True)
+    installation_date = Column(DateTime, server_default=func.now())
+    station_id = Column(Integer, ForeignKey("stations.id"), nullable=True, index=True)
+    is_active = Column(Boolean, default=True)
+
+    # Relationships
+    station = relationship("Station", back_populates="sensor_devices")
+    readings = relationship("Reading", back_populates="sensor_device")
+
+
+# ----------------------------
+# Measure
+# ----------------------------
+class Measure(Base):
+    __tablename__ = "measures"
+    
+    id = Column(Integer, primary_key=True, index=True)
+    name = Column(String(50), nullable=True, index=True)
+
+    # Relationships
+    readings = relationship("Reading", back_populates="measure")
+
+
+# ----------------------------
+# Reading
+# ----------------------------
+class Reading(Base):
+    __tablename__ = "readings"
+    
+    id = Column(Integer, primary_key=True, index=True)
+    sensor_device_id = Column(Integer, ForeignKey("sensor_devices.id"), nullable=True, index=True)
+    measure_id = Column(Integer, ForeignKey("measures.id"), nullable=True, index=True)
+    value = Column(Float, nullable=True)
+
+    # Relationships
+    sensor_device = relationship("SensorDevice", back_populates="readings")
+    measure = relationship("Measure", back_populates="readings")