diff --git a/advlabdb/customClasses.py b/advlabdb/customClasses.py index a987be6..024c3a6 100644 --- a/advlabdb/customClasses.py +++ b/advlabdb/customClasses.py @@ -106,6 +106,7 @@ class CustomModelView(ModelView): # Should not be a copy of column_formatters # because of link formatting. column_formatters_export = {} + # Used in the UserView because of create_user # Should not be touched in other views _skip_session_addition_on_model_creation = False @@ -207,6 +208,16 @@ class CustomModelView(ModelView): return formClass(get_form_data(), obj=obj) + def get_export_columns(self): + # Use column_details_list instead of column_list + + only_columns = self.column_export_list or self.column_details_list or self.scaffold_list_columns() + + return self.get_column_names( + only_columns=only_columns, + excluded_columns=self.column_export_exclude_list, + ) + class SecureAdminModelView(CustomModelView): can_export = True