mirror of
https://github.com/cisagov/manage.get.gov.git
synced 2025-07-22 18:56:15 +02:00
Update csv_export.py
This commit is contained in:
parent
e2bd982c27
commit
4013815a04
1 changed files with 22 additions and 22 deletions
|
@ -88,6 +88,13 @@ class BaseExport(ABC):
|
||||||
"""
|
"""
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def get_columns(cls):
|
||||||
|
"""
|
||||||
|
Returns the columns for CSV export. Override in subclasses as needed.
|
||||||
|
"""
|
||||||
|
return []
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def get_sort_fields(cls):
|
def get_sort_fields(cls):
|
||||||
"""
|
"""
|
||||||
|
@ -153,6 +160,21 @@ class BaseExport(ABC):
|
||||||
"""
|
"""
|
||||||
return []
|
return []
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def update_queryset(cls, queryset, **kwargs):
|
||||||
|
"""
|
||||||
|
Returns an updated queryset. Override in subclass to update queryset.
|
||||||
|
"""
|
||||||
|
return queryset
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def write_csv_before(cls, csv_writer, **kwargs):
|
||||||
|
"""
|
||||||
|
Write to csv file before the write_csv method.
|
||||||
|
Override in subclasses where needed.
|
||||||
|
"""
|
||||||
|
pass
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def annotate_and_retrieve_fields(
|
def annotate_and_retrieve_fields(
|
||||||
cls, initial_queryset, annotated_fields, related_table_fields=None, include_many_to_many=False, **kwargs
|
cls, initial_queryset, annotated_fields, related_table_fields=None, include_many_to_many=False, **kwargs
|
||||||
|
@ -218,32 +240,10 @@ class BaseExport(ABC):
|
||||||
)
|
)
|
||||||
return cls.annotate_and_retrieve_fields(model_queryset, annotated_fields, related_table_fields, **kwargs)
|
return cls.annotate_and_retrieve_fields(model_queryset, annotated_fields, related_table_fields, **kwargs)
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def update_queryset(cls, queryset, **kwargs):
|
|
||||||
"""
|
|
||||||
Returns an updated queryset. Override in subclass to update queryset.
|
|
||||||
"""
|
|
||||||
return queryset
|
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def get_model_annotation_dict(cls, **kwargs):
|
def get_model_annotation_dict(cls, **kwargs):
|
||||||
return convert_queryset_to_dict(cls.get_annotated_queryset(**kwargs), is_model=False)
|
return convert_queryset_to_dict(cls.get_annotated_queryset(**kwargs), is_model=False)
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def get_columns(cls):
|
|
||||||
"""
|
|
||||||
Returns the columns for CSV export. Override in subclasses as needed.
|
|
||||||
"""
|
|
||||||
return []
|
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def write_csv_before(cls, csv_writer, **kwargs):
|
|
||||||
"""
|
|
||||||
Write to csv file before the write_csv method.
|
|
||||||
Override in subclasses where needed.
|
|
||||||
"""
|
|
||||||
pass
|
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def export_data_to_csv(cls, csv_file, **kwargs):
|
def export_data_to_csv(cls, csv_file, **kwargs):
|
||||||
"""
|
"""
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue