62 lines
1.9 KiB
Python
62 lines
1.9 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
|
||
from Pixiv import PixivSpider
|
||
import redis_monitor
|
||
import sys
|
||
|
||
def show_main_menu():
|
||
"""显示主菜单并处理用户选择"""
|
||
while True:
|
||
print("\n=== PixivSpider ===")
|
||
print("1. 爬取每日排行榜")
|
||
print("2. Redis数据库操作")
|
||
print("3. 退出程序")
|
||
|
||
try:
|
||
choice = input("\n请选择操作 (1-3): ")
|
||
|
||
if choice == "1":
|
||
print("\n=== 启动PixivSpider ===")
|
||
print("确保已安装并启动Redis服务")
|
||
print("确保已准备好有效的Pixiv Cookie")
|
||
try:
|
||
PixivSpider.pixiv_main()
|
||
except redis.exceptions.ConnectionError:
|
||
print('错误：无法连接到Redis服务，请确保Redis服务正在运行')
|
||
except KeyboardInterrupt:
|
||
print('\n用户中断运行')
|
||
except Exception as e:
|
||
print(f'发生错误：{str(e)}')
|
||
|
||
elif choice == "2":
|
||
print("\n=== 启动Redis管理工具 ===")
|
||
redis_monitor.show_menu()
|
||
|
||
elif choice == "3":
|
||
print("\nbye!")
|
||
sys.exit(0)
|
||
|
||
else:
|
||
print("\n无效的选择，请重试")
|
||
|
||
except KeyboardInterrupt:
|
||
print("\n\n检测到Ctrl+C，正在安全退出...")
|
||
sys.exit(0)
|
||
except Exception as e:
|
||
print(f"\n发生错误：{str(e)}")
|
||
|
||
if __name__ == "__main__":
|
||
try:
|
||
import redis
|
||
import requests
|
||
except ImportError:
|
||
print('检测到缺少必要包！正在尝试安装！.....')
|
||
import os
|
||
os.system(r'pip install -r requirements.txt')
|
||
import redis
|
||
import requests
|
||
|
||
|
||
show_main_menu()
|