user data storing

This commit is contained in:
2017-08-24 18:51:20 +08:00
parent 56a70d8a7d
commit 58f16a3434
2 changed files with 24 additions and 2 deletions

25
bot.py
View File

@ -1,7 +1,7 @@
#! /usr/bin/env python3
# necessary modules
import os, time, requests, telepot
import os, time, json, requests, telepot
from telepot.loop import MessageLoop
from pprint import pprint
# kCOJ API
@ -138,8 +138,29 @@ def on_chat(msg):
else:
user.send_answer('', msg['document']['file_id'])
# restore
with open('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'])
MessageLoop(bot, on_chat).run_as_thread()
print("Started! Service is available.")
while True:
time.sleep(1)
bot.getMe()
bot.getMe()
# backup
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('users.json', 'w') as f:
json.dump(users_backup, f)

1
users.json Normal file
View File

@ -0,0 +1 @@
{}