social_media_auto_publisher/advanced_demo.py

#!/usr/bin/env python3
"""
高级演示测试 - 展示项目核心功能
"""

import asyncio
import sys
from pathlib import Path
from datetime import datetime
from dataclasses import dataclass, field
from enum import Enum
from typing import List, Optional, Dict, Any

# 使用我们之前测试过的可用模型定义
class PlatformType(Enum):
    XIAOHONGSHU = "xiaohongshu"
    DOUYIN = "douyin"

@dataclass
class ImageNote:
    title: str
    description: str
    images: List[str] = field(default_factory=list)
    tags: List[str] = field(default_factory=list)
    visibility: str = "public"
    cover_image: Optional[str] = None

@dataclass
class VideoContent:
    title: str
    description: str
    video_path: str
    tags: List[str] = field(default_factory=list)
    visibility: str = "public"
    cover_image: Optional[str] = None

@dataclass
class AccountInfo:
    platform: PlatformType
    username: str
    cookie_file: str
    is_active: bool = True

class SimplePublisher:
    """简化的发布器演示类"""

    def __init__(self):
        self.browser = None
        self.page = None

    async def setup_browser(self, headless=False):
        """设置浏览器"""
        try:
            from playwright.async_api import async_playwright

            self.playwright = await async_playwright().start()
            self.browser = await self.playwright.chromium.launch(
                headless=headless,
                args=[
                    '--no-sandbox',
                    '--disable-blink-features=AutomationControlled'
                ]
            )

            # 设置用户代理
            context = await self.browser.new_context(
                user_agent='Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36'
            )

            self.page = await context.new_page()

            # 注入反检测脚本
            await self.page.add_init_script("""
                Object.defineProperty(navigator, 'webdriver', {
                    get: () => undefined,
                });
            """)

            return True

        except Exception as e:
            print(f"❌ 浏览器设置失败: {e}")
            return False

    async def test_platform_access(self, platform, account_info):
        """测试平台访问（不登录）"""
        print(f"\n🌐 测试{platform.value}平台访问")
        print("-" * 40)

        try:
            if platform == PlatformType.XIAOHONGSHU:
                # 访问小红书创作者中心
                await self.page.goto("https://creator.xiaohongshu.com/", timeout=10000)
                await asyncio.sleep(2)

                # 检查页面是否正常加载
                title = await self.page.title()
                print(f"✅ 页面加载成功: {title}")

                # 检查是否有登录相关元素
                login_elements = await self.page.query_selector_all('[class*="login"], [class*="auth"], button:has-text("登录")')
                if login_elements:
                    print(f"✅ 检测到 {len(login_elements)} 个登录相关元素")

                return True

            elif platform == PlatformType.DOUYIN:
                # 访问抖音创作者中心
                await self.page.goto("https://creator.douyin.com/", timeout=10000)
                await asyncio.sleep(2)

                # 检查页面是否正常加载
                title = await self.page.title()
                print(f"✅ 页面加载成功: {title}")

                # 检查是否有登录相关元素
                login_elements = await self.page.query_selector_all('[class*="login"], [class*="auth"], button:has-text("登录")')
                if login_elements:
                    print(f"✅ 检测到 {len(login_elements)} 个登录相关元素")

                return True

        except Exception as e:
            print(f"❌ 平台访问失败: {e}")
            return False

    async def simulate_content_creation(self, content, platform):
        """模拟内容创建过程（不实际提交）"""
        print(f"\n📝 模拟{platform.value}内容创建")
        print("-" * 40)

        try:
            if platform == PlatformType.XIAOHONGSHU:
                # 模拟访问图文笔记发布页面
                await self.page.goto("https://creator.xiaohongshu.com/publish/publish", timeout=10000)
                await asyncio.sleep(3)

                # 检查是否到达发布页面
                url = self.page.url
                if "publish" in url:
                    print("✅ 成功访问图文笔记发布页面")

                    # 模拟查找标题输入框
                    title_input = await self.page.query_selector('input[placeholder*="标题"], textarea[placeholder*="标题"]')
                    if title_input:
                        print("✅ 找到标题输入框")
                        # 只模拟输入，不实际输入
                        print(f"📝 准备输入标题: {content.title}")

                    # 模拟查找内容输入框
                    content_input = await self.page.query_selector('textarea[placeholder*="内容"], textarea[placeholder*="描述"]')
                    if content_input:
                        print("✅ 找到内容输入框")
                        print(f"📝 准备输入描述: {content.description[:30]}...")

                    # 模拟查找图片上传区域
                    upload_area = await self.page.query_selector('[class*="upload"], [data-testid*="upload"], input[type="file"]')
                    if upload_area:
                        print("✅ 找到图片上传区域")
                        print(f"📷 准备上传 {len(content.images)} 张图片")

                    return True
                else:
                    print("❌ 未能访问发布页面，可能需要登录")
                    return False

            elif platform == PlatformType.DOUYIN:
                # 模拟访问视频上传页面
                await self.page.goto("https://creator.douyin.com/creator-micro/content/upload", timeout=10000)
                await asyncio.sleep(3)

                # 检查页面
                url = self.page.url
                if "upload" in url or "content" in url:
                    print("✅ 成功访问内容上传页面")

                    # 模拟查找标题输入框
                    title_input = await self.page.query_selector('input[placeholder*="标题"]')
                    if title_input:
                        print("✅ 找到标题输入框")
                        print(f"📝 准备输入标题: {content.title}")

                    # 模拟查找描述输入框
                    desc_input = await self.page.query_selector('textarea[placeholder*="描述"]')
                    if desc_input:
                        print("✅ 找到描述输入框")
                        print(f"📝 准备输入描述: {content.description[:30]}...")

                    # 模拟查找视频上传区域
                    video_upload = await self.page.query_selector('[class*="upload"], input[type="file"][accept*="video"]')
                    if video_upload:
                        print("✅ 找到视频上传区域")
                        print(f"🎬 准备上传视频: {content.video_path}")

                    return True
                else:
                    print("❌ 未能访问上传页面，可能需要登录")
                    return False

        except Exception as e:
            print(f"❌ 内容创建模拟失败: {e}")
            return False

    async def cleanup(self):
        """清理资源"""
        if self.page:
            await self.page.close()
        if self.browser:
            await self.browser.close()
        if hasattr(self, 'playwright'):
            await self.playwright.stop()

