45 lines
1.7 KiB
Python
45 lines
1.7 KiB
Python
from sqlalchemy import Column, Integer, String
|
||
|
||
from commons.models.mixin.operation_track import OperationTrackMixin
|
||
from core.extensions import db
|
||
|
||
|
||
class Material(db.Model, OperationTrackMixin):
|
||
id = Column('ID', String(128), primary_key=True)
|
||
parent_id = Column('PARENT_ID', String(128))
|
||
category_1 = Column('CATEGORY1', String(128), default='', comment='分类1')
|
||
category_2 = Column('CATEGORY2', String(128), default='', comment='分类2')
|
||
category_3 = Column('CATEGORY3', String(128), default='', comment='分类3')
|
||
category_4 = Column('CATEGORY4', String(128), default='', comment='分类4')
|
||
name = Column('NAME', String(128), default='', comment='名称')
|
||
unit = Column('UNIT', String(128), default='', comment='单位')
|
||
spec = Column('SPEC', String(128), default='', comment='规格')
|
||
tax = Column('TAX', Integer, default=0, comment='税率（%）')
|
||
is_builtin = Column('IS_BUILTIN', Integer, default=0, comment='是否初始内建类型（不允许删除）')
|
||
type = Column('TYPE', Integer, default=0, comment='材料类别（主材、地材）')
|
||
is_tree = Column('IS_TREE', Integer, default=0, comment='是否树')
|
||
|
||
__tablename__ = 'MATERIAL'
|
||
__table_args__ = (
|
||
{'comment': '材料'},
|
||
)
|
||
|
||
def __init__(self, *args, **kwargs):
|
||
super().__init__(*args, **kwargs)
|
||
|
||
@classmethod
|
||
def get_by_key(cls, id):
|
||
return cls.query.filter(cls.id == id).one_or_none()
|
||
|
||
def upsert(self):
|
||
result = self.get_by_key(self.id)
|
||
if not result:
|
||
session = db.session
|
||
session.add(self)
|
||
session.commit()
|
||
else:
|
||
session = db.session
|
||
self.id = result.id
|
||
session.add(result)
|
||
session.commit()
|