租戶隔離

一個客戶就是一個租戶，每個租戶的數據在數據表中都有個一個tenantid字段用來與其他租戶隔離

• 租戶識別
  

from flask import g, request
    
def get_tenant_from_request():
    auth = validate_auth(request.headers.get('Authorization'))
    return Tenant.query.get(auth.tenant_id)
        
def get_current_tenant():
    rv = getattr(g, 'current_tenant', None)
    if rv is = None:
        rv = get_tenant_from_request()
        g.current_tenant = rv
    return rv

• 自動租戶隔離

例如，每個租戶有自己的Project，像下面這樣批量直接修改project又忘記帶上tenantid查詢字段，就會把其他租戶的project一并修改了

def batch_update_projects(ids, changes):
    projects = Project.query.filter(
        Project.id.in_(ids) &
        Project.status != ProjectStatus.INVISIBLE
    )
    for project in projects: 
        update_projects(project, changes)

我們可以override Project的query屬性以及使用sqlalchemy的event listener來自動的進行租戶隔離，也就是說我們在使用orm查詢的時候，會自動帶上tenentid查詢字段，這樣無論是修改、刪除、查詢都只會是操作自己租戶下面的數據，畢竟在orm中，刪除和修改都是要先查詢的嘛

# model
class Project(TenantBoundMixin, db.Model):
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(100))
    status = db.Column(db.Integer)
    def __repr__(self):
        return '<Project name=%r>' % self.name

from sqlalchemy.ext.declarative import declared_attr
?
# 覆寫Project類的query_class屬性
class TenantQuery(db.Query):
    current_tenant_constrained = True
    def tenant_unconstrained_unsafe(self):
        rv = self._clone()
        rv.current_tenant_constrained = False
        return rv
# 通過mixin的方式，讓Project類多繼承這個類
# 從而覆寫了Project類的query_class屬性
class TenantBoundMixin(object):
    query_class = TenantQuery
    # 為project表定義了一個tenant_id字段
    @declared_attr
    def tenant_id(cls):
        return db.Column(db.Integer, db.ForeignKey('tenant.id'))
?
    @declared_attr
    def tenant(cls):
        return db.relationship(Tenant, uselist=False)
?
# event_listener的意思是，每次執行例如Project.query的時候都會執行
# 下面的鉤子函數，
# 這個鉤子函數會給query對象帶上ilter_by(tenant=get_current_tenant())
@db.event.listens_for(TenantQuery, 'before_compile', retval=True)
def ensure_tenant_constrained(query):
    for desc in query.column_descriptions:
        if hasattr(desc['type'], 'tenant') and \
            query.current_tenant_constrained:
              query = query.filter_by(tenant=get_current_tenant())
    return query

接下來我們來看一下使用

# 只查詢我這個租戶名下的project42
>>> Project.query.all()
[<Project name='project42'>]
# 去掉租戶約束，查詢所有project信息
>>> Project.query.tenant_unconstrained_unsafe().all()
[<Project name='project1'>, Project.name='project2', ...]

審計日志

def log(action, message=None):
  data = {
        'action': action
        'timestamp': datetime.utcnow()
  }
  if message is not None:
         data['message'] = message
  if request:
         data['ip'] = request.remote_addr
  user = get_current_user()
  if user is not None:
         data['user'] = User
  db.session.add(LogMessage(**data))

更多詳細信息，請參考

flask多租戶實踐
完整版
https://johnson329.github.io/flask/flask%E6%9E%84%E5%BB%BAsaas%E5%BA%94%E7%94%A8/