DataBridge

DataBridge — это сущность, предоставляющая временное хранилище данных, которое существует в рамках одной сессии приложения (от запуска до выхода). Она предназначена для обмена данными между обработчиками.

Основной способ получения доступа к DataBridge — через DI.

from argenta.di import FromDishka
from argenta import DataBridge, Response

# ... setting up router and other

def my_handler(response: Response, data_bridge: FromDishka[DataBridge]):
    # ... your code

Практический пример: Аутентификация

Рассмотрим пример, где команда login сохраняет токен аутентификации, а команда get-profile использует его.

from argenta import Router, Response, Command, DataBridge
from argenta.command import Flag
from argenta.di import FromDishka

router = Router(title="Authentication")


def authenticate_user(username: str) -> str:
    return f"token_for_{username}"


@router.command(Command("login", flags=Flag("username")))
def login_handler(response: Response, data_bridge: FromDishka[DataBridge]):
    username_flag = response.input_flags.get_flag_by_name("username")
    if not username_flag or not username_flag.input_value:
        print("Error: username must be specified using the --username flag.")
        return

    username = username_flag.input_value
    token = authenticate_user(username)

    data_bridge.update({"auth_token": token})
    print(f"Login successful! User '{username}' authenticated.")


@router.command("get-profile")
def get_profile_handler(response: Response, data_bridge: FromDishka[DataBridge]):
    token = data_bridge.get_by_key("auth_token")

    if not token:
        print("Error: you are not authenticated. Please run the 'login' command first.")
        return

    print(f"Loading profile using token: [yellow]{token}[/yellow]")


@router.command("logout")
def logout_handler(response: Response, data_bridge: FromDishka[DataBridge]):
    try:
        data_bridge.delete_by_key("auth_token")
        print("Logout successful. Session data cleared.")
    except KeyError:
        print("You were not authenticated anyway.")

Как это работает:

1. При вызове обработчика dishka автоматически внедряет экземпляр DataBridge.

2. Команда login --username <имя> вызывает login_handler, который через внедрённый data_bridge сохраняет токен.

3. Команда get-profile вызывает get_profile_handler, который так же получает data_bridge и извлекает из него токен.

---

class DataBridge

__init__(self, initial_data: dict | None = None)

Инициализирует хранилище. При использовании через DI вызывается автоматически.

update(self, data: dict) → None

Обновляет хранилище данными из словаря.

get_all(self) → dict

Возвращает все данные из хранилища.

get_by_key(self, key: str) → Any

Возвращает значение по ключу или None, если ключ не найден.

delete_by_key(self, key: str) → None

Удаляет значение по ключу. Вызывает KeyError, если ключ не найден.

clear_all(self) → None

Полностью очищает хранилище.