from datetime import datetime, timezone
from flask_sqlalchemy import SQLAlchemy

db = SQLAlchemy()


class Zipcode(db.Model):
    __tablename__ = "zipcodes"
    zip = db.Column(db.Integer, primary_key=True)
    results = db.Column(db.JSON, nullable=False)
    timestamp = db.Column(
        db.DateTime(timezone=True), default=datetime.now(timezone.utc), nullable=False
    )


class Weather(db.Model):
    __tablename__ = "weather"
    lat_long = db.Column(db.String(100), primary_key=True)
    results = db.Column(db.JSON, nullable=False)
    last_timestamp = db.Column(
        db.DateTime(timezone=True),
        default=datetime.now(timezone.utc),
        onupdate=datetime.now(timezone.utc),
        nullable=False,
    )


class Stats(db.Model):
    __tablename__ = "stats"
    id = db.Column(db.Integer, autoincrement=True, primary_key=True)
    lat_long = db.Column(db.String(100), nullable=False)
    timestamp = db.Column(
        db.DateTime(timezone=True), default=datetime.now(timezone.utc), nullable=False
    )