cupcakearmy/mercatus
1
0
Fork 0
mirror of https://github.com/cupcakearmy/mercatus.git synced 2024-11-01 08:14:10 +01: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 yaml import load, Loader
from enum import Enum
import pickle
config = load(open('./config.yml', 'r'), Loader=Loader)
persistence = PicklePersistence('./data.db')
DB_FILE = './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)