from api import db


class ContractCustomer(db.Model):
    __tablename__ = 'contract_customers'

    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
        }