我需要导出到我的数据库的csv用户,所以我或多或少地清楚(https://docs.djangoproject.com/en/dev/howto/outputting-csv/),但我需要选择先前导出的字段。
使用该模型是否可以显示您可以选择要导出的字段?
答案 0 :(得分:2)
在控制台中,您只需尝试
即可>>> from django.contrib.auth.models import User
>>> with open('myfile.csv', 'w') as csv:
... for user in User.objects.all():
... d = '%s, %s, %s,\n' % (user.username, user.last_name, user.first_name)
... csv.write(d)
其中将d
设置为您要保存的字段。
当然,如果你想多次这样做会很痛苦 - 创建一个custom admin command可能是一个更好的主意。
答案 1 :(得分:2)
针对您的问题的最佳解决方案是在您的用户管理员中创建一个操作。方法如下:转到您的admin.py并执行此操作
from django.contrib.auth import get_user_model
from django.http import HttpResponse
import csv, datetime
User = get_user_model()
def export_to_csv(modeladmin, request, queryset):
opts = modeladmin.model._meta
response = HttpResponse(content_type='text/csv')
response['Content-Disposition'] = 'attachment;' 'filename{}.csv'.format(opts.verbose_name)
writer = csv.writer(response)
fields = [field for field in opts.get_fields() if not field.many_to_many and not field.one_to_many]
# Write a first row with header information
writer.writerow([field.verbose_name for field in fields])
# Write data rows
for obj in queryset:
data_row = []
for field in fields:
value = getattr(obj, field.name)
if isinstance(value, datetime.datetime):
value = value.strftime('%d/%m/%Y')
data_row.append(value)
writer.writerow(data_row)
return response
export_to_csv.short_description = 'Export to CSV' #short description
@admin.register(User)
class UserAdmin(admin.ModelAdmin):
'''
'''
actions = [export_to_csv] #register the action in your model admin
通过此操作,您可以选择要导出其数据的特定用户并立即导出。
答案 2 :(得分:1)
从 Django 3.2 开始,您还必须通过添加装饰器将 export_as_csv
注册为“动作”。有关官方文档,请参阅此 link。