Django 数据导出为 CSV

最新推荐文章于 2024-07-26 21:19:18 发布

Leon_Jinhai_Sun

最新推荐文章于 2024-07-26 21:19:18 发布

阅读量97

点赞数 3

文章标签： python

本文链接：https://blog.csdn.net/Leon_Jinhai_Sun/article/details/139798463

版权

import django_setup
from doc.models import *
from doc.version_util import *
import sys

import csv
from datetime import datetime
from django.db.models import QuerySet


def export_to_csv(model_query: QuerySet, fieldnames: list, filename_prefix: str):
    now = datetime.now()
    formatted_time = now.strftime('%Y_%m_%d_%H%M%S')
    csv_filename = f'{filename_prefix}_{formatted_time}.csv'
    csv_data = list(model_query.values(*fieldnames))
    with open(csv_filename, mode='w', newline='', encoding='utf-8') as file:
        writer = csv.DictWriter(file, fieldnames=fieldnames)
        writer.writeheader()
        writer.writerows(csv_data)

    print(f'CSV file has been saved as {csv_filename} in the current directory.')
fieldnames = ['children_item', 'full_path']
valid_vis = VersionedItem.objects.filter(id=52).values(*fieldnames)
export_to_csv(valid_vis, fieldnames, 'exported_data')