social_media_auto_publisher/examples/douyin_examples.py

"""
抖音发布示例
演示抖音平台的各种发布场景。
"""
import asyncio
import sys
from pathlib import Path

# 添加项目根目录到Python路径
project_root = Path(__file__).parent.parent
sys.path.insert(0, str(project_root))

from core.publisher import Publisher
from core.models import VideoContent, PlatformType, AccountInfo
from utils.logger import setup_logger, get_logger

# 设置日志
setup_logger(level="INFO")
logger = get_logger(__name__)


async def douyin_video_publish_example():
    """抖音视频发布示例"""
    print("=" * 60)
    print("抖音视频发布示例")
    print("=" * 60)

    async with Publisher(headless=False) as publisher:
        # 设置账号
        account_name = "your_douyin_account"  # 替换为实际账号名
        print(f"1. 设置抖音账号: {account_name}")

        success = await publisher.setup_platform("douyin", account_name)
        if not success:
            print("❌ 账号设置失败，请检查网络和账号信息")
            return

        print("✅ 账号设置成功")

        # 创建多种类型的视频内容
        videos = [
            # 娱乐搞笑视频
            VideoContent(
                title="😂 今日最佳搞笑时刻",
                description="收集了一些超级搞笑的瞬间，保证让你笑出声！记得点赞关注哦~ #搞笑 #日常 #娱乐",
                video_path="media/videos/funny_moments.mp4",  # 替换为实际视频路径
                tags=["搞笑", "日常", "娱乐", "短视频"]
            ),

            # 生活技巧视频
            VideoContent(
                title="🔥 超实用生活技巧分享",
                description="分享5个超级实用的生活小技巧，让你的生活更加便利。学会了记得分享给朋友哦！#生活技巧 #实用 #分享",
                video_path="media/videos/life_hacks.mp4",  # 替换为实际视频路径
                tags=["生活技巧", "实用", "分享", "教程"]
            ),

            # 美食制作视频
            VideoContent(
                title="🍜 10分钟搞定一碗美味面条",
                description="忙碌工作日也能吃到美食！今天分享一个超简单的面条做法，10分钟就能搞定，味道还不错哦！#美食 #教程 #快手菜",
                video_path="media/videos/quick_noodles.mp4",  # 替换为实际视频路径
                tags=["美食", "教程", "快手菜", "面条"]
            ),

            # 健身运动视频
            VideoContent(
                title="💪 居家健身5分钟燃脂训练",
                description="不需要任何器械，在家就能做的燃脂训练！每天坚持5分钟，让你拥有好身材。#健身 #燃脂 #运动",
                video_path="media/videos/home_workout.mp4",  # 替换为实际视频路径
                tags=["健身", "燃脂", "运动", "居家"]
            )
        ]

        # 发布视频
        for i, video in enumerate(videos, 1):
            print(f"\n{i}. 发布视频: {video.title}")
            try:
                result = await publisher.publish("douyin", video, account_name)

                if result.success:
                    print(f"✅ 发布成功！")
                    print(f"   任务ID: {result.task_id}")
                    print(f"   耗时: {result.duration:.2f}秒")
                    if result.published_url:
                        print(f"   链接: {result.published_url}")
                else:
                    print(f"❌ 发布失败: {result.message}")
                    if result.error_details:
                        print(f"   错误详情: {result.error_details}")

            except Exception as e:
                print(f"❌ 发布异常: {e}")

            # 发布间隔（避免频繁操作）
            if i < len(videos):
                wait_time = 15 + (i * 5)  # 递增等待时间
                print(f"⏳ 等待{wait_time}秒后发布下一个视频...")
                await asyncio.sleep(wait_time)


async def douyin_batch_upload_example():
    """抖音批量上传示例"""
    print("=" * 60)
    print("抖音批量上传示例")
    print("=" * 60)

    # 批量准备视频内容
    video_contents = [
        VideoContent(
            title=f"批量测试视频 {i+1}",
            description=f"这是第{i+1}个批量上传的测试视频，内容为{i+1}。",
            video_path=f"media/videos/batch_test_{i+1}.mp4",  # 替换为实际视频路径
            tags=[f"批量测试{i+1}", "自动化", "测试"]
        )
        for i in range(3)
    ]

    async with Publisher(headless=False) as publisher:
        # 设置账号
        account_name = "your_douyin_account"  # 替换为实际账号名
        success = await publisher.setup_platform("douyin", account_name)

        if not success:
            print("❌ 账号设置失败")
            return

        print("✅ 账号设置成功")
        print(f"开始批量上传 {len(video_contents)} 个视频...")

        # 创建发布任务
        from core.models import PublishTask
        tasks = []
        for i, content in enumerate(video_contents):
            task = PublishTask(
                id=f"douyin_batch_{i+1}",
                platform=PlatformType.DOUYIN,
                account=AccountInfo(
                    platform=PlatformType.DOUYIN,
                    username=account_name,
                    cookie_file=f"{account_name}.json"
                ),
                content=content,
                max_retries=2  # 设置重试次数
            )
            tasks.append(task)

        # 执行批量上传
        try:
            results = await publisher.batch_publish(tasks)

            # 统计结果
            success_count = sum(1 for r in results if r.success)
            print(f"\n📊 批量上传完成:")
            print(f"   成功: {success_count}/{len(results)}")
            print(f"   失败: {len(results) - success_count}/{len(results)}")

            # 详细结果
            for result in results:
                status = "✅" if result.success else "❌"
                duration = f" ({result.duration:.1f}s)" if result.duration else ""
                print(f"   {status} {result.task_id}: {result.message}{duration}")

        except Exception as e:
            print(f"❌ 批量上传异常: {e}")


