125 lines
4.1 KiB
Python
125 lines
4.1 KiB
Python
#! /usr/bin/env python3
|
|
|
|
# necessary modules
|
|
import os, time, requests, telepot
|
|
from telepot.loop import MessageLoop
|
|
from pprint import pprint
|
|
# kCOJ API
|
|
import access
|
|
from interface import kuser
|
|
# configurations
|
|
import config
|
|
|
|
users = {}
|
|
bot = telepot.Bot(config.TOKEN)
|
|
|
|
def on_chat(msg):
|
|
content_type, chat_type, chat_id = telepot.glance(msg)
|
|
from_id = msg['from']['id']
|
|
pprint(msg) # for debug
|
|
print('content_type:', content_type) # for debug
|
|
print('chat_type:', chat_type) # for debug
|
|
print('chat_id:', chat_id) # for debug
|
|
print('from_id:', from_id) # for debug
|
|
print() # for debug
|
|
|
|
# create a user object
|
|
user = kuser(from_id, bot)
|
|
if str(from_id) in users:
|
|
user = users[str(from_id)]
|
|
else:
|
|
users[str(from_id)] = user
|
|
|
|
if content_type == 'text':
|
|
# pre-treat the command
|
|
command = [msg['text']]
|
|
if msg['text'][0] == '/':
|
|
command = msg['text'].replace('_', ' ').lower().split(' ')
|
|
command[0] = command[0].replace(config.NAME, '')
|
|
|
|
# first-time user
|
|
if user.status == '第一次用' and chat_type == 'private':
|
|
user.new_user()
|
|
|
|
# press password
|
|
elif user.status == '輸入學號' and chat_type == 'private':
|
|
user.press_password(msg['text'])
|
|
|
|
# login
|
|
elif user.status == '輸入密碼' and chat_type == 'private':
|
|
user.login_kcoj(msg['text'])
|
|
|
|
elif user.status == '舊的密碼' and chat_type == 'private':
|
|
if user.check_online() == True:
|
|
user.press_newpassword(msg['text'])
|
|
|
|
elif user.status == '修改密碼' and chat_type == 'private':
|
|
if user.check_online() == True:
|
|
user.change_password(msg['text'])
|
|
|
|
elif user.status == '上傳答案' and chat_type == 'private':
|
|
if user.check_online() == True:
|
|
user.send_answer(msg['text'], '')
|
|
|
|
elif command[0] == '/start' or command[0] == '首頁🏠':
|
|
if user.check_online() == True:
|
|
user.display_main()
|
|
|
|
elif command[0] == '/question' 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(from_id, bot)
|
|
users[str(from_id)] = user
|
|
user.logout_system()
|
|
|
|
elif command[0] == '/restart':
|
|
if str(from_id) in config.ADMIN:
|
|
bot.sendMessage(chat_id, "即將更新並重新啟動")
|
|
print("Restarting...")
|
|
time.sleep(1)
|
|
os._exit(0)
|
|
|
|
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)
|
|
|
|
elif content_type == 'document':
|
|
if user.status == '上傳答案' or user.status == '查看題目':
|
|
if user.check_online() == True:
|
|
if msg['document']['file_size'] > 167770000:
|
|
user.fail_send()
|
|
else:
|
|
user.send_answer('', msg['document']['file_id'])
|
|
|
|
MessageLoop(bot, on_chat).run_as_thread()
|
|
print("Started! Service is available.")
|
|
|
|
# for debug
|
|
def main():
|
|
pass
|
|
|
|
if __name__ == '__main__':
|
|
main()
|
|
while True:
|
|
time.sleep(1)
|
|
bot.getMe() |