base.py

#encoding:utf-8

from datetime import  datetime

from flask_sqlalchemy import SQLAlchemy as _SQLAlchemy, BaseQuery
from sqlalchemy import Column, Integer, SmallInteger
from contextlib import  contextmanager

from app.libs.error_code import NotFoundException


class SQLAlchemy(_SQLAlchemy):
  @contextmanager
  def auto_commit(self):
    try:
      yield
      self.session.commit()
    except Exception as e:
      db.session.rollback()
      raise e

class Query(BaseQuery):
  def filter_by(self, **kwargs):
    if 'status' not in kwargs.keys():
      kwargs['status'] = 1
    return super(Query, self).filter_by(**kwargs)

  def get_or_404(self, ident):
    rv = self.get(ident)
    if not rv:
      raise NotFoundException()
    return rv

  def first_or_404(self):
    rv = self.first()
    if not rv:
      raise NotFoundException()
    return rv

db = SQLAlchemy(query_class=Query)

class Base(db.Model):
  __abstract__ = True
  create_time = Column(Integer)
  status = Column(SmallInteger, default=1)

  def __init__(self):
    self.create_time = int(datetime.now().timestamp())

  #字典序列化 模型字典 key value时候 keys() 返回值时使用
  def __getitem__(self, item):
    return getattr(self, item)

  @property
  def create_datetime(self):
    if self.create_time:
      return datetime.fromtimestamp(self.create_time)
    else:
      return None

  def set_attrs(self, attrs_dict):
    for key, value in attrs_dict.items():
      if hasattr(self, key) and key != 'id':
        setattr(self, key, value)

  def delete(self):
    self.status = 0