35 lines
1004 B
Python
35 lines
1004 B
Python
# 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,
|
|
}
|