um-cseg/chez-betty

from .model import *
from . import vendor
from . import item

class ItemVendor(Base):
    __tablename__ = 'item_vendors'

    id          = Column(Integer, primary_key=True, nullable=False)
    vendor_id   = Column(Integer, ForeignKey("vendors.id"), nullable=False)
    item_id     = Column(Integer, ForeignKey("items.id"), nullable=False)
    item_number = Column(String(255), nullable=False)

    enabled     = Column(Boolean, default=True, nullable=False)

    vendor      = relationship(
                      vendor.Vendor,
                      primaryjoin="and_(ItemVendor.vendor_id==Vendor.id, ItemVendor.enabled==True)",
                      backref="items"
                  )
    item         = relationship(
                      item.Item,
                      primaryjoin="and_(ItemVendor.item_id==Item.id, ItemVendor.enabled==True)",
                      backref="vendors"
                  )

    def __init__(self, vendor, item, item_number, enabled=True):
        self.vendor_id   = vendor.id
        self.item_id     = item.id
        self.item_number = item_number
        self.enabled     = enabled

    @classmethod
    def from_id(cls, id):
        return DBSession.query(cls).filter(cls.id == id).one()

    @classmethod
    def from_number_fuzzy(cls, number):
        return DBSession.query(cls)\
            .filter(cls.item_number.like('%{}%'.format(number)))\
            .filter(cls.enabled)\
            .all()

@property
def __all_items(self):
    return DBSession.query(item.Item)\
            .join(ItemVendor)\
            .filter(ItemVendor.vendor_id == self.id)\
            .filter(ItemVendor.enabled==True)\
            .filter(item.Item.enabled==True)\
            .order_by(item.Item.name)
vendor.Vendor.all_items = __all_items