Fallback to MemorySession if sqlite3 is not available

This commit is contained in:
Lonami Exo
2018-08-02 15:17:44 +02:00
parent 16f7626ceb
commit 7a7923b317
2 changed files with 21 additions and 3 deletions

View File

@@ -1,7 +1,6 @@
import datetime
import json
import os
import sqlite3
from base64 import b64decode
from os.path import isfile as file_exists
@@ -13,6 +12,11 @@ from ..tl.types import (
InputPhoto, InputDocument, PeerUser, PeerChat, PeerChannel
)
try:
import sqlite3
except ImportError:
sqlite3 = None
EXTENSION = '.session'
CURRENT_VERSION = 4 # database version
@@ -27,6 +31,9 @@ class SQLiteSession(MemorySession):
"""
def __init__(self, session_id=None):
if sqlite3 is None:
raise ValueError('sqlite3 is not installed')
super().__init__()
self.filename = ':memory:'
self.save_entities = True