neuro_llc/neuro-reply-bot-reworked
2
0
Fork 0
mirror of https://github.com/MrSedan/neuro-reply-bot-reworked.git synced 2026-01-15 05:59:43 +03:00
Code Issues Packages Projects Releases Wiki Activity

Added redis storage using for future

Browse Source
This commit is contained in:
Sergey Elpashev
2024-02-11 14:16:11 +03:00
parent 981101d228
commit 57fec6bb00
4 changed files with 33 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
neuroapi/config.py
View File

@@ -6,5 +6,20 @@ from pydantic_settings import BaseSettings
class GlobalConfig(BaseSettings):
api_url: str = Field("http://localhost:3000", alias='API_URL')
# Redis config
redis_host: str = Field("localhost", alias="REDIS_HOST")
redis_port: int = Field(6379, alias="REDIS_PORT")
redis_password: str = Field('', alias="REDIS_PASSWORD")
redis_db: int = Field(0, alias='REDIS_DB')
# Bot tokens
token: Optional[str] = Field(None, alias='TOKEN')
proxy_token: Optional[str] = Field(None, alias='PROXY_TOKEN')
proxy_token: Optional[str] = Field(None, alias='PROXY_TOKEN')
@property
def redis_url(self):
return f'redis://:{self.redis_password}@{self.redis_host}:{self.redis_port}/{self.redis_db}'
class Config:
env_file = '.env'

7
neuroapi/types/_bot.py
View File

@@ -1,4 +1,5 @@
from aiogram import Bot, Dispatcher
from aiogram.fsm.storage.redis import RedisStorage
from pydantic import BaseModel
from handlers.handler import Handler
@@ -13,12 +14,12 @@ class NeuroApiBot:
_instances = {}
def __init__(self, token: str) -> None:
def __init__(self, token: str, storage: RedisStorage | None = None) -> None:
token_data = Token(token=token)
self.bot = Bot(token_data.token)
self.dp = Dispatcher()
self.dp = Dispatcher(storage=storage)
def __new__(cls, token: str) -> 'NeuroApiBot':
def __new__(cls, token: str, storage: RedisStorage | None = None) -> 'NeuroApiBot':
token_data = Token(token=token)
if token_data.token not in cls._instances:
cls._instances[token_data.token] = super(NeuroApiBot, cls).__new__(cls)