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

2023-12-13 12:10:27 +02:00
parent fe00d09a64
commit 29eaebe595
1 changed files with 64 additions and 0 deletions
@@ -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 typer
+import os
+import sys
+from typing_extensions import Annotated
 from rich.console import Console
 from notification_producer.producer_client import ProducerClient

@@ -58,3 +81,44 @@ def message(
    client = ProducerClient(ip, port)
    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))
+
+