from sqlalchemy import Column, Integer, String, Numeric, Date, UniqueConstraint, func

from commons.models.mixin.steel import SteelMixin
from commons.models.model import Model
from core.extensions import db


class SteelPlate(db.Model, Model, SteelMixin):
    __tablename__ = 'STEEL_PLATE'
    id = Column('ID', Integer, primary_key=True)
    name = Column('NAME', String(128), default='', comment='名称')
    spec = Column('SPEC', String(128), default='', comment='规格')
    material = Column('MATERIAL', String(64), default='', comment='材质')
    source = Column('SOURCE', String(64), default='', comment='产地')
    price = Column('PRICE', Numeric(16, 4), default=0, comment='价格')
    fluctuating = Column('FLUCTUATING', Numeric(16, 4), default=0, comment='浮动')
    date = Column('DATE', Date, comment='日期')

    __table_args__ = (
        UniqueConstraint(name, spec, material, source, date,  name='Idx_key'),
        {'comment': '中厚板'},
    )

    @classmethod
    def get_items(cls, year, month, spec_in=('12', '16-20', '22-28'), source_in=('三钢闽光',), name_in=('普中板', '工字钢')):
        query = cls.query
        query = cls.query_by_month(query, year, month)
        query = query.filter(cls.spec.in_(spec_in))
        query = query.filter(cls.source.in_(source_in))
        query = query.filter(cls.name.in_(name_in))
        query = query.with_entities(cls.spec, func.avg(cls.price))
        query = query.group_by(cls.spec)
        result = query.all()
        return result