caravel 控制某角色只能看到自己的表

背景：

      每个用户应该只能看到自己有权限的表，但是我拿到这个版本的时候 caravel还没做限制，即用户可以看到所有表。

     然后我就用自己的方法 限制了，如下：

    #views.py

class FilterTables(CaravelFilter):
def apply(self, query, func): # noqa
if any([r.name in ('Admin', 'Alpha') for r in get_user_roles()]):
return query
sql="""
select * from ab_view_menu where id in
(select view_menu_id from ab_permission_view where id in
(select permission_view_id from ab_permission_view_role where role_id=5)
and permission_id=31
)
"""
logging.info(sql)
result=db.session.execute(sql)
idsinfo=result.fetchall()
ids=[]
for item in idsinfo:
ids.append(item[1].split(':')[1].split(')')[0])
qry = query.filter(self.model.id.in_(ids))
return qry
增加了一个内部类之后，调用：

#views TableModelView 代码段中调用
base_filters = [['id', FilterTables, lambda: []]]
#post_add(self, table)函数前

当当当当，完毕。

回到GitHub上看了一眼，呀，版本升级，已经实现了呀。

看了下代码，人家风格更统一呢，

你看人家是这样写的：

#views.py

class TableSlice(CaravelFilter):
def apply(self, query, func): # noqa
if any([r.name in ('Admin', 'Alpha') for r in get_user_roles()]):
return query
perms = self.get_perms()
tables = []
for perm in perms:
match = re.search(r'\(id:(\d+)\)', perm)
tables.append(match.group(1))
qry = query.filter(self.model.id.in_(tables))
return qry

#views.py 调用
base_filters = [['id', TableSlice, lambda: []]]

都可以哦~ 你可以自己看情况来。