Add the CLI command to send a file content as a notification.

main
KKlochko 1 year ago
parent fe00d09a64
commit 29eaebe595

@ -1,5 +1,28 @@
##########################################################################
# Copyright (C) 2023 Kostiantyn Klochko <kklochko@protonmail.com> #
# #
# This file is part of notification-producer. #
# #
# notification-producer is free software: you can redistribute it and/or #
# modify it under the terms of the GNU Affero General Public License as #
# published by the Free Software Foundation, either version 3 of the #
# License, or (at your option) any later version. #
# #
# notification-producer is distributed in the hope that it will be #
# useful, but WITHOUT ANY WARRANTY; without even the implied warranty #
# of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU #
# Affero General Public License for more details. #
# #
# You should have received a copy of the GNU Affero General Public #
# License along with notification-producer. If not, see #
# <https://www.gnu.org/licenses/>. #
##########################################################################
import asyncio import asyncio
import typer import typer
import os
import sys
from typing_extensions import Annotated
from rich.console import Console from rich.console import Console
from notification_producer.producer_client import ProducerClient from notification_producer.producer_client import ProducerClient
@ -58,3 +81,44 @@ def message(
client = ProducerClient(ip, port) client = ProducerClient(ip, port)
asyncio.run(client.notify(notification_title, notification_message)) asyncio.run(client.notify(notification_title, notification_message))
@notify.command()
def file(
notification_message_path: Annotated[str, typer.Option(
"--file", "-f",
help="[b yellow]The message file[/] for the [y b]notification[/].",
rich_help_panel="Notification",
show_default=False,
)],
notification_title: str = typer.Option(
"notification", "--title", "-t",
help="[b yellow]The title[/] for the [y b]notification[/].",
rich_help_panel="Notification",
),
ip: str = typer.Option(
"127.0.0.1", "--ip", "-i",
help="The server's ip.",
rich_help_panel="Connection",
),
port: int = typer.Option(
5554, "--port", "-p",
help="The server's port for [b]producers[/].",
rich_help_panel="Connection",
),
):
"""
This command [b]send[/] the file content as [yellow b]one notification[/].
"""
client = ProducerClient(ip, port)
if not os.path.isfile(notification_message_path):
sys.stderr.write(f"The file ({notification_message_path}) does not exist or this is not a file. Please, check the path.\n")
sys.exit(2)
with open(notification_message_path) as file:
notification_message = file.read()
asyncio.run(client.notify(notification_title, notification_message))

Loading…
Cancel
Save