reduce connections to kcoj

This commit is contained in:
2017-08-23 15:33:38 +08:00
parent 69151f80aa
commit f94a61e9a8
2 changed files with 12 additions and 7 deletions

View File

@ -27,7 +27,7 @@ class kuser_api:
soup = BeautifulSoup(response.text, 'html.parser') soup = BeautifulSoup(response.text, 'html.parser')
return soup.find('a').get_text().strip() == '線上考試' return soup.find('a').get_text().strip() == '線上考試'
except requests.exceptions.Timeout: except requests.exceptions.Timeout:
return True return None
# list all questions, deadline and hand-in status # list all questions, deadline and hand-in status
def list_questions(self): def list_questions(self):

View File

@ -78,20 +78,25 @@ class kuser:
self.bot.sendMessage(self.chat_id, "哇...登入失敗,讓我們重新開始一次", reply_markup=ReplyKeyboardRemove()) self.bot.sendMessage(self.chat_id, "哇...登入失敗,讓我們重新開始一次", reply_markup=ReplyKeyboardRemove())
self.press_username() self.press_username()
def check_online(self): def check_online(self): # edit
self.status = '正常使用' self.status = '正常使用'
if self.api.login_kcoj(self.username, self.password) == None: result = self.api.check_online()
if result == None:
self.question = '題外' self.question = '題外'
self.bot.sendMessage(self.chat_id, "郭老網站離線中!", self.bot.sendMessage(self.chat_id, "郭老 Online Judge 離線中!",
reply_markup=ReplyKeyboardMarkup(keyboard=[ reply_markup=ReplyKeyboardMarkup(keyboard=[
["主畫面🏠"] ["主畫面🏠"]
], resize_keyboard=True)) ], resize_keyboard=True))
return False return False
elif self.api.check_online() == True: elif result == True:
return True return True
else: else:
self.fail_login() self.api.login_kcoj(self.username, self.password)
return False if self.api.check_online() == False:
self.fail_login()
return False
else:
return True
def logout_system(self): def logout_system(self):
self.status = '正常使用' self.status = '正常使用'