-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathdiscord_bot.py
225 lines (186 loc) · 9.57 KB
/
discord_bot.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
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
#!/user/bin/env python
# -*- coding: utf-8 -*-
import logging #可以測試哪裡有問題
import discord
import json
import re
from covid_info_bot import LokiResult, runLoki
import datetime
import vaccine_stock_api
from pprint import pprint
logging.basicConfig(level=logging.CRITICAL)
# <取得多輪對話資訊>
client = discord.Client()
# sideEffectTemplate ={
# "vaccine_shot":"",
# "side_effect": "",
# "side_effect_var":""
# }
# severeSideEffectTemplate = {
# "vaccine_shot" : "",
# "severe_side_effect" : ""
# }
# vaccineStockTemplate = {
# "vaccine_shot":"",
# "location":"",
# "vaccine_stock":""
# }
# 將全部意圖合為一個Template處理，
# 但不一定要每個intent都必須滿足才能結束對話，
# 而是當confirm = True時，就可以結束對話
allTemplate = {
"inquiry_type" : [],
"vaccine_shot" : [],
"side_effect": [],
"severe_side_effect" : [],
"location" : [],
"vaccine_stock" : [],
"group_num" : [],
"group_num_def" : [],
"response": "",
"followup": [],
"updatetime": datetime.datetime.now(), #新增datetime
"completed" : False
}
mscDICT = {
# "userID": {side_effectTemplate, vaccine_stockTemplate, severeSideEffectTemplate}
}
# </取得多輪對話資訊>
with open("account.info", encoding="utf-8") as f:
accountDICT = json.loads(f.read())
# 另一個寫法是：accountDICT = json.load(open("account.info", encoding="utf-8"))
def getLokiResult(inputSTR):
punctuationPat = re.compile("[,\.\?:;，。？、：；\n]+")
inputLIST = punctuationPat.sub("\n", inputSTR).split("\n")
filterLIST = []
resultDICT = runLoki(inputLIST, filterLIST)
print("Loki Result => {}".format(resultDICT))
return resultDICT,
@client.event #成功連線到discord的回答
async def on_ready():
logging.info("[READY INFO] {} has connected to Discord!".format(client.user))
print("[READY INFO] {} has connected to Discord!".format(client.user))
@client.event
async def on_message(message):
# if message.channel.name != "dt_intern":
# return
if not re.search("<@[!&]{}> ?".format(message.author.id), message.content): # 只有 @Bot 才會回應
return
if message.author == client.user:
return
# try:
print("message.author.id =", message.author.id, "\nmessage.content =", message.content)
msgSTR = re.sub("<@[!&]{}> ?".format(message.author.id), "", message.content) # 收到 User 的訊息，將 id 取代成 ""
print("msgSTR =", msgSTR)
replySTR = "" # Bot 回應訊息
if re.search("(hi|hello|哈囉|嗨|[你您]好)", msgSTR.lower()):
replySTR = "Hi 您好，想知道哪些疫苗資訊呢?"
await message.reply(replySTR)
return
lokiResultDICT = getLokiResult(msgSTR) # 取得 Loki 回傳結果
logging.info(lokiResultDICT)
if lokiResultDICT:
print(lokiResultDICT)
if message.author.id not in mscDICT: # 判斷 User 是否為第一輪對話
# mscDICT[message.author.id] = {
# "inquiry_type" : "",
# "vaccine_shot" : "",
# "side_effect": "",
# "severe_side_effect" : "",
# "location" : "",
# "vaccine_stock" : "",
# "group_num" : "",
# "group_def" : "",
# "response": "",
# "followup": [],
# "completed" : False
# }
mscDICT[message.author.id] = allTemplate
for k in lokiResultDICT.keys(): #注意!
mscDICT[message.author.id][k] = lokiResultDICT[k]
mscDICT[message.author.id]["followup"] = {}
else:
datetimeNow = datetime.datetime.now()
timeDIFF = datetimeNow - mscDICT[message.author.id]["updatetime"]
for k in lokiResultDICT.keys(): #注意!
mscDICT[message.author.id][k] = lokiResultDICT[k]
if timeDIFF.total_seconds() >= 300: # 回答時間超過五分鐘，對話內容重置
# mscDICT[message.author.id]= {
# "inquiry_type" : "",
# "vaccine_shot" : "",
# "side_effect": "",
# "severe_side_effect" : "",
# "location" : "",
# "vaccine_stock" : "",
# "group_num" : "", #未處理
# "group_def" : "", #未處理
# "response": "",
# "followup": [],
# "completed" : False
# }
mscDICT[message.author.id] = allTemplate
for k in lokiResultDICT.keys(): #注意!
mscDICT[message.author.id][k] = lokiResultDICT[k]
mscDICT[message.author.id]["followup"] = {}
# if confirm == False : 確認不完整資訊
# elif confirm == True : 問還要不要繼續問資訊
if lokiResultDICT:
# if mscDICT[message.author.id]["inquiry_type"] == "" and replySTR == "":
# replySTR = "\n請問要問關於疫苗的甚麼資訊呢？"
if mscDICT[message.author.id]["completed"] == False:
if mscDICT[message.author.id]["followup"] != []:
followupDICT = runLoki(mscDICT[message.author.id]["followup"], ["Probe", "confirm_check"])
for k in followupDICT.keys():
mscDICT[message.author.id][k] = followupDICT[k]
mscDICT[message.author.id]["followup"] = []
if mscDICT[message.author.id]["response"]:
replySTR = mscDICT[message.author.id]["response"]
if mscDICT[message.author.id]["inquiry_type"] == [] and replySTR == "":
replySTR = "\n請問要問關於疫苗的甚麼資訊呢？"
elif mscDICT[message.author.id]["inquiry_type"] == [] and mscDICT[message.author.id]["vaccine_shot"] != [] and replySTR == "":
replySTR = "你想知道關於這支疫苗的甚麼事情?"
if mscDICT[message.author.id]["inquiry_type"] == "side_effect" and mscDICT[message.author.id]["vaccine_shot"] == []:
replySTR = "你想要詢問哪隻疫苗的副作用?"
if mscDICT[message.author.id]["inquiry_type"] == "severe_side_effect" and mscDICT[message.author.id]["vaccine_shot"] == []:
replySTR = "你想要詢問哪隻疫苗的嚴重副作用?"
if mscDICT[message.author.id]["inquiry_type"] == "vaccine_stock" and mscDICT[message.author.id]["location"] == []:
replySTR = "請問您要詢問哪個地區的疫苗庫存呢?"
if mscDICT[message.author.id]["inquiry_type"] == "vaccine_stock" and mscDICT[message.author.id]["vaccine_shot"] == []:
replySTR = "請問您想知道哪個廠牌的疫苗庫存呢?"
if mscDICT[message.author.id]["inquiry_type"] == "vaccine_stock" and mscDICT[message.author.id]["location"] == [] and mscDICT[message.author.id]["vaccine_shot"] == []:
replySTR = "請問您想知道哪個廠牌以及哪個地區的疫苗庫存呢?"
if mscDICT[message.author.id]["inquirt_type"] == "group" and replySTR == "":
replySTR += """{}族群為{}。\n""".format("".join(mscDICT[message.author.id]["group_num"]), "".join(mscDICT[message.author.id]["group_num_def"]))
await message.reply(replySTR)
replySTR = "還想問其他的嗎?"
del mscDICT[message.author.id]
if "side_effect" in mscDICT[message.author.id]["inquiry_type"] and mscDICT[message.author.id]["vaccine_shot"] != []:
replySTR += """{}疫苗的常見副作用是{}。\n""".format("".join(mscDICT[message.author.id]["vaccine_shot"]), "".join(mscDICT[message.author.id]["side_effect"]))
await message.reply(replySTR)
replySTR = "還想問其他的嗎?"
del mscDICT[message.author.id]
if "severe_side_effect" in mscDICT[message.author.id]["inquiry_type"] and mscDICT[message.author.id]["vaccine_shot"] != []:
replySTR += """{}疫苗的常見副作用是{}。\n""".format("".join(mscDICT[message.author.id]["vaccine_shot"]), "".join(mscDICT[message.author.id]["severe_side_effect"]))
await message.reply(replySTR)
replySTR = "還想問其他的嗎?"
del mscDICT[message.author.id]
if "vaccine_stock" in mscDICT[message.author.id]["inquiry_type"] and mscDICT[message.author.id]["location"] != [] and mscDICT[message.author.id]["vaccine_shot"] != []:
replySTR = vaccine_stock_api.write_response(mscDICT[message.author.id])
await message.reply(replySTR)
replySTR = "還想問其他的嗎?"
del mscDICT[message.author.id]
if replySTR == "":
print("看到我就是種錯誤囉!")
else:
del mscDICT[message.author.id]
replySTR = "對話結束囉! 謝謝你使用Covid_Info_Bot! 請務必給我們五個星喔XDD"
print("mscDICT =")
pprint(mscDICT)
if replySTR: # 回應 User 訊息
await message.reply(replySTR)
return
#except Exception as e:
# logging.error("[MSG ERROR] {}".format(str(e)))
# print("[MSG ERROR] {}".format(str(e)))
if __name__ == "__main__":
client.run(accountDICT["discord_token"])