dds-cli 2.8.0__tar.gz → 2.8.1__tar.gz

{dds_cli-2.8.0 → dds_cli-2.8.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dds_cli
-Version: 2.8.0
+Version: 2.8.1
 Summary: A command line tool to manage data and projects in the SciLifeLab Data Delivery System.
 Home-page: https://github.com/ScilifelabDataCentre/dds_cli
 Author: SciLifeLab Data Centre

{dds_cli-2.8.0 → dds_cli-2.8.1}/dds_cli/__main__.py

@@ -2070,15 +2070,22 @@ def deactivate_motd(click_ctx, motd_id):
 # -- dds motd send -- #
 @motd_group_command.command(name="send")
 @click.argument("motd_id", metavar="[MOTD_ID]", nargs=1, type=int, required=True)
+@click.option(
+    "--unit-only",
+    is_flag=True,
+    required=False,
+    default=False,
+    help="Only send MOTD to Unit Admins and Unit Personnel.",
+)
 @click.pass_obj
-def send_motd(click_ctx, motd_id):
+def send_motd(click_ctx, motd_id, unit_only):
     """Send motd as email to all users."""
     try:
         with dds_cli.motd_manager.MotdManager(
             no_prompt=click_ctx.get("NO_PROMPT", False),
             token_path=click_ctx.get("TOKEN_PATH"),
         ) as sender:
-            sender.send_motd(motd_id=motd_id)
+            sender.send_motd(motd_id=motd_id, unit_only=unit_only)
     except (
         dds_cli.exceptions.AuthenticationError,
         dds_cli.exceptions.ApiResponseError,

{dds_cli-2.8.0 → dds_cli-2.8.1}/dds_cli/motd_manager.py

@@ -126,13 +126,13 @@ class MotdManager(dds_cli.base.DDSBaseClass):
         )
         LOG.info(response_message)
 
-    def send_motd(self, motd_id: int) -> None:
+    def send_motd(self, motd_id: int, unit_only=False) -> None:
         """Send specific MOTD to users."""
         response_json, _ = dds_cli.utils.perform_request(
             endpoint=DDSEndpoint.MOTD_SEND,
             headers=self.token,
             method="post",
-            json={"motd_id": motd_id},
+            json={"motd_id": motd_id, "unit_only": unit_only},
             error_message="Failed sending the MOTD to users",
         )
 

{dds_cli-2.8.0 → dds_cli-2.8.1}/dds_cli/version.py

@@ -2,4 +2,4 @@
 
 # Do not change bump the major version unless absolutely necessary - makes incompatible with API
 # If mid or minor version reaches 9, continue to 10, 11 etc.
-__version__ = "2.8.0"
+__version__ = "2.8.1"

{dds_cli-2.8.0 → dds_cli-2.8.1}/dds_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dds_cli
-Version: 2.8.0
+Version: 2.8.1
 Summary: A command line tool to manage data and projects in the SciLifeLab Data Delivery System.
 Home-page: https://github.com/ScilifelabDataCentre/dds_cli
 Author: SciLifeLab Data Centre

{dds_cli-2.8.0 → dds_cli-2.8.1}/tests/test_motd_manager.py

@@ -222,3 +222,30 @@ def test_add_new_motd_ok(caplog: LogCaptureFixture):
                 logging.INFO,
                 "Response from API about adding a MOTD.",
             ) in caplog.record_tuples
+
+
+# send_motd
+
+
+def test_send_motd_all(caplog: LogCaptureFixture):
+    """Send a MOTD to all users."""
+
+    motd_id = 1
+    response_message = "Response from API about sending a MOTD."
+    returned_response: Dict = {"message": response_message}
+
+    with caplog.at_level(logging.INFO):
+        # Create mocker
+        with Mocker() as mock:
+            # Create mocked request - real request not executed
+            mock.post(DDSEndpoint.MOTD_SEND, status_code=200, json=returned_response)
+
+            with motd_manager.MotdManager(authenticate=False, no_prompt=True) as mtdm:
+                mtdm.token = {}  # required, otherwise none
+                mtdm.send_motd(motd_id=motd_id, unit_only=False)  # Send motd
+
+            assert (
+                "dds_cli.motd_manager",
+                logging.INFO,
+                response_message,
+            ) in caplog.record_tuples