Files
baijiahao_text_to_image/create_test_excel.py

69 lines
3.7 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 -*-
"""
创建测试Excel文件用于验证图片文章挂靠效果
"""
from openpyxl import Workbook
import json
def create_test_articles_excel():
"""创建测试文章Excel文件"""
# 创建测试文章数据
articles_data = [
['ID', '标题', '内容', '标签'], # 表头
[1, '人工智能发展趋势', '人工智能技术正在快速发展,在各个领域都有广泛应用。机器学习、深度学习等技术不断突破,推动着社会进步。', json.dumps(['人工智能', '科技', '趋势'], ensure_ascii=False)],
[2, '健康饮食的重要性', '合理膳食是保持身体健康的基础。均衡摄入各种营养素,有助于提高免疫力,预防疾病。', json.dumps(['健康', '饮食', '营养'], ensure_ascii=False)],
[3, '环境保护与可持续发展', '环境保护是当今世界面临的重要挑战。通过可持续发展策略,我们可以平衡经济发展与生态保护。', json.dumps(['环保', '可持续发展', '生态'], ensure_ascii=False)],
[4, '数字化转型对企业的影响', '数字化转型正在重塑企业运营模式。通过引入新技术,企业可以提升效率,优化客户体验。', json.dumps(['数字化', '企业', '转型'], ensure_ascii=False)],
[5, '教育创新的未来方向', '教育创新是培养未来人才的关键。利用新技术手段,可以创造更加个性化和高效的学习环境。', json.dumps(['教育', '创新', '学习'], ensure_ascii=False)]
]
# 创建Excel工作簿
wb = Workbook()
ws = wb.active
if ws:
ws.title = '测试文章数据'
# 将数据添加到工作表
for row_data in articles_data:
ws.append(row_data)
# 保存Excel文件
wb.save('test_articles.xlsx')
print('测试文章Excel文件已创建: test_articles.xlsx')
def create_test_images_excel():
"""创建测试图片Excel文件"""
# 创建测试图片数据
images_data = [
['ID', '图像ID', '图像名称', '图像URL', '标签名称', '关键词名称', '部门名称', '附加文章数量'], # 表头
[1, 'IMG001', 'AI概念图', 'https://example.com/images/ai_concept.jpg', '人工智能,科技,趋势', 'AI,Machine Learning,Deep Learning', '科技部', 2],
[2, 'IMG002', '健康饮食图', 'https://example.com/images/healthy_food.jpg', '健康,饮食,营养', 'Nutrition,Health,Diet', '生活部', 1],
[3, 'IMG003', '环保地球图', 'https://example.com/images/environment.jpg', '环保,可持续发展,生态', 'Environment,Sustainability,Eco', '环保部', 3],
[4, 'IMG004', '数字化办公图', 'https://example.com/images/digital_office.jpg', '数字化,企业,转型', 'Digital,Enterprise,Transformation', '科技部', 0],
[5, 'IMG005', '教育创新图', 'https://example.com/images/education_innovation.jpg', '教育,创新,学习', 'Education,Innovation,Learning', '教育部', 4],
[6, 'IMG006', '网络安全图', 'https://example.com/images/cyber_security.jpg', '安全,网络,防护', 'Security,Cyber,Protection', '安全部', 1]
]
# 创建Excel工作簿
wb = Workbook()
ws = wb.active
if ws:
ws.title = '测试图片数据'
# 将数据添加到工作表
for row_data in images_data:
ws.append(row_data)
# 保存Excel文件
wb.save('test_images.xlsx')
print('测试图片Excel文件已创建: test_images.xlsx')
if __name__ == "__main__":
create_test_articles_excel()
create_test_images_excel()
print('\n两个测试Excel文件已创建完成可用于测试图片文章挂靠效果。')