mirror of
https://github.com/revanced/revanced-polling-api.git
synced 2025-04-29 22:24:26 +02:00
24 lines
636 B
Python
24 lines
636 B
Python
import os
|
|
import toml
|
|
from redis import asyncio as aioredis
|
|
|
|
# Load config
|
|
|
|
config: dict = toml.load("config.toml")
|
|
|
|
# Redis connection parameters
|
|
|
|
redis_config: dict[ str, str | int ] = {
|
|
"url": f"redis://{os.environ['REDIS_URL']}",
|
|
"port": os.environ['REDIS_PORT'],
|
|
}
|
|
|
|
class RedisConnector:
|
|
"""Implements the RedisConnector class for the ReVanced API"""
|
|
|
|
@staticmethod
|
|
def connect(database: str) -> aioredis.Redis:
|
|
"""Connect to Redis"""
|
|
redis_url = f"{redis_config['url']}:{redis_config['port']}/{database}"
|
|
return aioredis.from_url(redis_url, encoding="utf-8", decode_responses=True)
|