async def douyin_upload_with_progress_example():
    """抖音上传进度监控示例"""
    print("=" * 60)
    print("抖音上传进度监控示例")
    print("=" * 60)

    async with Publisher(headless=False) as publisher:
        # 设置账号
        account_name = "your_douyin_account"  # 替换为实际账号名
        success = await publisher.setup_platform("douyin", account_name)

        if not success:
            print("❌ 账号设置失败")
            return

        print("✅ 账号设置成功")

        # 创建一个较大的视频文件用于测试进度监控
        video = VideoContent(
            title="上传进度测试视频 📊",
            description="这是一个用于测试上传进度监控功能的视频，文件比较大，可以清楚地看到上传进度。",
            video_path="media/videos/large_video.mp4",  # 替换为实际的大视频路径
            tags=["测试", "上传", "进度监控"]
        )

        print(f"\n开始上传视频: {video.title}")
        print("注意：这会打开浏览器窗口，你可以看到实际的上传过程")

        try:
            result = await publisher.publish("douyin", video, account_name)

            if result.success:
                print(f"✅ 视频上传成功！")
                print(f"   任务ID: {result.task_id}")
                print(f"   总耗时: {result.duration:.2f}秒")
            else:
                print(f"❌ 视频上传失败: {result.message}")

        except Exception as e:
            print(f"❌ 上传异常: {e}")


async def douyin_content_validation_example():
    """抖音内容验证示例"""
    print("=" * 60)
    print("抖音内容验证示例")
    print("=" * 60)

    async with Publisher() as publisher:
        # 测试各种内容验证场景
        test_cases = [
            # 有效内容
            {
                "name": "有效的视频内容",
                "content": VideoContent(
                    title="有效标题",
                    description="有效描述",
                    video_path="test.mp4"
                ),
                "expected": True
            },

            # 无效内容 - 空标题
            {
                "name": "空标题",
                "content": VideoContent(
                    title="",
                    description="有效描述",
                    video_path="test.mp4"
                ),
                "expected": False
            },

            # 无效内容 - 空描述
            {
                "name": "空描述",
                "content": VideoContent(
                    title="有效标题",
                    description="",
                    video_path="test.mp4"
                ),
                "expected": False
            },

            # 无效内容 - 不支持的格式
            {
                "name": "不支持的文件格式",
                "content": VideoContent(
                    title="有效标题",
                    description="有效描述",
                    video_path="test.avi"  # 假设不支持avi格式
                ),
                "expected": False
            },

            # 无效内容 - 视频时长过长
            {
                "name": "视频时长过长",
                "content": VideoContent(
                    title="有效标题",
                    description="有效描述",
                    video_path="test.mp4",
                    duration=1200  # 20分钟，假设平台限制为10分钟
                ),
                "expected": False
            }
        ]

        print("开始内容验证测试...")
        for i, test_case in enumerate(test_cases, 1):
            print(f"\n{i}. 测试: {test_case['name']}")

            try:
                is_valid, error_msg = await publisher.validate_content(
                    "douyin",
                    test_case['content']
                )

                if is_valid == test_case['expected']:
                    print(f"   ✅ 验证结果符合预期")
                else:
                    print(f"   ❌ 验证结果不符合预期")
                    print(f"      期望: {test_case['expected']}")
                    print(f"      实际: {is_valid}")

                if not is_valid:
                    print(f"   错误信息: {error_msg}")

            except Exception as e:
                print(f"   ❌ 验证异常: {e}")


async def douyin_quick_publish_example():
    """抖音快速发布示例"""
    print("=" * 60)
    print("抖音快速发布示例")
    print("=" * 60)

    try:
        from core.publisher import publish_to_douyin

        # 快速发布单个视频
        print("1. 使用快速发布功能...")
        result = await publish_to_douyin(
            title="🎬 快速发布测试视频",
            description="这是使用快速发布功能上传的测试视频，简单方便！#快速发布 #测试 #自动化",
            video_path="media/videos/quick_test.mp4",  # 替换为实际视频路径
            tags=["快速发布", "测试", "自动化"],
            account_name="your_douyin_account",  # 替换为实际账号名
            headless=False
        )

        if result.success:
            print(f"✅ 快速发布成功！")
            print(f"   任务ID: {result.task_id}")
            print(f"   耗时: {result.duration:.2f}秒")
        else:
            print(f"❌ 快速发布失败: {result.message}")

    except Exception as e:
        print(f"❌ 快速发布异常: {e}")


async def main():
    """主函数"""
    print("🚀 抖音发布示例")
    print("=" * 60)
    print("请确保:")
    print("1. 替换示例中的账号名称")
    print("2. 准备好视频文件")
    print("3. 确保网络连接正常")
    print("4. 视频文件格式符合要求（mp4、mov等）")
    print("=" * 60)

    try:
        # 运行示例
        print("选择要运行的示例:")
        print("1. 视频发布")
        print("2. 批量上传")
        print("3. 上传进度监控")
        print("4. 内容验证")
        print("5. 快速发布")

        choice = input("\n请输入选择 (1-5): ").strip()

        if choice == "1":
            await douyin_video_publish_example()
        elif choice == "2":
            await douyin_batch_upload_example()
        elif choice == "3":
            await douyin_upload_with_progress_example()
        elif choice == "4":
            await douyin_content_validation_example()
        elif choice == "5":
            await douyin_quick_publish_example()
        else:
            print("❌ 无效选择")

    except KeyboardInterrupt:
        print("\n⏹️ 用户中断")
    except Exception as e:
        print(f"\n❌ 运行异常: {e}")
        import traceback
        traceback.print_exc()


if __name__ == "__main__":
    asyncio.run(main())