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

fix upload answer filename again

Browse Source
This commit is contained in:
Pin Lin
2018-04-14 17:13:35 +08:00
parent b02dd08410
commit 7c62c85627
1 changed files with 11 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
interface.py
View File

@ -227,13 +227,20 @@ class Kuser:
def send_answer(self, text, file_id): def send_answer(self, text, file_id):
self._status = '正常使用' self._status = '正常使用'
# define filename
filename = self._username + self._question
if self._api.list_questions()[self._question][3] == '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 "移除失敗",