fix: 修复地材混入计算结果的问题

This commit is contained in:
han0
2025-03-19 15:23:09 +08:00
parent 1e4c665870
commit d6389510ae
3 changed files with 9 additions and 2 deletions

View File

@@ -35,6 +35,13 @@ class Material(db.Model, Model, OperationTrackMixin):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
@classmethod
def list(cls, type=None):
query = cls.query
if type:
query = query.filter(cls.type == type)
return query.all()
@classmethod
def get_by_key(cls, code):
return cls.query.filter(cls.code == code).one_or_none()