def create_sample_content():
    """创建示例内容"""
    contents = []

    # 小红书图文笔记
    xhs_note = ImageNote(
        title="今日美食分享 🍜| 超好吃的家常面",
        description="今天做了特别好吃的家常面，汤底浓郁，面条劲道，配上特制的酱料，简直是绝配！详细做法在图片里，大家一定要试试看~\n\n#家常美食 #面条 #美食分享 #简单易学",
        images=["delicious_noodles1.jpg", "delicious_noodles2.jpg", "delicious_noodles3.jpg"],
        tags=["家常美食", "面条", "美食分享", "简单易学", "今日菜单"]
    )
    contents.append((xhs_note, PlatformType.XIAOHONGSHU))

    # 抖音视频
    douyin_video = VideoContent(
        title="10分钟搞定快手早餐 🥪",
        description="分享一个超级简单的快手早餐做法，10分钟就能搞定，营养又美味！上班族和学生党都适合~\n\n#快手早餐 #早餐教程 #简单易学 #美食vlog",
        video_path="quick_breakfast.mp4",
        tags=["快手早餐", "早餐教程", "简单易学", "美食vlog", "生活技巧"]
    )
    contents.append((douyin_video, PlatformType.DOUYIN))

    return contents

async def main():
    """主演示函数"""
    print("🚀 社交媒体自动发布器 - 高级功能演示")
    print("📋 这个演示会访问真实平台页面，但不会发布任何内容")
    print("=" * 70)

    # 创建示例内容和账号
    contents = create_sample_content()

    # 创建测试账号
    test_accounts = {
        PlatformType.XIAOHONGSHU: AccountInfo(
            platform=PlatformType.XIAOHONGSHU,
            username="demo_xhs_user",
            cookie_file="demo_xhs_user.json"
        ),
        PlatformType.DOUYIN: AccountInfo(
            platform=PlatformType.DOUYIN,
            username="demo_douyin_user",
            cookie_file="demo_douyin_user.json"
        )
    }

    # 创建发布器
    publisher = SimplePublisher()

    try:
        # 设置浏览器（显示模式，便于观察）
        print("\n🌐 1. 设置浏览器环境")
        if await publisher.setup_browser(headless=False):
            print("✅ 浏览器设置成功")
        else:
            print("❌ 浏览器设置失败")
            return False

        # 测试每个平台
        for content, platform in contents:
            print(f"\n{'='*70}")
            print(f"📱 测试平台: {platform.value}")
            print(f"📝 内容: {content.title}")
            print(f"{'='*70}")

            # 1. 测试平台访问
            if await publisher.test_platform_access(platform, test_accounts[platform]):

                # 2. 模拟内容创建
                await publisher.simulate_content_creation(content, platform)

                # 3. 显示内容预览
                print(f"\n📋 内容预览:")
                print(f"   标题: {content.title}")
                print(f"   描述: {content.description[:100]}...")
                print(f"   标签: {', '.join(content.tags)}")

                if isinstance(content, ImageNote):
                    print(f"   图片: {len(content.images)} 张")
                elif isinstance(content, VideoContent):
                    print(f"   视频: {content.video_path}")

                print(f"\n⏳ 等待3秒后测试下一个平台...")
                await asyncio.sleep(3)

            else:
                print(f"❌ {platform.value}平台访问失败")

        print(f"\n{'='*70}")
        print("🎉 演示测试完成！")
        print("\n📝 演示总结:")
        print("✅ 浏览器环境设置成功")
        print("✅ 平台页面访问测试")
        print("✅ 内容创建流程模拟")
        print("✅ 发布界面元素检测")

        print(f"\n📋 实际使用时:")
        print("1. 准备真实的图片和视频文件")
        print("2. 运行完整的登录流程（扫码登录）")
        print("3. 使用真实的账号信息")
        print("4. 执行实际的内容发布")

        return True

    except Exception as e:
        print(f"\n❌ 演示过程中出现异常: {e}")
        import traceback
        traceback.print_exc()
        return False

    finally:
        # 清理资源
        await publisher.cleanup()

if __name__ == "__main__":
    print("⚠️  注意: 这个演示会打开浏览器窗口，访问真实网站页面")
    print("⚠️  但不会进行任何登录操作或内容发布")
    print("⚠️  请在30秒内观察浏览器行为，或按Ctrl+C退出")
    print("\n按回车键继续演示，或按Ctrl+C退出...")
    try:
        input()
        result = asyncio.run(main())
        sys.exit(0 if result else 1)
    except KeyboardInterrupt:
        print("\n⏹️ 用户中断演示")
        sys.exit(0)