Travel-Agent/app/keyboards/travel.py

__all__ = ("get",)

from aiogram import types
from aiogram.utils.keyboard import InlineKeyboardBuilder

from app.models.travel import Travel


def get(travel: Travel):
    builder = InlineKeyboardBuilder()

    builder.row(
        types.InlineKeyboardButton(
            text="📝 Change title",
            callback_data=f"travel_change_{travel.id}_title",
        ),
        types.InlineKeyboardButton(
            text="ℹ️ Change description",
            callback_data=f"travel_change_{travel.id}_description",
        ),
    )
    builder.row(
        types.InlineKeyboardButton(
            text="🗺️ Locations",
            callback_data=f"travel_locations_page_{travel.id}_0",
        ),
        types.InlineKeyboardButton(
            text="➕ Add location",
            callback_data=f"travel_add_location_{travel.id}",
        ),
    )
    builder.row(
        types.InlineKeyboardButton(
            text="👤 Users",
            callback_data=f"travel_users_page_{travel.id}_0",
        ),
        types.InlineKeyboardButton(
            text="➕ Add user",
            callback_data=f"travel_add_user_{travel.id}",
        ),
    )
    builder.row(
        types.InlineKeyboardButton(
            text="📝 Notes",
            callback_data=f"travel_notes_page_{travel.id}_0",
        ),
        types.InlineKeyboardButton(
            text="➕ Add note",
            callback_data=f"travel_add_note_{travel.id}",
        ),
    )
    builder.row(
        types.InlineKeyboardButton(
            text="🗺️ Routes",
            callback_data=f"travel_routes_{travel.id}",
        ),
    )
    builder.row(
        types.InlineKeyboardButton(
            text="❌ Delete travel",
            callback_data=f"travel_delete_{travel.id}",
        ),
    )
    builder.row(
        types.InlineKeyboardButton(
            text="⬅️",
            callback_data="travels",
        ),
    )

    return builder.as_markup()


def get_public(travel: Travel):
    builder = InlineKeyboardBuilder()

    builder.row(
        types.InlineKeyboardButton(
            text="🗺️ Locations",
            callback_data=f"travel_locations_{travel.id}",
        ),
    )
    builder.row(
        types.InlineKeyboardButton(
            text="👤 Users",
            callback_data=f"travel_users_{travel.id}",
        ),
    )
    builder.row(
        types.InlineKeyboardButton(
            text="📝 Notes",
            callback_data=f"travel_notes_{travel.id}",
        ),
        types.InlineKeyboardButton(
            text="➕ Add note",
            callback_data=f"travel_add_note_{travel.id}",
        ),
    )
    builder.row(
        types.InlineKeyboardButton(
            text="🗺️ Routes",
            callback_data=f"travel_routes_{travel.id}",
        ),
    )
    builder.row(
        types.InlineKeyboardButton(
            text="⬅️",
            callback_data="travels",
        ),
    )

    return builder.as_markup()