Allow dataset owners to see their datasets

CybercentreCanada · May 18, 2022 · ce73018 · ce73018
1 parent 4005527
commit ce73018
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/superset/views/base.py b/superset/views/base.py
@@ -613,10 +613,19 @@ def apply(self, query: Query, value: Any) -> Query:
  return query
  datasource_perms = security_manager.user_view_menu_names("datasource_access")
  schema_perms = security_manager.user_view_menu_names("schema_access")
+ owner_ids_query = (
+ db.session.query(models.SqlaTable.id)
+ .join(models.SqlaTable.owners)
+ .filter(
+ security_manager.user_model.id
+ == security_manager.user_model.get_user_id()
+ )
+ )
  return query.filter(
  or_(
  self.model.perm.in_(datasource_perms),
  self.model.schema_perm.in_(schema_perms),
+ models.SqlaTable.id.in_(owner_ids_query),
  )
  )