Лучший способ не нарушать принцип DRY. Django/Виды на основе классов

Вот представление, которое я использовал для загрузки документа excel на основе данных, которые у меня есть:

class ExcelDownloadAllView(TemplateView):
    def get(self, request, *args, **kwargs):
        response = HttpResponse(content_type='application/ms-excel')
        response['Content-Disposition'] = 'attachment; filename="All_of_Projects.xls"'

        wb = xlwt.Workbook(encoding='utf-8')
        ws = wb.add_sheet('All Projects Data') # this will make a sheet named Users Data

        # Sheet header, first row
        row_num = 0

        font_style = xlwt.XFStyle()
        font_style.font.bold = True

        columns = ['Username', 'First Name', 'Last Name', 'Email Address', 'Summary']
        rows = BugTracker.objects.all().values_list('project_number','assignee','priority','status','summary',) 

        for col_num in range(len(columns)):
            ws.write(row_num, col_num, columns[col_num], font_style) # at 0 row 0 column 

        # Sheet body, remaining rows
        font_style = xlwt.XFStyle()
        for row in rows:
            row_num += 1
            for col_num in range(len(row)):
                ws.write(row_num, col_num, row[col_num], font_style)

        wb.save(response)

        return response

Это работает хорошо, но теперь я хочу создать другой CBV с другим фильтром "row=". Как лучше всего это сделать, используя принципы DRY?

Я пробовал что-то вроде. Но переменная rows будет перезаписана предыдущим классом.

class ExcelDownloadOpenView(ExcelDownloadAllView):
    
    def get(self, request, *args, **kwargs):
        rows = BugTracker.objects.all().values_list('project_number','assignee', 'priority')
        return super().get(request, *args, **kwargs)

Есть предложения без копирования всего метода и замены переменной rows?

Напишите функцию для создания файла и прикрепления его к ответу, и расположите эту функцию во внешнем файле, чтобы очистить ваш код. Например, допустим, модуль (приложение) называется core:

utils.py

import  xlwt
from django.http import HttpResponse

def create_sheet_response(columns, rows, filename):
    response = HttpResponse(headers={
        'Content-Type': 'application/vnd.ms-excel',
        'Content-Disposition': f'attachment; filename={filename}',
    })

    wb = xlwt.Workbook(encoding='utf-8')
    ws = wb.add_sheet('All Projects Data')

    row_num = 0

    font_style = xlwt.XFStyle()
    font_style.font.bold = True

    for col_num in range(len(columns)):
        ws.write(row_num, col_num, columns[col_num], font_style)

    font_style = xlwt.XFStyle()
    for row in rows:
        row_num += 1
        for col_num in range(len(row)):
            ws.write(row_num, col_num, row[col_num], font_style)

    wb.save(response)

    return response

views.py:

from core.models import BugTracker
from core.utils import create_sheet_response

from django.views.generic import TemplateView

class ExcelDownloadAllView(TemplateView):
    def get(self, request, *args, **kwargs):
        filename = 'All_of_Projects.xls'
        columns = ['Username', 'First Name', 'Last Name', 'Email Address', 'Summary']

        fields = 'project_number','assignee','priority','status','summary'
        rows = BugTracker.objects.all().values_list(*fields)

        return create_sheet_response(columns, rows, filename)

class ExcelDownloadOpenView(TemplateView):
    def get(self, request, *args, **kwargs):
        filename = 'NotAll_of_Projects.xls'
        columns = ['Username', 'First Name', 'Last Name']

        fields = 'project_number','assignee', 'priority'
        rows = BugTracker.objects.all().values_list(*fields)

        return create_sheet_response(columns, rows, filename)
Вернуться на верх