PinLin/KCOJ_bot
Archived
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix upload answer filename

Browse Source
This commit is contained in:
Pin Lin
2018-04-14 16:55:27 +08:00
parent f1420f89c8
commit 0b8980b763
2 changed files with 14 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
bot.py
View File

@ -128,7 +128,7 @@ def on_chat(msg):
elif user._status == '上傳答案': elif user._status == '上傳答案':
if user.check_online(chat_id, msg['message_id']) == True: if user.check_online(chat_id, msg['message_id']) == True:
user.send_answer(msg['text'], '') user.send_answer(msg['text'], '', user.list_questions(chat_id)[user._question][3])
else: else:
if user.check_online(chat_id, msg['message_id']) == True: if user.check_online(chat_id, msg['message_id']) == True:
@ -140,7 +140,7 @@ def on_chat(msg):
if msg['document']['file_size'] > 167770000: if msg['document']['file_size'] > 167770000:
user.send_failed() user.send_failed()
else: else:
user.send_answer('', msg['document']['file_id']) user.send_answer('', msg['document']['file_id'], user.list_questions(chat_id)[user._question][3])
def backup_db(): def backup_db():
users_backup = {} users_backup = {}

17
interface.py
View File

@ -225,15 +225,22 @@ class Kuser:
["首頁🏠", "回題目📜"] ["首頁🏠", "回題目📜"]
], resize_keyboard=True)) ], resize_keyboard=True))
def send_answer(self, text, file_id): def send_answer(self, text, file_id, language):
self._status = '正常使用' self._status = '正常使用'
# define filename
filename = self._username + self._question
if language == 'Python':
filename += '.py'
else:
filename += '.c'
if text != '': if text != '':
with open(self._username + self._question + '.c', 'w') as f: with open(filename, 'w') as f:
f.write(text) f.write(text)
else: else:
bot.download_file(file_id, self._username + self._question + '.c') bot.download_file(file_id, filename)
self._api.delete_answer(self._question) self._api.delete_answer(self._question)
if self._api.upload_answer(self._question, self._username + self._question + '.c') == True: if self._api.upload_answer(self._question, filename) == True:
bot.sendMessage(self._userid, "上傳成功", bot.sendMessage(self._userid, "上傳成功",
reply_markup=ReplyKeyboardMarkup(keyboard=[ reply_markup=ReplyKeyboardMarkup(keyboard=[
["首頁🏠", "回題目📜"], ["首頁🏠", "回題目📜"],
@ -246,7 +253,7 @@ class Kuser:
["首頁🏠", "回題目📜"], ["首頁🏠", "回題目📜"],
["登出🚪", "改密碼💱", "幫助📚"] ["登出🚪", "改密碼💱", "幫助📚"]
], resize_keyboard=True)) ], resize_keyboard=True))
os.remove(self._username + self._question + '.c') os.remove(filename)
def delete_answer(self): def delete_answer(self):
bot.sendMessage(self._userid, "移除成功" if self._api.delete_answer(self._question) == True else "移除失敗", bot.sendMessage(self._userid, "移除成功" if self._api.delete_answer(self._question) == True else "移除失敗",