31 lines
1.0 KiB
Python
31 lines
1.0 KiB
Python
|
# -*- coding: utf-8 -*-
|
|||
|
|
"""
|
|||
|
|
main.py - 支持扫码登录和 cookie 登录模式选择
|
|||
|
|
"""
|
|||
|
|
import os
|
|||
|
|
|
|||
|
|
def main():
|
|||
|
|
print("请选择登录模式：")
|
|||
|
|
print("1. 扫码登录（首次登录/刷新 cookie）")
|
|||
|
|
print("2. 使用已保存 cookie 自动发文")
|
|||
|
|
choice = input("请输入模式编号（1或2）：").strip()
|
|||
|
|
if choice == '1':
|
|||
|
|
from scan_qrcode import scan_qrcode
|
|||
|
|
scan_qrcode()
|
|||
|
|
elif choice == '2':
|
|||
|
|
cookie_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'cookies.json')
|
|||
|
|
if not os.path.exists(cookie_path):
|
|||
|
|
print("未检测到 cookie 文件，请先扫码登录！")
|
|||
|
|
return
|
|||
|
|
from use_cookie import get_test_content, auto_publish_xhs
|
|||
|
|
title, desc, image_list = get_test_content()
|
|||
|
|
if not image_list:
|
|||
|
|
print('未找到图片文件，请将图片放入test文件夹')
|
|||
|
|
else:
|
|||
|
|
auto_publish_xhs(title, desc, image_list)
|
|||
|
|
else:
|
|||
|
|
print("输入无效，请输入 1 或 2！")
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
main()
|