def check_storage(model, color): storage = Storage.select().where(Storage.model == model, Storage.color == color).get() if storage and storage.num > 0: return True else: return False
def get_storage(model): """通过车型获取 该车型所有库存分类 :param model: :type model: :return: 该车型所有库存 :rtype: json """ storage = Storage.select().where(Storage.model == model) return storage
def get_storage_total(model): """通过车型获取 该车型库存总数 :param model: :type model: :return: :rtype: """ storage = Storage.select().where(Storage.model == model) count = 0 for one in storage: count += one.num return count
def modify_num(model, color, num): """ :param model: :type model: :param color: :type color: :param num: :type num: :return: number of row update, 0 if not find, error if modify_json is wrong :rtype: int """ query = Storage.update(num=num).where(Storage.model == model, Storage.color == color) return query.execute()
def add(**kwargs): """ add storage eg = { "model": "E100小龟", "color": "红", "num": 50 } :param kwargs: :type kwargs: :return: the added json :rtype: json """ store_list = Storage.create(**kwargs) return store_list
def get_by_model_color(model, color): storage = Storage.select().where(Storage.model == model, color == color) return storage
def get_by_id(store_list_id): return Storage.get(Storage.id == store_list_id)
def count_all(): count = Storage.select().count() return count
def get_all_paginate(page, paginate_by): storage = Storage.select().paginate(page, paginate_by) return storage
def get_all(): storage = Storage.select() return storage
def get(*query, **kwargs): store_list = Storage.get(*query, **kwargs) return store_list
def increment_num(model, color): query = Storage.update(num=Storage.num + 1) \ .where(Storage.model == model, Storage.color == color) return query.execute()