from app import db, ma
from datetime import datetime


class Service_Service(db.Model):
    __tablename__ = 'service_service'
    __table_args__ = {"schema": "public"}
    
    id = db.Column(db.Integer,
                   primary_key=True,
                   autoincrement=True,
                   unique=False)
    
    customer_id = db.Column(db.INTEGER)
    customer_name = db.Column(db.VARCHAR(1000))
    customer_address = db.Column(db.VARCHAR(1000))
    customer_town = db.Column(db.VARCHAR(140))
    customer_state = db.Column(db.VARCHAR(140))
    customer_zip = db.Column(db.VARCHAR(10))
    # tune-up = 0
    # no heat = 1
    # fix = 2
    # tank = 3
    # other = 4
    type_service_call = db.Column(db.INTEGER)
    when_ordered = db.Column(db.DATETIME())
    scheduled_date = db.Column(db.DATETIME())
    description = db.Column(db.TEXT())

    service_cost = db.Column(db.Numeric(10, 2), nullable=False)
    
    payment_type = db.Column(db.INTEGER, nullable=True)
    payment_card_id = db.Column(db.INTEGER, nullable=True)
    payment_status = db.Column(db.INTEGER, nullable=True)

    
class Service_Service_schema(ma.SQLAlchemyAutoSchema):
    class Meta:
        model = Service_Service
    scheduled_date = ma.DateTime(format='%Y-%m-%dT%H:%M:%S')
    when_ordered = ma.DateTime(format='%Y-%m-%dT%H:%M:%S')



class Service_Parts(db.Model):
    __tablename__ = 'service_parts'
    __table_args__ = {"schema": "public"}
    
    id = db.Column(db.Integer,
                   primary_key=True,
                   autoincrement=True,
                   unique=False)
    
    customer_id = db.Column(db.INTEGER)
    oil_filter = db.Column(db.VARCHAR(100))
    oil_filter_2 = db.Column(db.VARCHAR(100))
    oil_nozzle = db.Column(db.VARCHAR(10))
    oil_nozzle_2 = db.Column(db.VARCHAR(10))
    hot_water_tank = db.Column(db.INTEGER)

    
class Service_Parts_schema(ma.SQLAlchemyAutoSchema):
    class Meta:
        model = Service_Parts
    scheduled_date = ma.DateTime(format='%Y-%m-%dT%H:%M:%S')
    when_ordered = ma.DateTime(format='%Y-%m-%dT%H:%M:%S')



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

    id = db.Column(db.Integer,
                   primary_key=True,
                   autoincrement=True,
                   unique=False)

    customer_id = db.Column(db.INTEGER)
    contract_plan = db.Column(db.INTEGER, default=0)  # 0=no contract, 1=standard, 2=premium
    contract_years = db.Column(db.INTEGER, default=1)
    contract_start_date = db.Column(db.DATE())


class Service_Plans_schema(ma.SQLAlchemyAutoSchema):
    class Meta:
        model = Service_Plans
    contract_start_date = ma.DateTime(format='%Y-%m-%d')