Ubuntu
58 lines
2.0 KiB
Python
58 lines
2.0 KiB
Python
"""Notification commands: unread counts and notification feeds."""
|
|
|
|
import click
|
|
|
|
from ..command_normalizers import normalize_unread_summary
|
|
from ..formatter import console, render_notifications
|
|
from ._common import handle_command, structured_output_options
|
|
|
|
|
|
@click.command()
|
|
@click.option(
|
|
"--type", "notif_type",
|
|
type=click.Choice(["mentions", "likes", "connections"]),
|
|
default="mentions",
|
|
help="Notification type: mentions (评论和@), likes (赞和收藏), connections (新增关注)",
|
|
)
|
|
@click.option("--cursor", default="", help="Pagination cursor")
|
|
@click.option("--num", default=20, help="Number of items per page")
|
|
@structured_output_options
|
|
@click.pass_context
|
|
def notifications(ctx, notif_type: str, cursor: str, num: int, as_json: bool, as_yaml: bool):
|
|
"""View notifications (mentions, likes, connections)."""
|
|
def _load_notifications(client):
|
|
if notif_type == "mentions":
|
|
return client.get_notification_mentions(cursor=cursor, num=num)
|
|
if notif_type == "likes":
|
|
return client.get_notification_likes(cursor=cursor, num=num)
|
|
return client.get_notification_connections(cursor=cursor, num=num)
|
|
|
|
handle_command(
|
|
ctx,
|
|
action=_load_notifications,
|
|
render=lambda data: render_notifications(data, notif_type),
|
|
as_json=as_json,
|
|
as_yaml=as_yaml,
|
|
)
|
|
|
|
|
|
@click.command()
|
|
@structured_output_options
|
|
@click.pass_context
|
|
def unread(ctx, as_json: bool, as_yaml: bool):
|
|
"""Show unread notification counts."""
|
|
def _render_unread(data):
|
|
summary = normalize_unread_summary(data)
|
|
console.print(f"📬 未读通知: [bold]{summary['unread_count']}[/bold]")
|
|
console.print(f" 💬 评论和@: {summary['mentions']}")
|
|
console.print(f" ❤️ 赞和收藏: {summary['likes']}")
|
|
console.print(f" 👥 新增关注: {summary['connections']}")
|
|
|
|
handle_command(
|
|
ctx,
|
|
action=lambda client: client.get_unread_count(),
|
|
render=_render_unread,
|
|
as_json=as_json,
|
|
as_yaml=as_yaml,
|
|
)
|