import asyncio
from auto_media_publisher.utils.login_helper import create_and_run_login
import sys
        
async def main():
    if len(sys.argv) < 3:
        print("python test_login.py [USERNAME] [PLATFORM]")
        print("douyin, kuaishou, shipinhao, xiaohongshu, toutiaohao, bilibili, baijiahao, tiktok, twitterX, all")
        return
    proxy_url = sys.argv[3] if len(sys.argv) > 3 else None
    await create_and_run_login(user=sys.argv[1],platform=sys.argv[2],proxy=proxy_url)

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