cupcakearmy/mercatus
1
0
Fork 0
mirror of https://github.com/cupcakearmy/mercatus.git synced 2024-12-22 16:16:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

default db data

Browse Source
This commit is contained in:
cupcakearmy 2019-08-27 15:33:44 +02:00
parent b7336b6a97
commit 840f1b4511
1 changed files with 16 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
src/Utils.py
View File

@ -2,9 +2,23 @@ from telegram import Update
from telegram.ext import PicklePersistence, Updater from telegram.ext import PicklePersistence, Updater
from yaml import load, Loader from yaml import load, Loader
from enum import Enum from enum import Enum
import pickle
config = load(open('./config.yml', 'r'), Loader=Loader) DB_FILE = './data.db'
persistence = PicklePersistence('./data.db') CONFIG_FILE = './config.yml'
DEFAULT_DATA = {
'user_data': {},
'chat_data': {},
'conversations': {},
}
try:
pickle.load(open(DB_FILE, 'rb'))
except:
pickle.dump(DEFAULT_DATA, open(DB_FILE, 'wb'))
config = load(open(CONFIG_FILE, 'r'), Loader=Loader)
persistence = PicklePersistence(DB_FILE)
updater: Updater = Updater(config['token'], use_context=True, persistence=persistence) updater: Updater = Updater(config['token'], use_context=True, persistence=persistence)