Files
baijiahao_text_to_image/create_test_csv.py

61 lines
4.6 KiB
Python
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
创建测试CSV文件用于验证图片文章挂靠效果
"""
import csv
import json
def create_test_articles_csv():
"""创建测试文章CSV文件"""
# 创建测试文章数据与20260108文件夹中的图片主题相关
articles_data = [
{'ID': 1, '标题': '美丽的自然风景欣赏', '内容': '自然界的风景总是让人感到心旷神怡。无论是山川河流还是森林草原,大自然的美景总能带给我们视觉上的享受和心灵上的宁静。', '标签': json.dumps(['自然', '风景', '美丽'], ensure_ascii=False)},
{'ID': 2, '标题': '户外活动的乐趣', '内容': '走出室内,亲近大自然是一种极好的放松方式。户外活动不仅能锻炼身体,还能让我们欣赏到美丽的自然风光。', '标签': json.dumps(['户外', '活动', '自然'], ensure_ascii=False)},
{'ID': 3, '标题': '摄影艺术中的自然之美', '内容': '摄影师们常常将镜头对准大自然的美景,捕捉那些令人惊叹的瞬间。每一张风景照片都是对自然之美的独特诠释。', '标签': json.dumps(['摄影', '自然', '艺术'], ensure_ascii=False)},
{'ID': 4, '标题': '风景旅游推荐指南', '内容': '想要寻找美丽的风景胜地吗?这里有几处绝佳的风景旅游目的地,每一处都有其独特的魅力和美景等待你的探索。', '标签': json.dumps(['旅游', '风景', '推荐'], ensure_ascii=False)},
{'ID': 5, '标题': '数字图像处理技术', '内容': '现代数字图像处理技术使得图片的缩放、裁剪和优化变得更加容易。无论是原图还是缩略图,都能在保持质量的同时方便使用。', '标签': json.dumps(['图像处理', '技术', '缩略图'], ensure_ascii=False)}
]
# 写入CSV文件
with open('test_articles.csv', 'w', newline='', encoding='utf-8-sig') as csvfile:
fieldnames = ['ID', '标题', '内容', '标签']
writer = csv.DictWriter(csvfile, fieldnames=fieldnames)
writer.writeheader()
for row in articles_data:
writer.writerow(row)
print('测试文章CSV文件已创建: test_articles.csv')
def create_test_images_csv():
"""创建测试图片CSV文件"""
# 创建测试图片数据使用20260108文件夹中的图片
images_data = [
{'ID': 1, '图像ID': 'IMG001', '图像名称': '风景图1', '图像URL': '20260108/1767867138994556.png', '标签名称': '风景,自然,美丽', '关键词名称': 'Landscape,Nature,Beauty', '部门名称': '生活部', '附加文章数量': 2},
{'ID': 2, '图像ID': 'IMG002', '图像名称': '风景图2', '图像URL': '20260108/1767867148035776.png', '标签名称': '自然,风光,户外', '关键词名称': 'Nature,Landscape,Outdoor', '部门名称': '生活部', '附加文章数量': 1},
{'ID': 3, '图像ID': 'IMG003', '图像名称': '风景图3', '图像URL': '20260108/1767867156936619.png', '标签名称': '景色,自然,美丽', '关键词名称': 'Scenery,Nature,Beautiful', '部门名称': '生活部', '附加文章数量': 3},
{'ID': 4, '图像ID': 'IMG004', '图像名称': '风景图4', '图像URL': '20260108/1767867165665952.png', '标签名称': '自然风光,户外,美景', '关键词名称': 'Natural Scenery,Outdoor,Beautiful View', '部门名称': '生活部', '附加文章数量': 0},
{'ID': 5, '图像ID': 'IMG005', '图像名称': '缩略图1', '图像URL': '20260108/1767867138994556_thumb.png', '标签名称': '缩略图,小图,预览', '关键词名称': 'Thumbnail,Small Image,Preview', '部门名称': '技术部', '附加文章数量': 4},
{'ID': 6, '图像ID': 'IMG006', '图像名称': '缩略图2', '图像URL': '20260108/1767867148035776_thumb.png', '标签名称': '缩略图,预览,小尺寸', '关键词名称': 'Thumbnail,Preview,Small Size', '部门名称': '技术部', '附加文章数量': 1}
]
# 写入CSV文件
with open('test_images.csv', 'w', newline='', encoding='utf-8-sig') as csvfile:
fieldnames = ['ID', '图像ID', '图像名称', '图像URL', '标签名称', '关键词名称', '部门名称', '附加文章数量']
writer = csv.DictWriter(csvfile, fieldnames=fieldnames)
writer.writeheader()
for row in images_data:
writer.writerow(row)
print('测试图片CSV文件已创建: test_images.csv')
if __name__ == "__main__":
create_test_articles_csv()
create_test_images_csv()
print('\n两个测试CSV文件已创建完成可用于测试图片文章挂靠效果。')