# coding=UTF-8

from api.extensions import db
from flask_sqlalchemy import BaseQuery


class ContractCustomerQuery(BaseQuery):
    def reseller_store(self, url, territory):
        return self.filter(
            ContractCustomer.dealerstore == True,
            ContractCustomer.dealerurl == url,
            ContractCustomer.country == territory,
        ).first()


class ContractCustomer(db.Model):

    __tablename__ = "contract_customers"

    query_class = ContractCustomerQuery

    id = db.Column("contractcustomerid", db.Integer, primary_key=True)
    country = db.Column(db.String(2))
    pricepremium = db.Column(db.Numeric, nullable=False, default=0)
    dealerstore = db.Column(db.Boolean, nullable=False, default=False)
    dealerurl = db.Column(db.String(255), nullable=False)

    def to_json(self):
        return {
            "id": self.id,
            "pricepremium": self.pricepremium,
            "dealerstore": self.dealerstore,
            "dealerurl": self.dealerurl,
        }