diff --git a/telethon/client/telegrambaseclient.py b/telethon/client/telegrambaseclient.py index 02cb05e9..56e1926f 100644 --- a/telethon/client/telegrambaseclient.py +++ b/telethon/client/telegrambaseclient.py @@ -202,7 +202,7 @@ class TelegramBaseClient(abc.ABC): if isinstance(session, str) or session is None: try: session = SQLiteSession(session) - except ValueError: + except ImportError: import warnings warnings.warn( 'The sqlite3 module is not available under this ' diff --git a/telethon/sessions/sqlite.py b/telethon/sessions/sqlite.py index ebb773e3..f01843dc 100644 --- a/telethon/sessions/sqlite.py +++ b/telethon/sessions/sqlite.py @@ -14,8 +14,10 @@ from ..tl.types import ( try: import sqlite3 -except ImportError: + sqlite3_err = None +except ImportError as e: sqlite3 = None + sqlite3_err = type(e) EXTENSION = '.session' CURRENT_VERSION = 5 # database version @@ -32,7 +34,7 @@ class SQLiteSession(MemorySession): def __init__(self, session_id=None): if sqlite3 is None: - raise ValueError('sqlite3 is not installed') + raise sqlite3_err super().__init__() self.filename = ':memory:'