Django Admin中增加導(dǎo)出CSV功能過程解析
參考 https://books.agiliq.com/projects/django-admin-cookbook/en/latest/export.html
在使用Django Admin時, 對于列表我們有時需要提供數(shù)據(jù)導(dǎo)出功能, 如下圖:
增加導(dǎo)出CSV功能

在Django Admin中每個模型的Admin類(繼承至admin.ModelAdmin), 我們可以通過actions增加支持的動作, 值為當(dāng)前類存在的方法名, 例如:
.......
@admin.register(Issue)
class IssueAdmin(admin.ModelAdmin):
......
actions = ['export_as_csv'] # 增加動作, 對應(yīng)相應(yīng)的方法名
def export_as_csv(self, request, queryset): # 具體的導(dǎo)出csv方法的實現(xiàn)
pass
export_as_csv.short_description = '導(dǎo)出CSV' # 該動作在admin中的顯示文字
導(dǎo)出CSV方法詳細實現(xiàn)如下:
def export_as_csv(self, request, queryset):
meta = self.model._meta # 用于確定導(dǎo)出的文件名, 格式為: app名.模型類名
field_names = [field.name for field in meta.fields] # 所有屬性名
response = HttpResponse(content_type='text/csv') # 指定響應(yīng)內(nèi)容類型
response['Content-Disposition'] = f'attachment; filename={meta}.csv'
response.charset = 'utf-8-sig' # 可選, 修改編碼為帶BOM的utf-8格式(Excel打開不會有亂碼)
writer = csv.writer(response)
writer.writerow(field_names) # 將屬性名寫入csv
for obj in queryset: # 遍歷要導(dǎo)出的對象列表
row = writer.writerow([getattr(obj, field) for field in field_names]) # 將當(dāng)前對象的各屬性值寫入csv
return response
由于導(dǎo)出CSV動作可以作為各個模型的通用動作, 我們可以封裝成一個Mixin類使用, 完整代碼如下:
import csv
from django.contrib import admin
from django.http import HttpResponse
from .models import Issue
class ExportCsvMixin(object):
def export_as_csv(self, request, queryset):
meta = self.model._meta
field_names = [field.name for field in meta.fields]
response = HttpResponse(content_type='text/csv')
response['Content-Disposition'] = f'attachment; filename={meta}.csv'
response.charset = 'utf-8-sig'
writer = csv.writer(response)
writer.writerow(field_names)
for obj in queryset:
row = writer.writerow([getattr(obj, field) for field in field_names])
return response
export_as_csv.short_description = '導(dǎo)出CSV'
@admin.register(Issue)
class IssueAdmin(admin.ModelAdmin, ExportCsvMixin):
fields = ('key', 'summary', 'status', 'project',
'origin', 'components', 'prj_level', 'prj_category',
'assignee', 'origin_person', 'pm', 'dev_manager', 'test_manager', 'tester', 'fe_dev', 'backend_dev',
'plan_begin', 'plan_end', 'fe_plan_begin', 'fe_plan_end', 'test_plan_begin',
'test_plan_end', 'backend_plan_begin', 'backend_plan_end',
'created', 'reopen', 'prd_begin', 'prd_end', 'dev_begin', 'dev_end',
'test_begin', 'test_end', 'pm_check', 'ready', 'pause', 'done',
'pm_take', 'dev_take', 'test_take', 'total_take',
'tags',
)
readonly_fields = fields
list_display = ('key', 'summary', 'status', 'origin', 'components', 'created', 'visit')
list_filter = ('origin', 'components', 'status', 'tags')
search_fields = ('key', 'summary')
date_hierarchy = 'created'
actions = ['export_as_csv']
以上就是本文的全部內(nèi)容,希望對大家的學(xué)習(xí)有所幫助,也希望大家多多支持腳本之家。
相關(guān)文章
Python+tkinter使用80行代碼實現(xiàn)一個計算器實例
這篇文章主要介紹了Python+tkinter使用80行代碼實現(xiàn)一個計算器實例,具有一定借鑒價值,需要的朋友可以參考下2018-01-01
python使用scapy模塊實現(xiàn)ARP掃描的過程
這篇文章主要介紹了python使用scapy模塊實現(xiàn)ARP掃描的過程,本文給大家介紹的非常詳細,對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值,需要的朋友可以參考下2021-01-01

