-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathWordPress评论监控.py
175 lines (156 loc) · 7.28 KB
/
WordPress评论监控.py
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
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
import asyncio
import discord
from datetime import datetime
import logging
import requests
import os
import time
import platform
import zipfile
import json
import aiohttp
from discord.ext import tasks
LogFileName = ""
LogPath = ""
WorkDirectory = ""
class DiscordConfig(object):
def __init__(self):
self.Ignore_List = []
self.__Bot_Token = ""
self.URL = ""
self.Admin_User_ID = 0
self.Channel_ID = 0
self.Channel_Message = True
def From_Dict(self, Dict: dict):
self.Ignore_List = Dict["Ignore_List"]
self.__Bot_Token = Dict["_DiscordConfig__Bot_Token"]
self.URL = Dict["URL"]
self.Admin_User_ID = Dict["Admin_User_ID"]
self.Channel_ID = Dict["Channel_ID"]
self.Channel_Message = Dict["Channel_Message"]
return self
def Get_Token(self):
return self.__Bot_Token
def Set_Token(self, Token: str):
self.__Bot_Token = Token
class DiscordClient(discord.Client):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.Message = None
self.Config = None
# an attribute we can access from our task
async def on_ready(self):
logging.info(f'以{BotClient.user}登录')
@tasks.loop(minutes=1)
async def TimerTrigger(self) -> None:
logging.info(f'现在是{datetime.now().replace().strftime("%Y-%m-%d %H:%M:%S")}，开始检测评论状态。')
Compress_LogFile()
response = None
async with aiohttp.ClientSession() as session:
async with session.get('https://www.thewhitedog9487.xyz/wp-json/wp/v2/comments') as response:
response = await response.json()
for comment in response:
RemoteCommentDate = int(time.mktime(time.strptime(comment["date"], '%Y-%m-%dT%H:%M:%S')))
with open('最新评论发布时间.txt', 'r', encoding="utf-8") as r:
FileCommentDate = r.readline().replace('\n', '')
FileCommentDate = int(time.mktime(time.strptime(FileCommentDate, '%Y年%m月%d日 %H时%M分%S秒')))
if int(FileCommentDate) < RemoteCommentDate:
logging.info('有新评论')
with open('最新评论发布时间.txt', 'w', encoding="utf-8") as w:
w.write(time.strftime('%Y年%m月%d日 %H时%M分%S秒', time.localtime(RemoteCommentDate)))
self.Message = f'评论者：{comment["author_name"]}\n评论内容：{comment["content"]["rendered"]}评论链接：{comment["link"]}\n评论时间：{time.strftime("%Y年%m月%d日 %H时%M分%S秒", time.localtime(RemoteCommentDate))}'
for user in self.Config.Ignore_List:
if comment["author_name"] == user:
logging.info("由于发送评论的用户位于忽略列表当中，因此本次新评论不会发送提醒。")
logging.info(self.Message)
return
if self.Config.Channel_Message:
channel = self.get_channel(self.Config.Channel_ID)
await channel.send(f"<@{self.Config.Admin_User_ID}>\n{self.Message}")
else:
user = await self.fetch_user(self.Config.Admin_User_ID)
await user.send(self.Message)
logging.info('已发送新通知\n')
elif int(FileCommentDate) > RemoteCommentDate:
logging.info('有评论被删除\n')
elif int(FileCommentDate) == RemoteCommentDate:
logging.info('没有新评论\n')
break
async def setup_hook(self) -> None:
self.TimerTrigger.start()
@TimerTrigger.before_loop
async def Before_Loop(self):
await self.wait_until_ready() # wait until the bot logs in
def Compress_LogFile():
global LogFileName
global LogPath
global WorkDirectory
if os.path.exists(LogFileName):
if os.path.getsize(LogFileName) > 1024 * 1024 * 1: # 1MB
os.chdir(LogPath)
with zipfile.ZipFile(datetime.now().replace().strftime("%Y-%m-%d %H:%M:%S") + ".zip", 'w',
zipfile.ZIP_DEFLATED) as zf:
zf.write("WordPress评论监控.log")
os.remove(LogFileName)
os.chdir(WorkDirectory)
def Initialization():
global LogFileName
global LogPath
global WorkDirectory
WorkDirectory = os.getcwd()
if platform.system() == 'Windows':
LogPath = os.curdir + "\\日志\\"
elif platform.system() == 'Linux':
LogPath = "/var/log/thewhitedog9487/"
else:
print("不认识的操作系统，请联系开发者寻求适配。")
exit()
LogFileName = LogPath + "WordPress评论监控.log"
if not os.path.exists(LogPath):
os.makedirs(LogPath)
Compress_LogFile()
logging.basicConfig(filename=LogFileName, level=logging.INFO, encoding="utf-8")
logging.info(f'程序从{datetime.now().replace().strftime("%Y-%m-%d %H:%M:%S")}开始运行')
if (not os.path.exists("配置文件.json")) or (os.path.exists("配置文件.json") and (os.path.getsize("配置文件.json") == 0)):
with open("配置文件.json", "w", encoding="utf-8") as w:
json.dump(DiscordConfig().__dict__, w, indent=4)
logging.info('首次运行')
logging.error('配置文件模板已生成，请填写配置文件')
exit()
with open("配置文件.json", "r", encoding="utf-8") as r:
Config = DiscordConfig().From_Dict(json.load(r))
if Config.Get_Token() == "":
logging.error('请填写机器人令牌')
logging.error('这是必填项，如果为空程序将不能运行')
exit()
if Config.URL == "":
logging.error('请填写要监测博客的地址')
logging.error('这是必填项，如果为空程序将不能运行')
exit()
if Config.Channel_Message:
logging.debug('程序在频道中运行')
if Config.Channel_ID == "":
logging.error('请填写频道ID')
logging.error('这是必填项，如果为空程序将不能运行')
exit()
else:
logging.debug('程序在私信中运行')
if Config.Admin_User_ID == "":
logging.error('请填写Discord管理员用户ID')
logging.error('这是必填项，如果为空程序将不能运行')
exit()
if not os.path.exists('最新评论发布时间.txt') or os.path.getsize('最新评论发布时间.txt') == 0:
if not str(Config.URL).startswith(("http://", "https://")):
logging.error("监测链接必须以http://或者https://打头！")
exit()
with open('最新评论发布时间.txt', 'w', encoding="utf-8") as w:
w.write(datetime.strptime(
requests.get(Config.URL + '/wp-json/wp/v2/comments').json()[0]["date"],
'%Y-%m-%dT%H:%M:%S').strftime('%Y年%m月%d日 %H时%M分%S秒'))
return Config
if __name__ == '__main__':
intent = discord.Intents.default()
intent.messages = True
BotClient = DiscordClient(intents=intent)
BotClient.Config = Initialization()
BotClient.run(BotClient.Config.Get_Token())