diff --git a/api/models/__init__.py b/api/models/__init__.py index 596ca08..90f7c40 100644 --- a/api/models/__init__.py +++ b/api/models/__init__.py @@ -1,2 +1,3 @@ from .designer import Designer from .category import Category +from .contract_customer import ContractCustomer \ No newline at end of file diff --git a/api/models/contract_customer.py b/api/models/contract_customer.py new file mode 100644 index 0000000..22f3252 --- /dev/null +++ b/api/models/contract_customer.py @@ -0,0 +1,19 @@ +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 + }