diff options
Diffstat (limited to 'ishtar_common/models_imports.py')
| -rw-r--r-- | ishtar_common/models_imports.py | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/ishtar_common/models_imports.py b/ishtar_common/models_imports.py index 6358f070a..d4f8b5698 100644 --- a/ishtar_common/models_imports.py +++ b/ishtar_common/models_imports.py @@ -353,6 +353,24 @@ class ImporterType(models.Model): newclass = type(name, (Importer,), args) return newclass + def get_columns(self, importer_class=None): + """ + :param importer_class: importer class - if not provided get from self + :return: (columns: list, columns_names: list) - column attributes, column labels + """ + if not importer_class: + importer_class = self.get_importer_class() + cols, col_names = [], [] + for formater in importer_class.LINE_EXPORT_FORMAT: + if not formater: + cols.append("") + col_names.append("") + continue + cols.append(formater.export_field_name) + col_names.append(formater.label) + return cols, col_names + + def save(self, *args, **kwargs): if not self.slug: self.slug = create_slug(ImporterType, self.name) |