You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
58 lines
1.6 KiB
58 lines
1.6 KiB
import asyncio
|
|
import typer
|
|
from rich.console import Console
|
|
from notification_producer.producer_client import ProducerClient
|
|
|
|
console = Console()
|
|
cli_app = typer.Typer(rich_markup_mode="rich")
|
|
notify = typer.Typer()
|
|
cli_app.add_typer(notify, name="notify", help="To send notification.")
|
|
|
|
|
|
@cli_app.command()
|
|
def up(
|
|
ip: str = typer.Option(
|
|
"127.0.0.1", "--ip", "-i",
|
|
help="The server's ip.",
|
|
),
|
|
port: int = typer.Option(
|
|
5554, "--port", "-p",
|
|
help="The server's port for [b]producers[/].",
|
|
),
|
|
):
|
|
"""
|
|
This command [b]start[/] the [yellow b]producer[/] to send messages in [b]the interactive mode[/].
|
|
"""
|
|
|
|
client = ProducerClient(ip, port)
|
|
asyncio.run(client.notify_interactive())
|
|
|
|
|
|
@notify.command()
|
|
def message(
|
|
ip: str = typer.Option(
|
|
"127.0.0.1", "--ip", "-i",
|
|
help="The server's ip.",
|
|
),
|
|
port: int = typer.Option(
|
|
5554, "--port", "-p",
|
|
help="The server's port for [b]producers[/].",
|
|
),
|
|
notification_title: str = typer.Option(
|
|
"notification", "--title", "-t",
|
|
help="[b yellow]The title[/] for the [y b]notification[/]."
|
|
),
|
|
# TODO as argument (required)
|
|
notification_message: str = typer.Option(
|
|
'hello', "--title", "-t",
|
|
help="[b yellow]The message[/] for the [y b]notification[/]."
|
|
),
|
|
):
|
|
"""
|
|
This command [b]send[/] [yellow b]one notification[/].
|
|
"""
|
|
|
|
client = ProducerClient(ip, port)
|
|
asyncio.run(client.notify(notification_title, notification_message))
|
|
|