from auto_media_publisher.task.task_generator import UserManager


manager = UserManager()
manager.add_user("user02", city="上海", proxy="http://127.0.0.1:7890")
manager.add_user_platform("user02", "kuaishou")
manager.add_user_platform("user02", "xiaohongshu")
#manager.remove_user_platform("user01", "douyin")
manager.enable_user("user01")

print(manager.get_user_platforms("user01"))
print(manager.get_enabled_users())