66 lines
1.7 KiB
Python
66 lines
1.7 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
"""
|
|||
|
|
正确的dataclass继承测试
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
from dataclasses import dataclass, field
|
|||
|
|
from datetime import datetime
|
|||
|
|
from enum import Enum
|
|||
|
|
from typing import List, Optional, Dict, Any
|
|||
|
|
|
|||
|
|
# 定义枚举
|
|||
|
|
class PlatformType(Enum):
|
|||
|
|
XIAOHONGSHU = "xiaohongshu"
|
|||
|
|
DOUYIN = "douyin"
|
|||
|
|
|
|||
|
|
# 正确的基类定义 - 所有必需参数在前
|
|||
|
|
@dataclass
|
|||
|
|
class BaseContent:
|
|||
|
|
title: str
|
|||
|
|
description: str
|
|||
|
|
tags: List[str] = field(default_factory=list)
|
|||
|
|
visibility: str = "public"
|
|||
|
|
|
|||
|
|
# 正确的子类定义 - 只有必需参数
|
|||
|
|
@dataclass
|
|||
|
|
class ImageNote(BaseContent):
|
|||
|
|
images: List[str] = field(default_factory=list)
|
|||
|
|
|
|||
|
|
@dataclass
|
|||
|
|
class VideoNote(BaseContent):
|
|||
|
|
video_path: str
|
|||
|
|
images: List[str] = field(default_factory=list)
|
|||
|
|
|
|||
|
|
def test_correct_design():
|
|||
|
|
print('🚀 开始正确的dataclass设计测试')
|
|||
|
|
print('=' * 50)
|
|||
|
|
|
|||
|
|
# 测试枚举
|
|||
|
|
print(f'✅ 平台枚举: {PlatformType.XIAOHONGSHU.value}')
|
|||
|
|
|
|||
|
|
# 测试图文笔记
|
|||
|
|
note = ImageNote(
|
|||
|
|
title="测试笔记标题",
|
|||
|
|
description="测试笔记描述",
|
|||
|
|
images=["img1.jpg", "img2.jpg"],
|
|||
|
|
tags=["测试", "笔记"]
|
|||
|
|
)
|
|||
|
|
print(f'✅ 图文笔记: {note.title}, {len(note.images)}张图片')
|
|||
|
|
|
|||
|
|
# 测试视频内容
|
|||
|
|
video = VideoNote(
|
|||
|
|
title="测试视频标题",
|
|||
|
|
description="测试视频描述",
|
|||
|
|
video_path="test.mp4",
|
|||
|
|
images=["video_cover.jpg"],
|
|||
|
|
tags=["测试", "视频"]
|
|||
|
|
)
|
|||
|
|
print(f'✅ 视频内容: {video.title}, 路径: {video.video_path}')
|
|||
|
|
|
|||
|
|
print('=' * 50)
|
|||
|
|
print('🎉 正确的dataclass设计测试成功！')
|
|||
|
|
print('💡 关键：子类只能添加默认值参数，不能添加必需参数')
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
test_correct_design()
|