Лучший способ не нарушать принцип 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)