#!/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文件已创建完成，可用于测试图片文章挂靠效果。')