feat: add Planka service provider with full notification and command support
Webhook-based provider for Planka (self-hosted Kanban board) with: - 15 event types (cards, boards, lists, comments, tasks, attachments, labels) - Bearer token webhook authentication - Async API client for boards/cards/lists - 30 notification templates (en/ru) + 26 command templates (en/ru) - Bot commands: /status, /boards, /cards, /lists - Default tracking config, template config, command config seeded on startup - DB migration for 15 new tracking_config columns - Frontend: provider config UI with auto-name, Planka-specific hints - Frontend: tracking config event toggles for all 15 Planka events
This commit is contained in:
@@ -0,0 +1,193 @@
|
||||
"""Planka-specific bot command handler."""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import logging
|
||||
from typing import Any
|
||||
|
||||
import aiohttp
|
||||
from sqlmodel import select
|
||||
from sqlmodel.ext.asyncio.session import AsyncSession
|
||||
|
||||
from ..database.engine import get_engine
|
||||
from ..database.models import (
|
||||
CommandConfig, CommandTracker, EventLog,
|
||||
NotificationTracker, ServiceProvider, TelegramBot,
|
||||
)
|
||||
from ..services import make_planka_provider
|
||||
from .base import ProviderCommandHandler
|
||||
from .handler import _render_cmd_template, _get_notification_trackers_for_providers
|
||||
|
||||
_LOGGER = logging.getLogger(__name__)
|
||||
|
||||
_PLANKA_COMMANDS = {"status", "boards", "cards", "lists"}
|
||||
|
||||
|
||||
class PlankaCommandHandler(ProviderCommandHandler):
|
||||
"""Handles Planka-specific bot commands."""
|
||||
|
||||
provider_type = "planka"
|
||||
|
||||
def get_provider_commands(self) -> set[str]:
|
||||
return _PLANKA_COMMANDS
|
||||
|
||||
def get_rate_categories(self) -> dict[str, str]:
|
||||
return {
|
||||
"boards": "api", "cards": "api", "lists": "api",
|
||||
}
|
||||
|
||||
async def handle(
|
||||
self,
|
||||
cmd: str,
|
||||
args: str,
|
||||
count: int,
|
||||
locale: str,
|
||||
response_mode: str,
|
||||
providers_map: dict[int, ServiceProvider],
|
||||
cmd_templates: dict[str, dict[str, str]],
|
||||
bot: TelegramBot,
|
||||
ctx_tuples: list[tuple[CommandTracker, CommandConfig, ServiceProvider]],
|
||||
) -> str | list[dict[str, Any]] | None:
|
||||
if cmd == "status":
|
||||
ctx = await _cmd_status(providers_map)
|
||||
return _render_cmd_template(cmd_templates, "status", locale, ctx)
|
||||
if cmd == "boards":
|
||||
ctx = await _cmd_boards(providers_map)
|
||||
return _render_cmd_template(cmd_templates, "boards", locale, ctx)
|
||||
if cmd == "cards":
|
||||
ctx = await _cmd_cards(providers_map, count)
|
||||
return _render_cmd_template(cmd_templates, "cards", locale, ctx)
|
||||
if cmd == "lists":
|
||||
ctx = await _cmd_lists(providers_map)
|
||||
return _render_cmd_template(cmd_templates, "lists", locale, ctx)
|
||||
return None
|
||||
|
||||
|
||||
def _get_tracked_board_ids(
|
||||
providers_map: dict[int, ServiceProvider],
|
||||
trackers: list[NotificationTracker],
|
||||
) -> list[tuple[ServiceProvider, str]]:
|
||||
"""Get (provider, board_id) tuples from tracked collection_ids."""
|
||||
boards: list[tuple[ServiceProvider, str]] = []
|
||||
for tracker in trackers:
|
||||
provider = providers_map.get(tracker.provider_id)
|
||||
if not provider or provider.type != "planka":
|
||||
continue
|
||||
if not provider.config.get("api_key"):
|
||||
continue
|
||||
for board_id in (tracker.collection_ids or []):
|
||||
entry = (provider, board_id)
|
||||
if entry not in boards:
|
||||
boards.append(entry)
|
||||
# Also check filters.collections
|
||||
for board_id in (tracker.filters or {}).get("collections", []):
|
||||
entry = (provider, board_id)
|
||||
if entry not in boards:
|
||||
boards.append(entry)
|
||||
return boards[:20]
|
||||
|
||||
|
||||
async def _cmd_status(providers_map: dict[int, ServiceProvider]) -> dict[str, Any]:
|
||||
provider_ids = set(providers_map.keys())
|
||||
trackers = await _get_notification_trackers_for_providers(provider_ids)
|
||||
tracked_boards = _get_tracked_board_ids(providers_map, trackers)
|
||||
|
||||
# Last event
|
||||
engine = get_engine()
|
||||
async with AsyncSession(engine) as session:
|
||||
tracker_ids = [t.id for t in trackers]
|
||||
if tracker_ids:
|
||||
result = await session.exec(
|
||||
select(EventLog)
|
||||
.where(EventLog.tracker_id.in_(tracker_ids))
|
||||
.order_by(EventLog.created_at.desc()).limit(1)
|
||||
)
|
||||
last_event = result.first()
|
||||
else:
|
||||
last_event = None
|
||||
last_str = last_event.created_at.strftime("%Y-%m-%d %H:%M") if last_event else "-"
|
||||
|
||||
return {
|
||||
"boards_count": len(tracked_boards),
|
||||
"last_event": last_str,
|
||||
}
|
||||
|
||||
|
||||
async def _cmd_boards(providers_map: dict[int, ServiceProvider]) -> dict[str, Any]:
|
||||
provider_ids = set(providers_map.keys())
|
||||
trackers = await _get_notification_trackers_for_providers(provider_ids)
|
||||
tracked_boards = _get_tracked_board_ids(providers_map, trackers)
|
||||
|
||||
boards_data: list[dict[str, Any]] = []
|
||||
async with aiohttp.ClientSession() as http:
|
||||
for provider, board_id in tracked_boards:
|
||||
planka = make_planka_provider(http, provider)
|
||||
all_boards = await planka.client.get_boards()
|
||||
for b in all_boards:
|
||||
if str(b.get("id", "")) == board_id:
|
||||
boards_data.append({"name": b.get("name", board_id)})
|
||||
break
|
||||
else:
|
||||
boards_data.append({"name": board_id})
|
||||
|
||||
return {"boards": boards_data}
|
||||
|
||||
|
||||
async def _cmd_cards(
|
||||
providers_map: dict[int, ServiceProvider], count: int,
|
||||
) -> dict[str, Any]:
|
||||
provider_ids = set(providers_map.keys())
|
||||
trackers = await _get_notification_trackers_for_providers(provider_ids)
|
||||
tracked_boards = _get_tracked_board_ids(providers_map, trackers)
|
||||
|
||||
all_cards: list[dict[str, Any]] = []
|
||||
async with aiohttp.ClientSession() as http:
|
||||
for provider, board_id in tracked_boards:
|
||||
planka = make_planka_provider(http, provider)
|
||||
cards = await planka.client.get_board_cards(board_id, limit=count)
|
||||
lists = await planka.client.get_board_lists(board_id)
|
||||
lists_by_id = {str(lst.get("id", "")): lst.get("name", "") for lst in lists}
|
||||
|
||||
boards = await planka.client.get_boards()
|
||||
board_name = board_id
|
||||
for b in boards:
|
||||
if str(b.get("id", "")) == board_id:
|
||||
board_name = b.get("name", board_id)
|
||||
break
|
||||
|
||||
for card in cards:
|
||||
list_id = str(card.get("listId", ""))
|
||||
all_cards.append({
|
||||
"name": card.get("name", ""),
|
||||
"list_name": lists_by_id.get(list_id, ""),
|
||||
"board_name": board_name,
|
||||
})
|
||||
|
||||
return {"cards": all_cards[:count]}
|
||||
|
||||
|
||||
async def _cmd_lists(providers_map: dict[int, ServiceProvider]) -> dict[str, Any]:
|
||||
provider_ids = set(providers_map.keys())
|
||||
trackers = await _get_notification_trackers_for_providers(provider_ids)
|
||||
tracked_boards = _get_tracked_board_ids(providers_map, trackers)
|
||||
|
||||
all_lists: list[dict[str, Any]] = []
|
||||
async with aiohttp.ClientSession() as http:
|
||||
for provider, board_id in tracked_boards:
|
||||
planka = make_planka_provider(http, provider)
|
||||
lists = await planka.client.get_board_lists(board_id)
|
||||
|
||||
boards = await planka.client.get_boards()
|
||||
board_name = board_id
|
||||
for b in boards:
|
||||
if str(b.get("id", "")) == board_id:
|
||||
board_name = b.get("name", board_id)
|
||||
break
|
||||
|
||||
for lst in lists:
|
||||
all_lists.append({
|
||||
"name": lst.get("name", ""),
|
||||
"board_name": board_name,
|
||||
})
|
||||
|
||||
return {"lists": all_lists}
|
||||
Reference in New Issue
Block a user