from app import db, ma
from datetime import datetime


class Admin_Company(db.Model):
    __tablename__ = 'admin_company'
    __table_args__ = {"schema": "public"}

    id = db.Column(db.Integer,
                   primary_key=True,
                   autoincrement=True,
                   unique=False)
    creation_date = db.Column(db.TIMESTAMP(), default=datetime.utcnow())
    account_prefix = db.Column(db.VARCHAR(5))
    company_name = db.Column(db.VARCHAR(250))
    company_address = db.Column(db.VARCHAR(250))
    company_town = db.Column(db.VARCHAR(100))
    company_zip = db.Column(db.VARCHAR(25))
    company_state = db.Column(db.INTEGER())
    company_phone_number = db.Column(db.VARCHAR(50))

class Admin_Company_schema(ma.SQLAlchemyAutoSchema):
    class Meta:
        model = Admin_Company

# --- ADD THIS ENTIRE NEW MODEL ---
class Call(db.Model):
    __tablename__ = "call_call"
    __table_args__ = {"schema": "public"}

    id = db.Column(db.Integer, primary_key=True, index=True)
    current_phone = db.Column(db.String(500))
    created_at = db.Column(db.DateTime, default=datetime.utcnow)