# coding=UTF-8

from flask_sqlalchemy import BaseQuery
from api import db


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
        }