116 lines
3.8 KiB
Python
116 lines
3.8 KiB
Python
#! /usr/bin/env python3
|
|
|
|
# necessary modules
|
|
import time, requests, telepot
|
|
from telepot.loop import MessageLoop
|
|
from pprint import pprint
|
|
# kCOJ API
|
|
import access
|
|
from interface import kuser
|
|
# configurations
|
|
import config
|
|
|
|
def on_chat(msg):
|
|
content_type, chat_type, chat_id = telepot.glance(msg)
|
|
pprint(msg) # for debug
|
|
print(content_type, chat_type, chat_id) # for debug
|
|
|
|
# create a user object
|
|
user = kuser(chat_id, bot)
|
|
if str(chat_id) in users:
|
|
user = users[str(chat_id)]
|
|
else:
|
|
users[str(chat_id)] = user
|
|
|
|
if content_type == 'text':
|
|
# pre-treat the command
|
|
command = [msg['text']]
|
|
if msg['text'][0] == '/':
|
|
command = msg['text'].replace('_', ' ').lower().split(' ')
|
|
|
|
# first-time user
|
|
if user.status == '第一次用':
|
|
user.new_user()
|
|
|
|
# press password
|
|
elif user.status == '輸入學號':
|
|
user.press_password(msg['text'])
|
|
|
|
# login
|
|
elif user.status == '輸入密碼':
|
|
user.login_kcoj(msg['text'])
|
|
|
|
elif user.status == '舊的密碼':
|
|
if user.check_online() == True:
|
|
user.press_newpassword(msg['text'])
|
|
|
|
elif user.status == '修改密碼':
|
|
if user.check_online() == True:
|
|
user.change_password(msg['text'])
|
|
|
|
elif user.status == '上傳答案':
|
|
if user.check_online() == True:
|
|
user.send_answer(msg['text'], '')
|
|
|
|
elif command[0] == '/start' or command[0] == '重新整理🔃' or command[0] == '回主畫面🏠':
|
|
if user.check_online() == True:
|
|
user.display_main()
|
|
|
|
elif command[0] == '/question' or command[0] == '查看題庫📝' or command[0] == '回到題庫📝' or command[0] == '重新載入🔃':
|
|
if user.check_online() == True:
|
|
if len(command) > 1:
|
|
user.display_question(command[1])
|
|
else:
|
|
user.display_questions()
|
|
|
|
elif command[0] == '/help' or command[0] == '提供幫助📚':
|
|
if user.check_online() == True:
|
|
user.help_you()
|
|
|
|
elif command[0] == '/password' or command[0] == '修改密碼💱':
|
|
if user.check_online() == True:
|
|
user.press_oldpassword()
|
|
|
|
elif command[0] == '/logout' or command[0] == '登出帳號🚪':
|
|
user = kuser(chat_id, bot)
|
|
users[str(chat_id)] = user
|
|
user.logout_system()
|
|
|
|
elif user.question != '題外':
|
|
if user.check_online() == True:
|
|
if command[0] == '/upload' or command[0] == '上傳答案📮':
|
|
user.upload_answer()
|
|
elif command[0] == '/result' or command[0] == '查看結果☑️':
|
|
user.list_results()
|
|
elif command[0] == '/passer' or command[0] == '通過名單🌐':
|
|
user.list_passers()
|
|
elif command[0] == '回到題目📜':
|
|
user.display_question(user.question)
|
|
else:
|
|
bot.sendMessage(chat_id, "快去寫扣啦！")
|
|
|
|
elif content_type == 'document':
|
|
if user.status == '上傳答案' or user.status == '查看題目':
|
|
if user.check_online() == True:
|
|
if msg['document']['file_size'] > 167770000:
|
|
pass
|
|
else:
|
|
user.send_answer('', msg['document']['file_id'])
|
|
else:
|
|
bot.sendMessage(chat_id, "是擅長寫扣的朋友呢！")
|
|
else:
|
|
bot.sendMessage(chat_id, "我不是來看這些的。")
|
|
|
|
users = {}
|
|
bot = telepot.Bot(config.TOKEN)
|
|
MessageLoop(bot, on_chat).run_as_thread()
|
|
|
|
# for debug
|
|
def main():
|
|
pass
|
|
|
|
if __name__ == '__main__':
|
|
main()
|
|
while True:
|
|
time.sleep(100)
|
|
bot.getMe() |