allow selection of dbs where csv can be uploaded to (#5393)

This commit is contained in:
timifasubaa
2018-07-17 15:38:16 -07:00
committed by GitHub
parent 73ec526913
commit 7f8eaee18d
4 changed files with 40 additions and 8 deletions

View File

@@ -49,8 +49,10 @@ def filter_not_empty_values(value):
class CsvToDatabaseForm(DynamicForm):
# pylint: disable=E0211
def all_db_items():
return db.session.query(models.Database)
def csv_enabled_dbs():
return db.session.query(
models.Database).filter_by(
allow_csv_upload=True).all()
name = StringField(
_('Table Name'),
@@ -64,7 +66,7 @@ class CsvToDatabaseForm(DynamicForm):
FileRequired(), FileAllowed(['csv'], _('CSV Files Only!'))])
con = QuerySelectField(
_('Database'),
query_factory=all_db_items,
query_factory=csv_enabled_dbs,
get_pk=lambda a: a.id, get_label=lambda a: a.database_name)
sep = StringField(
_('Delimiter'),