From 7c62c8562716b44f8bd5b0df1e8c45b7f8502f36 Mon Sep 17 00:00:00 2001 From: Pin Lin Date: Sat, 14 Apr 2018 17:13:35 +0800 Subject: [PATCH] fix upload answer filename again --- interface.py | 15 +++++++++++---- 1 file changed, 11 insertions(+), 4 deletions(-) diff --git a/interface.py b/interface.py index 9edb820..edebb6a 100644 --- a/interface.py +++ b/interface.py @@ -227,13 +227,20 @@ class Kuser: def send_answer(self, text, file_id): 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 != '': - with open(self._username + self._question + '.c', 'w') as f: + with open(filename, 'w') as f: f.write(text) else: - bot.download_file(file_id, self._username + self._question + '.c') + bot.download_file(file_id, filename) 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, "上傳成功", reply_markup=ReplyKeyboardMarkup(keyboard=[ ["首頁🏠", "回題目📜"], @@ -246,7 +253,7 @@ class Kuser: ["首頁🏠", "回題目📜"], ["登出🚪", "改密碼💱", "幫助📚"] ], resize_keyboard=True)) - os.remove(self._username + self._question + '.c') + os.remove(filename) def delete_answer(self): bot.sendMessage(self._userid, "移除成功" if self._api.delete_answer(self._question) == True else "移除失敗",