fix: Fixing schemas_allowed_for_upload field in database connection UX (#15750)

This commit is contained in:
Hugh A. Miles II
2021-07-19 12:12:37 -04:00
committed by GitHub
parent 281cb92bfa
commit 7d0f6ab7f5
2 changed files with 6 additions and 2 deletions

View File

@@ -20,6 +20,7 @@ import enum
import json
import logging
import textwrap
from ast import literal_eval
from contextlib import closing
from copy import deepcopy
from datetime import datetime
@@ -668,6 +669,10 @@ class Database(
self,
) -> List[str]:
allowed_databases = self.get_extra().get("schemas_allowed_for_csv_upload", [])
if isinstance(allowed_databases, str):
allowed_databases = literal_eval(allowed_databases)
if hasattr(g, "user"):
extra_allowed_databases = config["ALLOWED_USER_CSV_SCHEMA_FUNC"](
self, g.user