Added the set_permissions function.

Author: LTLA

src/pygobbler/__init__.py

@@ -39,3 +39,4 @@
 from .refresh_usage import refresh_usage
 from .approve_probation import approve_probation
 from .reject_probation import reject_probation
+from .set_permissions import set_permissions

src/pygobbler/set_permissions.py

@@ -0,0 +1,54 @@
+from typing import Optional, List, Dict
+from . import _utils as ut
+from . import fetch_permissions
+
+
+def set_permissions(project: str, registry: str, staging: str, url: str, owners: Optional[List] = None, uploaders: Optional[Dict] = None, append: bool = True):
+    """
+    Set the owner and uploader permissions for a project.
+
+    Args:
+        project:
+            Name of the project.
+
+        registry:
+            Path to the Gobbler registry.
+
+        staging:
+            Path to the staging directory.
+
+        url:
+            URL of the REST API.
+
+        owners:
+            List of user IDs for owners of this project. If None, no change is
+            made to the existing owners in the project permissions.
+
+        uploaders:
+            List of dictionaries specifying the authorized uploaders for this
+            project.  See the ``uploaders`` field in the return value of
+            :py:func:`~fetch_permissions.fetch_permissions` for  the expected
+            format. If None, no change is made to the existing uploaders.
+
+        append:
+            Whether ``owners`` and ``uploaders`` should be appended to the
+            existing owners and uploaders, respectively. If False, the
+            ``owners`` and ``uploaders`` are used to replace the existing
+            values in the project permissions.
+    """
+    perms = {}
+
+    if append:
+        oldperms = fetch_permissions(project, registry=registry, url=url)
+        if owners is not None:
+            oldset = set(oldperms["owners"])
+            perms["owners"] = oldperms["owners"] + list(filter(lambda x : x not in oldset, owners))
+        if uploaders is not None:
+            perms["uploaders"] = oldperms["uploaders"] + uploaders
+    else:
+        if owners is not None:
+            perms["owners"] = owners
+        if uploaders is not None:
+            perms["uploaders"] = uploaders
+
+    ut.dump_request(staging, url, "set_permissions", { "project": project, "permissions": perms })

tests/test_set_permissions.py

@@ -0,0 +1,49 @@
+import pygobbler as pyg
+
+
+def test_set_permissions():
+    _, staging, registry, url = pyg.start_gobbler()
+    pyg.remove_project("test-perms", staging=staging, url=url)
+    pyg.create_project("test-perms", staging=staging, url=url, owners=["LTLA"])
+
+    until = "2022-02-02T02:20:02.02Z"
+    pyg.set_permissions("test-perms",
+        owners=["jkanche"], 
+        uploaders=[ { "id": "lawremi", "until": until } ],
+        staging=staging,
+        url=url,
+        registry=registry
+    )
+
+    perms = pyg.fetch_permissions("test-perms", registry=registry, url=url)
+    assert perms["owners"] == [ "LTLA", "jkanche" ]
+    assert len(perms["uploaders"]) == 1
+    assert perms["uploaders"][0]["id"] == "lawremi"
+    assert perms["uploaders"][0]["until"] == until
+
+    # Checking uploader appending, while also checking owners=NULL.
+    pyg.set_permissions("test-perms", uploaders=[ { "id": "ArtifactDB-bot", "trusted": True } ], staging=staging, url=url, registry=registry)
+    perms = pyg.fetch_permissions("test-perms", registry=registry, url=url)
+    assert perms["owners"] == [ "LTLA", "jkanche" ]
+    assert len(perms["uploaders"]) == 2
+    assert perms["uploaders"][0]["id"] == "lawremi"
+    assert perms["uploaders"][1]["id"] == "ArtifactDB-bot"
+    assert perms["uploaders"][1]["trusted"]
+
+    # Checking union of owners, and also that uploaders=NULL works.
+    pyg.set_permissions("test-perms", owners=[ "PeteHaitch", "LTLA" ], staging=staging, url=url, registry=registry)
+    perms = pyg.fetch_permissions("test-perms", registry=registry, url=url)
+    assert perms["owners"] == [ "LTLA", "jkanche", "PeteHaitch" ]
+    assert len(perms["uploaders"]) == 2
+
+    # Resetting the owners back.
+    pyg.set_permissions("test-perms", owners=[ "LTLA" ], append=False, staging=staging, url=url, registry=registry)
+    perms = pyg.fetch_permissions("test-perms", registry=registry, url=url)
+    assert perms["owners"] == [ "LTLA" ]
+    assert len(perms["uploaders"]) == 2
+
+    # Now resetting the uploaders. 
+    pyg.set_permissions("test-perms", uploaders=[], append=False, staging=staging, url=url, registry=registry)
+    perms = pyg.fetch_permissions("test-perms", registry=registry, url=url)
+    assert perms["owners"] == [ "LTLA" ]
+    assert len(perms["uploaders"]) == 0