from sqlalchemy import create_engine, Column, Integer, String, DateTime, Float, ForeignKey
# from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker, relationship, declarative_base
from datetime import datetime, timezone

# Create a base class for declarative class definitions
Base = declarative_base()

# Define the Sensor table
class Sensor(Base):
    __tablename__ = 'sensors'
    id = Column(Integer, primary_key=True)
    mqtt_name = Column(String(50))  # Removed unique=True to allow multiple pool sensors with same name
    mqtt_id = Column(String(50))
    position = Column(String(50))
    room = Column(String(50))
    battery = Column(Float)
    rain_offset = Column(Float, default=0.0)  # Cumulative offset for rain sensor resets
    last_rain_value = Column(Float, default=0.0)  # Last reported rain value (for reset detection)
    node_id = Column(Integer, nullable=True)  # For pool sensors: the nodeId (1, 2, etc.) that generates mqtt_id
    sensor_type = Column(String(50), nullable=True)  # Sensor type: 'BME280', 'DS18B20', 'Bresser-6in1', etc.

# Define the TemperatureInside table
class TemperatureInside(Base):
    __tablename__ = 'temperature_inside'
    id = Column(Integer, primary_key=True)
    sensor_id = Column(Integer, ForeignKey('sensors.id'))
    sensor = relationship('Sensor', backref='temperature_inside')
    timestamp = Column(DateTime, default=datetime.utcnow)
    temperature_c = Column(Float)

# Define the TemperatureOutside table
class TemperatureOutside(Base):
    __tablename__ = 'temperature_outside'
    id = Column(Integer, primary_key=True)
    sensor_id = Column(Integer, ForeignKey('sensors.id'))
    sensor = relationship('Sensor', backref='temperature_outside')
    timestamp = Column(DateTime, default=datetime.utcnow)
    temperature_c = Column(Float)


# Define the HumidityInside table
class HumidityInside(Base):
    __tablename__ = 'humidity_inside'
    id = Column(Integer, primary_key=True)
    sensor_id = Column(Integer, ForeignKey('sensors.id'))
    sensor = relationship('Sensor', backref='humidity_inside')
    timestamp = Column(DateTime, default=datetime.utcnow)
    humidity = Column(Float)

# Define the HumidityInside table
class HumidityOutside(Base):
    __tablename__ = 'humidity_outside'
    id = Column(Integer, primary_key=True)
    sensor_id = Column(Integer, ForeignKey('sensors.id'))
    sensor = relationship('Sensor', backref='humidity_outside')
    timestamp = Column(DateTime, default=datetime.utcnow)
    humidity = Column(Float)

# Define the AirPressure table
class AirPressure(Base):
    __tablename__ = 'air_pressure'
    id = Column(Integer, primary_key=True)
    sensor_id = Column(Integer, ForeignKey('sensors.id'))
    sensor = relationship('Sensor', backref='air_pressure')
    timestamp = Column(DateTime, default=datetime.utcnow)
    pressure_rel = Column(Float)

# Define the wind table
class Wind(Base):
    __tablename__ = 'wind'
    id = Column(Integer, primary_key=True)
    sensor_id = Column(Integer, ForeignKey('sensors.id'))
    sensor = relationship('Sensor', backref='wind')
    timestamp = Column(DateTime, default=datetime.utcnow)
    average_speed = Column(Float)
    direction = Column(Float)
    gust = Column(Float)

# Define the precipitation table
class Precipitation(Base):
    __tablename__ = 'precipitation'
    id = Column(Integer, primary_key=True)
    sensor_id = Column(Integer, ForeignKey('sensors.id'))
    sensor = relationship('Sensor', backref='precipitation')
    timestamp = Column(DateTime, default=datetime.utcnow)
    precipitation = Column(Float)