71 lines
1.7 KiB
Python
71 lines
1.7 KiB
Python
# coding=UTF-8
|
|
|
|
from flask_sqlalchemy import BaseQuery
|
|
from api import db
|
|
|
|
|
|
class Currencies(db.Model):
|
|
|
|
__tablename__ = 'product-currencies'
|
|
|
|
iso3char = db.Column(db.String(3), primary_key=True)
|
|
exchange_rate = db.Column(db.Numeric, default=1)
|
|
|
|
|
|
class MaterialQuery(BaseQuery):
|
|
|
|
def wallpapers(self):
|
|
return self.filter(Material.name.in_(Material.WALLPAPER_MATERIALS))
|
|
|
|
def canvas(self):
|
|
return self.filter(Material.name.in_(Material.CANVAS_MATERIALS))
|
|
|
|
|
|
class Material(db.Model):
|
|
|
|
__tablename__ = 'product-materials'
|
|
|
|
query_class = MaterialQuery
|
|
|
|
WALLPAPER_MATERIALS = ['standard-wallpaper',
|
|
# 'self-adhesive-wallpaper', // isn't sold now
|
|
'premium-wallpaper']
|
|
CANVAS_MATERIALS = ['standard-canvas']
|
|
|
|
id = db.Column('materialid', db.Integer, primary_key=True)
|
|
name = db.Column('material', db.String(255), nullable=False)
|
|
material_price = db.relationship('MaterialPrice', uselist=False)
|
|
|
|
@property
|
|
def price(self):
|
|
if self.material_price is not None:
|
|
return float(self.material_price.price)
|
|
return None
|
|
|
|
@price.setter
|
|
def price(self, value):
|
|
if self.material_price:
|
|
self.material_price.price = value
|
|
|
|
def to_json(self):
|
|
return {
|
|
'id': self.id,
|
|
'name': self.name,
|
|
'price': float(self.price)
|
|
}
|
|
|
|
def __repr__(self):
|
|
return self.name
|
|
|
|
|
|
class MaterialPrice(db.Model):
|
|
|
|
__tablename__ = 'v_materialprice'
|
|
|
|
groupid = db.Column(db.Integer)
|
|
materialid = db.Column(db.Integer, db.ForeignKey(
|
|
'product-materials.materialid'))
|
|
price = db.Column(db.Numeric)
|
|
|
|
__table_args__ = (db.PrimaryKeyConstraint(groupid, materialid),)
|