def get_message(id=None, send_status=None, order_by='id', order_type='desc', page_num=1, page_size=10, to_dict=True): '''查询数据库message记录''' data = Message.select() if id: data = data.where(Message.id == id, Message.is_deleted == 0).first() else: if send_status is not None: data = data.where(Message.send_status == send_status) order_field = getattr(Message, order_by) if order_type.lower() == 'desc': data = data.where(Message.is_deleted == 0).order_by(-order_field) else: data = data.where(Message.is_deleted == 0).order_by(+order_field) if page_size > 0 and page_num > 0: data = data.paginate(page_num, page_size) if to_dict: if isinstance(data, peewee.SelectQuery): data = [model2dict(i) for i in data] else: data = model2dict(data) return data
def update_message(id, send_status=None, is_deleted=None, to_dict=True): data = Message.select().where(Message.id == id).first() if not data: return if send_status is not None: data.send_status = send_status if is_deleted is not None: data.is_deleted = is_deleted data.save() if to_dict: data = model2dict(data) return data
def get_record(cls, id=None, order_by='id', order_type='desc', to_dict=True): if id is not None: data = cls.select().where(cls.id == id, cls.is_deleted == 0).first() else: order_field = getattr(cls, order_by) if order_type.lower() == 'asc': data = cls.select().where( cls.is_deleted == 0).order_by(-order_field) else: data = cls.select().where( cls.is_deleted == 0).order_by(+order_field) if to_dict: if isinstance(data, peewee.SelectQuery): data = [model2dict(i) for i in data] else: data = model2dict(data) return data
def update_record(cls, id, send_status=None, is_deleted=None, to_dict=True): data = cls.select().where(cls.id == id, cls.is_deleted == 0).first() if not data: return if send_status is not None: data.send_status = send_status if is_deleted is not None: data.is_deleted = is_deleted data.save() if to_dict: data = model2dict(data) return data
def add_record(cls, message, to_dict=True): data = cls.create(message=message) if to_dict: data = model2dict(data) return data
def add_message(message, to_dict=True): data = Message.create(message=message) if to_dict: data = model2dict(data) return data