592 lines
23 KiB
Python
592 lines
23 KiB
Python
#! /usr/bin/env python3
|
||
|
||
# modules
|
||
import os
|
||
import sys
|
||
import time
|
||
import json
|
||
from random import choice
|
||
from pprint import pprint
|
||
import telepot
|
||
from telepot.loop import MessageLoop
|
||
from telepot.namedtuple import ReplyKeyboardMarkup, ReplyKeyboardRemove
|
||
# config
|
||
from config import NAME, URL, TOKEN
|
||
from kcoj import KCOJ
|
||
import promote
|
||
import external
|
||
|
||
bot = telepot.Bot(TOKEN)
|
||
users = {}
|
||
|
||
class Kuser:
|
||
def __init__(self, userid, username='', password='', status='第一次用', question='題外'):
|
||
self.userid = userid
|
||
self.username = username
|
||
self.password = password
|
||
self.status = status
|
||
self.question = question
|
||
self.api = KCOJ(URL)
|
||
|
||
# 新使用者要登入
|
||
def new_user(self):
|
||
self.help()
|
||
self.press_username()
|
||
|
||
# 輸入學號
|
||
def press_username(self):
|
||
self.question = '題外'
|
||
self.status = '輸入學號'
|
||
bot.sendMessage(self.userid, "請輸入您的學號：", reply_markup=ReplyKeyboardRemove())
|
||
|
||
# 輸入密碼
|
||
def press_password(self, text):
|
||
self.question = '題外'
|
||
self.status = '輸入密碼'
|
||
self.username = text
|
||
bot.sendMessage(self.userid,
|
||
"輸入完可刪除訊息以策安全！\n"
|
||
"請輸入您的密碼：",
|
||
reply_markup=ReplyKeyboardRemove()
|
||
)
|
||
|
||
# 輸入舊密碼
|
||
def press_oldpassword(self):
|
||
self.question = '題外'
|
||
self.status = '舊的密碼'
|
||
bot.sendMessage(self.userid,
|
||
"請輸入要原本的舊密碼：",
|
||
reply_markup=ReplyKeyboardMarkup(keyboard=[
|
||
["首頁🏠"]
|
||
], resize_keyboard=True)
|
||
)
|
||
|
||
# 輸入新密碼
|
||
def press_newpassword(self, text):
|
||
self.question = '題外'
|
||
# 判斷舊密碼是否輸入正確
|
||
if text == self.password:
|
||
# 正確舊密碼
|
||
self.status = '修改密碼'
|
||
bot.sendMessage(self.userid,
|
||
"使用此功能請務必小心！\n"
|
||
"請輸入要設定的新密碼：",
|
||
reply_markup=ReplyKeyboardMarkup(keyboard=[
|
||
["首頁🏠"]
|
||
], resize_keyboard=True)
|
||
)
|
||
else:
|
||
# 錯誤舊密碼
|
||
self.status = '正常使用'
|
||
bot.sendMessage(self.userid, "密碼錯誤！",
|
||
reply_markup=ReplyKeyboardMarkup(keyboard=[
|
||
["首頁🏠"]
|
||
], resize_keyboard=True)
|
||
)
|
||
|
||
# 開始修改密碼
|
||
def change_password(self, text):
|
||
self.question = '題外'
|
||
self.status = '正常使用'
|
||
self.password = text
|
||
bot.sendMessage(self.userid,
|
||
"修改成功！" if self.api.change_password(self.password) else "修改失敗。",
|
||
reply_markup=ReplyKeyboardMarkup(keyboard=[
|
||
["首頁🏠"]
|
||
], resize_keyboard=True)
|
||
)
|
||
|
||
# 執行登入
|
||
def login(self, text):
|
||
self.question = '題外'
|
||
self.status = '正常使用'
|
||
self.password = text
|
||
bot.sendMessage(self.userid, "登入中...", reply_markup=ReplyKeyboardRemove())
|
||
if self.check_online(self.userid):
|
||
self.show_homepage(self.userid)
|
||
|
||
# 登入失敗
|
||
def login_failed(self, chat_id, message_id):
|
||
self.question = '題外'
|
||
self.status = '正常使用'
|
||
# 判斷使用者從哪操作
|
||
if chat_id != self.userid:
|
||
# 從群組操作
|
||
bot.sendMessage(chat_id, "登入失敗，請先私訊我重新登入 KCOJ", reply_to_message_id=message_id)
|
||
else:
|
||
# 從私訊操作
|
||
bot.sendMessage(self.userid, "哇...登入失敗，讓我們重新開始", reply_markup=ReplyKeyboardRemove())
|
||
self.press_username()
|
||
|
||
# 網站連接失敗
|
||
def connect_failed(self, chat_id, message_id):
|
||
self.question = '題外'
|
||
self.status = '正常使用'
|
||
# 群組操作
|
||
if chat_id != self.userid:
|
||
bot.sendMessage(chat_id, "KCOJ 離線中！請稍後再試", reply_to_message_id=message_id)
|
||
# 私訊操作
|
||
else:
|
||
bot.sendMessage(self.userid, "KCOJ 離線中！請稍後再試",
|
||
reply_markup=ReplyKeyboardMarkup(keyboard=[
|
||
["首頁🏠", "幫助📚"]
|
||
], resize_keyboard=True)
|
||
)
|
||
|
||
# 確認登入是否正常
|
||
def check_online(self, chat_id, message_id=''):
|
||
result = self.api.check_online()
|
||
# 判斷是否可連接上
|
||
if result == None:
|
||
# 連接失敗
|
||
self.connect_failed(chat_id, message_id)
|
||
return False
|
||
else:
|
||
# 連接成功
|
||
# 判斷是否登入
|
||
if result == False:
|
||
# 沒有登入的話再嘗試重新登入一次
|
||
self.api.login(self.username, self.password, 2)
|
||
result = self.api.check_online()
|
||
# 再次確認是否登入＆連接
|
||
if result == None:
|
||
# 連接失敗
|
||
self.connect_failed(chat_id, message_id)
|
||
elif result == False:
|
||
# 登入失敗
|
||
self.login_failed(chat_id, message_id)
|
||
# 回傳登入狀態
|
||
return result
|
||
|
||
# 登出
|
||
def logout(self):
|
||
self.question = '題外'
|
||
self.status = '正常使用'
|
||
bot.sendMessage(self.userid, "您現在已經是登出的狀態。", reply_markup=ReplyKeyboardRemove())
|
||
self.press_username()
|
||
|
||
# 秀出主畫面
|
||
def show_homepage(self, chat_id):
|
||
self.question = '題外'
|
||
self.status = '正常使用'
|
||
# 題目列表字典
|
||
q_dict = self.api.list_questions()
|
||
# 題目列表字串
|
||
q_str = ''
|
||
# 將字典內容根據格式附加到字串上
|
||
for key in q_dict.keys():
|
||
# 只顯示期限未到的作業
|
||
if q_dict[key][1] == '期限未到':
|
||
q_str += (
|
||
"📗<b>{NUM}</b> (DL: {DL})\n"
|
||
" [[{LANG}]] [[{STATUS}]]{STAT_ICON} /question_{NUM}\n"
|
||
"\n".format(
|
||
NUM=key,
|
||
DL=q_dict[key][0],
|
||
LANG=q_dict[key][3],
|
||
STATUS=q_dict[key][2],
|
||
STAT_ICON=("⚠️" if q_dict[key][2] == '未繳' else "✅")
|
||
)
|
||
)
|
||
# 顯示主頁面
|
||
bot.sendMessage(chat_id,
|
||
# 畫面格式
|
||
"💁 <b>{NAME}</b> {BOT_NAME}\n"
|
||
"➖➖➖➖➖\n"
|
||
"📝<i>可繳交的作業</i>\n"
|
||
"\n"
|
||
"{Q_STR}"
|
||
"➖➖➖➖➖\n"
|
||
"{PROMOTE}"
|
||
# 填入資訊
|
||
"".format(NAME=self.username, BOT_NAME=NAME, Q_STR=q_str, PROMOTE=choice(promote.sentences)),
|
||
parse_mode='HTML',
|
||
reply_markup=
|
||
# 群組內不顯示按鈕
|
||
ReplyKeyboardRemove() if chat_id != self.userid else
|
||
# 私訊內顯示按鈕
|
||
ReplyKeyboardMarkup(keyboard=[
|
||
["題庫📝"],
|
||
["登出🚪", "改密碼💱", "幫助📚"]
|
||
], resize_keyboard=True),
|
||
disable_web_page_preview=False
|
||
)
|
||
|
||
# 列出題目列表
|
||
def list_questions(self, chat_id):
|
||
self.question = '題外'
|
||
self.status = '正常使用'
|
||
# 題目列表字典
|
||
q_dict = self.api.list_questions()
|
||
# 題目列表字串
|
||
q_str = ''
|
||
# 將字典內容根據格式附加到字串上
|
||
for key in q_dict.keys():
|
||
q_str += (
|
||
"{DL_ICON}<b>{NUM}</b> (DL: {DL})\n"
|
||
" [[{LANG}]] [[{STATUS}]]{STAT_ICON} /question_{NUM}\n"
|
||
"\n".format(
|
||
DL_ICON=("📗" if q_dict[key][1] == '期限未到' else "📕"),
|
||
NUM=key,
|
||
DL=q_dict[key][0],
|
||
LANG=q_dict[key][3],
|
||
STATUS=q_dict[key][2],
|
||
STAT_ICON=("⚠️" if q_dict[key][2] == '未繳' else "✅")
|
||
)
|
||
)
|
||
# 顯示題目列表並將訊息存起來
|
||
msg = bot.sendMessage(chat_id,
|
||
# 畫面格式
|
||
"💁 <b>{NAME}</b> {BOT_NAME}\n"
|
||
"➖➖➖➖➖\n"
|
||
"📝<i>所有作業</i>\n"
|
||
"\n"
|
||
"{Q_STR}"
|
||
"➖➖➖➖➖\n"
|
||
"{PROMOTE}"
|
||
# 填入資訊
|
||
"".format(NAME=self.username, BOT_NAME=NAME, Q_STR=q_str, PROMOTE=choice(promote.sentences)),
|
||
parse_mode='HTML',
|
||
reply_markup=
|
||
# 群組內不顯示按鈕
|
||
ReplyKeyboardRemove() if chat_id != self.userid else
|
||
# 私訊內顯示按鈕
|
||
ReplyKeyboardMarkup(keyboard=[
|
||
["首頁🏠", "更新🔃"],
|
||
["登出🚪", "改密碼💱", "幫助📚"]
|
||
], resize_keyboard=True) ,
|
||
disable_web_page_preview=False
|
||
)
|
||
# 顯示點我到頂的訊息
|
||
bot.sendMessage(chat_id, "點我到題庫頂", reply_to_message_id=msg['message_id'])
|
||
|
||
def show_question(self, number, chat_id):
|
||
self.question = number
|
||
self.status = '查看題目'
|
||
# 如果外部有指定題目內容
|
||
if number in external.QUESTION:
|
||
# 顯示外部內容
|
||
EXT = True
|
||
content = external.QUESTION[number]
|
||
else:
|
||
# 顯示內部內容
|
||
EXT = False
|
||
content = '```' + self.api.show_question(number) + '```'
|
||
# 題目資訊字典
|
||
q_info = self.api.list_questions()[number]
|
||
|
||
# 顯示題目內容並將訊息存起來
|
||
msg = bot.sendMessage(chat_id,
|
||
"💁 *{NAME}* [{BOT_NAME}]\n"
|
||
"➖➖➖➖➖\n"
|
||
"{DL_ICON}*{NUM}* (DL: {DL})\n"
|
||
" [[[{LANG}]]] [[[{STATUS}]]]{STAT_ICON}\n"
|
||
"\n"
|
||
"{CONTENT}\n".format(
|
||
NAME=self.username,
|
||
BOT_NAME=NAME,
|
||
DL_ICON=("📗" if q_info[1] == '期限未到' else "📕"),
|
||
NUM=number,
|
||
DL=q_info[0],
|
||
LANG=q_info[3],
|
||
STATUS=q_info[2],
|
||
STAT_ICON=("⚠️" if q_info[2] == '未繳' else "✅"),
|
||
CONTENT=content
|
||
),
|
||
parse_mode='Markdown',
|
||
reply_markup=
|
||
# 群組內不顯示按鈕
|
||
ReplyKeyboardRemove() if chat_id != self.userid else
|
||
# 私訊內顯示按鈕
|
||
ReplyKeyboardMarkup(keyboard=[
|
||
["首頁🏠", "題庫📝"],
|
||
["交作業📮" if q_info[1] == '期限未到' else '', "看結果☑️" if q_info[2] == '已繳' else '', "通過者🌐"],
|
||
["登出🚪", "改密碼💱", "幫助📚"]
|
||
], resize_keyboard=True)
|
||
)
|
||
if EXT == False:
|
||
# 顯示點我到頂的訊息
|
||
bot.sendMessage(chat_id, "點我到題目頂", reply_to_message_id=msg['message_id'])
|
||
|
||
def help(self):
|
||
bot.sendMessage(self.userid,
|
||
"這裡是 Kuo C Online Judge Bot！\n"
|
||
"可以簡稱 KCOJ Bot，目前定居於 [{BOT_NAME}]\n"
|
||
"作用是讓大家可以方便的透過我使用郭老程設課的 Online Judge\n"
|
||
"➡️[OJ 傳送門]({URL})\n"
|
||
"操作很簡單（？）\n\n"
|
||
"還是稍微提幾個需要注意的地方：\n"
|
||
"1. 📗代表還可以繳交的作業，📕代表已經不能繳交的作業\n"
|
||
"2. ⚠️代表還沒有繳交的作業，✅代表已經繳交的作業\n"
|
||
"3. 其實在查看題目的畫面就可以用「拖曳」的方式 *上傳作業📮*\n"
|
||
"4. *刪除作業⚔️* 的功能被放在 *上傳作業📮* 裡面\n"
|
||
"5. 學號與密碼將以「明文」方式儲存，因為....\n"
|
||
"6. 郭老的 Online Judge 其實也是以「明文」方式儲存您的帳號密碼\n"
|
||
"7. 我以我的人格擔保，不會使用您提供的資訊侵害您的權利\n\n"
|
||
"本專案採用 *MIT License*\n"
|
||
"聯絡我請私訊 @PinLin\n"
|
||
"原始碼被託管於 GitHub，如果想要鼓勵我的話可以幫我按個星星> </\n"
|
||
"網址如下：\n"
|
||
"[https://github.com/PinLin/KCOJ_bot]".format(BOT_NAME=NAME, URL=URL),
|
||
parse_mode='Markdown'
|
||
)
|
||
|
||
def upload_answer(self):
|
||
self.status = '上傳答案'
|
||
q = self.api.list_questions()[self.question]
|
||
q_str = "💁 <b>" + self.username + "</b> " + NAME + "\n"
|
||
q_str += "➖➖➖➖➖\n"
|
||
q_str += "📗" if q[1] == '期限未到' else "📕"
|
||
q_str += "<b>" + self.question + "</b> (DL: " + q[0] + ")\n [[" + q[3] + "]] [[" + q[2] + "]]"
|
||
q_str += "⚠️" if q[2] == '未繳' else "✅"
|
||
bot.sendMessage(self.userid,
|
||
q_str + "\n\n現在請把你的程式碼讓我看看（請別超過 20 MB）\n"
|
||
"可以使用「文字訊息」或是「傳送檔案」的方式\n"
|
||
"（注意：可在程式碼前後加上單獨成行的 ``` 避免可能的錯誤。）", parse_mode='HTML',
|
||
reply_markup=ReplyKeyboardMarkup(keyboard=[
|
||
["刪除作業⚔️"] if self.api.list_questions()[self.question][2] == '已繳' else [],
|
||
["首頁🏠", "回題目📜"]
|
||
], resize_keyboard=True))
|
||
|
||
def send_answer(self, text, file_id):
|
||
self.status = '正常使用'
|
||
# define filename
|
||
filename = sys.path[0] + '/' + self.username + self.question
|
||
if self.api.list_questions()[self.question][3] == 'Python':
|
||
filename += '.py'
|
||
else:
|
||
filename += '.c'
|
||
|
||
if text != '':
|
||
with open(filename, 'w') as f:
|
||
f.write(text)
|
||
else:
|
||
bot.download_file(file_id, filename)
|
||
self.api.delete_answer(self.question)
|
||
if self.api.upload_answer(self.question, filename) == True:
|
||
bot.sendMessage(self.userid, "上傳成功",
|
||
reply_markup=ReplyKeyboardMarkup(keyboard=[
|
||
["首頁🏠", "回題目📜"],
|
||
["看結果☑️"],
|
||
["登出🚪", "改密碼💱", "幫助📚"]
|
||
], resize_keyboard=True))
|
||
else:
|
||
bot.sendMessage(self.userid, "上傳失敗",
|
||
reply_markup=ReplyKeyboardMarkup(keyboard=[
|
||
["首頁🏠", "回題目📜"],
|
||
["登出🚪", "改密碼💱", "幫助📚"]
|
||
], resize_keyboard=True))
|
||
os.remove(filename)
|
||
|
||
def delete_answer(self):
|
||
bot.sendMessage(self.userid, "移除成功" if self.api.delete_answer(self.question) == True else "移除失敗",
|
||
reply_markup=ReplyKeyboardMarkup(keyboard=[
|
||
["首頁🏠", "回題目📜"],
|
||
["登出🚪", "改密碼💱", "幫助📚"]
|
||
], resize_keyboard=True))
|
||
|
||
def send_failed(self):
|
||
self.status = '正常使用'
|
||
bot.sendMessage(self.userid, "檔案不能超過 20 MB！上傳失敗",
|
||
reply_markup=ReplyKeyboardMarkup(keyboard=[
|
||
["首頁🏠", "回題目📜"],
|
||
["登出🚪", "改密碼💱", "幫助📚"]
|
||
], resize_keyboard=True))
|
||
|
||
def list_passers(self):
|
||
self.status = '正常使用'
|
||
q = self.api.list_questions()[self.question]
|
||
q_str = "💁 <b>" + self.username + "</b> " + NAME + "\n"
|
||
q_str += "➖➖➖➖➖\n"
|
||
q_str += "📗" if q[1] == '期限未到' else "📕"
|
||
q_str += "<b>" + self.question + "</b> (DL: " + q[0] + ")\n [[" + q[3] + "]] [[" + q[2] + "]]"
|
||
q_str += "⚠️" if q[2] == '未繳' else "✅"
|
||
q_str += "<code>\n"
|
||
for passer in self.api.list_passers(self.question):
|
||
q_str += "\n" + passer
|
||
reply = bot.sendMessage(self.userid, q_str + "</code>",
|
||
parse_mode='HTML',
|
||
reply_markup=ReplyKeyboardMarkup(keyboard=[
|
||
["首頁🏠", "回題目📜"],
|
||
["登出🚪", "改密碼💱", "幫助📚"]
|
||
], resize_keyboard=True))
|
||
bot.sendMessage(self.userid, "點我到名單頂", reply_to_message_id=reply['message_id'])
|
||
|
||
def list_results(self):
|
||
self.status = '正常使用'
|
||
q = self.api.list_questions()[self.question]
|
||
q_str = "💁 <b>" + self.username + "</b> " + NAME + "\n"
|
||
q_str += "➖➖➖➖➖\n"
|
||
q_str += "📗" if q[1] == '期限未到' else "📕"
|
||
q_str += "<b>" + self.question + "</b> (DL: " + q[0] + ")\n"
|
||
for result in self.api.list_results(self.question, self.username):
|
||
q_str += "\n測試編號 <code>" + result[0] + "</code>："
|
||
q_str += "✔️ " if result[1] == '通過測試' else "❌ "
|
||
q_str += result[1]
|
||
bot.sendMessage(self.userid, q_str,
|
||
parse_mode='HTML',
|
||
reply_markup=ReplyKeyboardMarkup(keyboard=[
|
||
["首頁🏠", "回題目📜"],
|
||
["交作業📮" if q[1] == '期限未到' else '', "通過者🌐"],
|
||
["登出🚪", "改密碼💱", "幫助📚"]
|
||
], resize_keyboard=True))
|
||
|
||
def on_chat(msg):
|
||
content_type, chat_type, chat_id = telepot.glance(msg)
|
||
from_id = msg['from']['id']
|
||
|
||
# create a user object
|
||
user = Kuser(from_id)
|
||
if str(from_id) in users:
|
||
user = users[str(from_id)]
|
||
else:
|
||
users[str(from_id)] = user
|
||
|
||
# debug message
|
||
# ==========================================================
|
||
pprint(msg)
|
||
# name
|
||
if 'last_name' in msg['from']:
|
||
last_name = msg['from']['last_name']
|
||
else:
|
||
last_name = ''
|
||
print("😊 student_name:", msg['from']['first_name'], last_name, "😊")
|
||
# id
|
||
print("😯 student_id:", user.username, "😯")
|
||
print()
|
||
# ==========================================================
|
||
|
||
if content_type == 'text':
|
||
# pre-treat the command
|
||
command = [msg['text']]
|
||
if msg['text'].startswith('/'):
|
||
command = msg['text'].replace(NAME, '').replace('_', ' ').lower().split(' ')
|
||
|
||
# test connection
|
||
if command[0] == '/ping':
|
||
bot.sendMessage(chat_id, "*PONG*",
|
||
parse_mode='Markdown',
|
||
reply_to_message_id=msg['message_id'])
|
||
|
||
# help message
|
||
elif command[0] == '/help' or command[0] == '幫助📚':
|
||
if chat_type == 'private':
|
||
user.help()
|
||
|
||
# first-time user
|
||
elif user.status == '第一次用':
|
||
if chat_type == 'private':
|
||
user.new_user()
|
||
|
||
# press password
|
||
elif user.status == '輸入學號':
|
||
if chat_type == 'private':
|
||
user.press_password(msg['text'])
|
||
|
||
# login
|
||
elif user.status == '輸入密碼':
|
||
if chat_type == 'private':
|
||
user.login(msg['text'])
|
||
|
||
# homepage
|
||
elif command[0] == '/start' or command[0] == '首頁🏠':
|
||
if user.check_online(chat_id, msg['message_id']) == True:
|
||
user.show_homepage(chat_id)
|
||
|
||
elif command[0] == '/question' or command[0] == '題庫📝' or command[0] == '更新🔃':
|
||
if user.check_online(chat_id, msg['message_id']) == True:
|
||
if len(command) > 1:
|
||
user.show_question(command[1], chat_id)
|
||
else:
|
||
user.list_questions(chat_id)
|
||
|
||
elif chat_type == 'private':
|
||
if command[0] == '/password' or command[0] == '改密碼💱':
|
||
if user.check_online(chat_id, msg['message_id']) == True:
|
||
user.press_oldpassword()
|
||
|
||
elif command[0] == '/logout' or command[0] == '登出🚪':
|
||
user = Kuser(from_id)
|
||
users[str(from_id)] = user
|
||
user.logout()
|
||
|
||
elif (command[0] == '/delete' or command[0] == '刪除作業⚔️') and user.question != '題外':
|
||
if user.check_online(chat_id, msg['message_id']) == True:
|
||
user.delete_answer()
|
||
|
||
elif (command[0] == '/upload' or command[0] == '交作業📮') and user.question != '題外':
|
||
if user.check_online(chat_id, msg['message_id']) == True:
|
||
user.upload_answer()
|
||
|
||
elif (command[0] == '/result' or command[0] == '看結果☑️') and user.question != '題外':
|
||
if user.check_online(chat_id, msg['message_id']) == True:
|
||
user.list_results()
|
||
|
||
elif (command[0] == '/passer' or command[0] == '通過者🌐') and user.question != '題外':
|
||
if user.check_online(chat_id, msg['message_id']) == True:
|
||
user.list_passers()
|
||
|
||
elif command[0] == '回題目📜' and user.question != '題外':
|
||
if user.check_online(chat_id, msg['message_id']) == True:
|
||
user.show_question(user.question, chat_id)
|
||
|
||
elif user.status == '舊的密碼':
|
||
if user.check_online(chat_id, msg['message_id']) == True:
|
||
user.press_newpassword(msg['text'])
|
||
|
||
elif user.status == '修改密碼':
|
||
if user.check_online(chat_id, msg['message_id']) == True:
|
||
user.change_password(msg['text'])
|
||
|
||
elif user.status == '上傳答案':
|
||
if user.check_online(chat_id, msg['message_id']) == True:
|
||
user.send_answer(msg['text'], '')
|
||
|
||
else:
|
||
if user.check_online(chat_id, msg['message_id']) == True:
|
||
bot.sendMessage(chat_id, "(ˊ・ω・ˋ)")
|
||
|
||
elif content_type == 'document':
|
||
if user.status == '上傳答案' or user.status == '查看題目':
|
||
if user.check_online(chat_id, msg['message_id']) == True:
|
||
if msg['document']['file_size'] > 167770000:
|
||
user.send_failed()
|
||
else:
|
||
user.send_answer('', msg['document']['file_id'])
|
||
|
||
def backup_db():
|
||
users_backup = {}
|
||
for key in users.keys():
|
||
user = users[key]
|
||
users_backup[key] = {
|
||
'userid': user.userid,
|
||
'username': user.username,
|
||
'password': user.password,
|
||
'status': user.status,
|
||
'question': user.question
|
||
}
|
||
with open(sys.path[0] + '/users.json', 'w') as f:
|
||
json.dump(users_backup, f, indent=' ')
|
||
|
||
def restore_db():
|
||
with open(sys.path[0] + '/users.json', 'r') as f:
|
||
users_restore = json.load(f)
|
||
for key in users_restore.keys():
|
||
user = users_restore[key]
|
||
users[key] = Kuser(user['userid'], user['username'], user['password'], user['status'], user['question'])
|
||
|
||
# restore
|
||
restore_db()
|
||
|
||
# start this bot
|
||
MessageLoop(bot, on_chat).run_as_thread()
|
||
print("Started! Service is available.")
|
||
|
||
while True:
|
||
time.sleep(60)
|
||
|
||
# keep bot alive
|
||
bot.getMe()
|
||
|
||
# backup
|
||
backup_db() |