mirror/pyTelegramBotAPI
1
0
Fork 0
mirror of https://github.com/eternnoir/pyTelegramBotAPI.git synced 2023-08-10 21:12:57 +03:00
Code Releases Activity

Merge pull request #2015 from Badiboy/master

Browse Source 
Fix aioredis version check for regular redis
This commit is contained in:
Badiboy 2023-07-14 13:00:03 +03:00 committed by GitHub
commit 0f52ca688f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
telebot/asyncio_storage/redis_storage.py
View File

@ -1,10 +1,11 @@
from telebot.asyncio_storage.base_storage import StateStorageBase, StateContext from telebot.asyncio_storage.base_storage import StateStorageBase, StateContext
import json import json
redis_installed = True redis_installed = True
is_actual_aioredis = False
try: try:
import aioredis import aioredis
is_actual_aioredis = True
except ImportError: except ImportError:
try: try:
from redis import asyncio as aioredis from redis import asyncio as aioredis
@ -23,10 +24,10 @@ class StateRedisStorage(StateStorageBase):
if not redis_installed: if not redis_installed:
raise ImportError('AioRedis is not installed. Install it via "pip install aioredis"') raise ImportError('AioRedis is not installed. Install it via "pip install aioredis"')
if is_actual_aioredis:
aioredis_version = tuple(map(int, aioredis.__version__.split(".")[0])) aioredis_version = tuple(map(int, aioredis.__version__.split(".")[0]))
if aioredis_version < (2,): if aioredis_version < (2,):
raise ImportError('Invalid aioredis version. Aioredis version should be >= 2.0.0') raise ImportError('Invalid aioredis version. Aioredis version should be >= 2.0.0')
self.redis = aioredis.Redis(host=host, port=port, db=db, password=password) self.redis = aioredis.Redis(host=host, port=port, db=db, password=password)
self.prefix = prefix self.prefix = prefix