UESTC_BBS_Reptile/main.py at main · chenxvb/UESTC_BBS_Reptile · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
from src.myjson import json_read, json_write
import src.hepan as hp
import src.weibang as wb
import src.report as rp
import time
import os
import json

keywords = []

tample_config = {
    "keywords": [],
    "FangTang_APIkey": "",
    "HePan_Cookies": "",
    "WeiBang_Cookies": "",
    "telegram_cookie": "",
    "telegram_chatid": "",
    "telegram_url": ""

}

def report(data:str, func):

    if func == hp:
        rp.sendMsg("河畔更新：" + data)

    if func == wb:
        rp.sendMsg("微帮更新：" + data)

def keywordCheck(data: list, func):
    tmp = data
    data = func.getRankStr()

    if func == hp:
        json_write("./data/hepanRank.json", data)

    if func == wb:
        json_write("./data/weibangRank.json", data)

    if tmp != data:  # 检查是否更新
        # print(data)
        # print(tmp)
        if func == hp:
            print("\n河畔更新：", data[0])
        if func == wb:
            print("\n微帮更新：", data[0])

        if keywords == []:
            report(data[0], func)

        for i in keywords:
            if i in data[0]:
                report(data[0], func)
                # rp.xiaoWei(data[0])

    return data

if __name__ == "__main__":

    hepanFlag = True
    weibangFlag = True

    if "config.json" not in os.listdir("./"):
        print("[+] 未检测到配置文件，请生成后填入信息")
        json_write("./config.json", tample_config)
        exit()

    with open("./config.json", 'r', encoding="utf-8") as f:
        tmp = json.load(f)
        keywords = tmp["keywords"]
        if tmp["HePan_Cookies"] == '':
            hepanFlag = False
            print("未输入河畔 Cookie, 不启动河畔检索")
        if tmp["WeiBang_Cookies"] == '':
            weibangFlag = False
            print("未输入微帮 Cookie, 不启动微帮检索")


    if weibangFlag:
        print("检查微帮 json")
        if os.path.exists("./data/weibangRank.json"):
            print("读取微帮 json")
            rankWeibang = json_read("./data/weibangRank.json")
        else:
            print("微帮 json 不存在")
            rankWeibang = []


    if hepanFlag:
        print("检查河畔 json")
        if os.path.exists("./data/hepanRank.json"):
            print("读取河畔 json")
            rankHepan = json_read("./data/hepanRank.json")
        else:
            print("河畔 json 不存在")
            rankHepan = []

    print("\n爬虫启动！\n")

    while (True):
        try:
            if hepanFlag:
                rankHepan = keywordCheck(rankHepan, hp)
            if weibangFlag:
                rankWeibang = keywordCheck(rankWeibang, wb)

            time.sleep(60)

        except KeyboardInterrupt:
            print("用户中断 KeyboardInterrupt，保存中...")

            if hepanFlag:
                print("Saving: hepanRank.json")
                json_write("./data/hepanRank.json", rankHepan)

            if weibangFlag:
                print("Saving: weibangRank.json")
                json_write("./data/weibangRank.json", rankWeibang)

            print("Over....")
            exit(0)

        except Exception as error:
            print("error: " + error)

            if hepanFlag:
                print("Saving: hepanRank.json")
                json_write("hepanRank.json", rankHepan)

            if weibangFlag:
                print("Saving: weibangRank.json")
                json_write("weibangRank.json", rankWeibang)

            print("Over....")
            exit(0)