PixivSpider/main.py

62 lines
1.9 KiB
Python
Raw Normal View History

2024-12-20 12:35:06 +08:00
#!/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()