Skip to content

Commit

Permalink
fix: remove expose (#19700)
Browse files Browse the repository at this point in the history
* bumping shillelagh

* remove expose
  • Loading branch information
AAfghahi committed Apr 17, 2022
1 parent ae70212 commit 57157c8
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 17 deletions.
5 changes: 1 addition & 4 deletions superset/databases/api.py
Original file line number Diff line number Diff line change
Expand Up @@ -169,10 +169,7 @@ class DatabaseRestApi(BaseSupersetModelRestApi):

edit_columns = add_columns

search_filters = {
"allow_file_upload": [DatabaseUploadEnabledFilter],
"expose_in_sqllab": [DatabaseFilter],
}
search_filters = {"allow_file_upload": [DatabaseUploadEnabledFilter]}

list_select_columns = list_columns + ["extra", "sqlalchemy_uri", "password"]
order_columns = [
Expand Down
13 changes: 1 addition & 12 deletions superset/databases/filters.py
Original file line number Diff line number Diff line change
Expand Up @@ -69,8 +69,6 @@ class DatabaseUploadEnabledFilter(BaseFilter): # pylint: disable=too-few-public
def apply(self, query: Query, value: Any) -> Query:
filtered_query = query.filter(Database.allow_file_upload)

database_perms = security_manager.user_view_menu_names("database_access")
schema_access_databases = can_access_databases("schema_access")
datasource_access_databases = can_access_databases("datasource_access")

if hasattr(g, "user"):
Expand All @@ -82,19 +80,10 @@ def apply(self, query: Query, value: Any) -> Query:
if len(allowed_schemas):
return filtered_query

filtered_query = filtered_query.filter(
return filtered_query.filter(
or_(
cast(Database.extra, JSON)["schemas_allowed_for_file_upload"]
is not None,
cast(Database.extra, JSON)["schemas_allowed_for_file_upload"] != [],
)
)

return filtered_query.filter(
or_(
self.model.perm.in_(database_perms),
self.model.database_name.in_(
[*schema_access_databases, *datasource_access_databases]
),
)
)
2 changes: 1 addition & 1 deletion tests/integration_tests/databases/api_tests.py
Original file line number Diff line number Diff line change
Expand Up @@ -1135,7 +1135,7 @@ def test_get_allow_file_upload_false_csv(self):
uri = f"api/v1/database/?q={prison.dumps(arguments)}"
rv = self.client.get(uri)
data = json.loads(rv.data.decode("utf-8"))
assert data["count"] == 0
assert data["count"] == 1

def test_get_allow_file_upload_filter_no_permission(self):
"""
Expand Down

0 comments on commit 57157c8

Please sign in to comment.