add function for adding assignees to issues

ogr/abstract.py

@@ -299,6 +299,14 @@ def add_label(self, *labels: str) -> None:
         """
         raise NotImplementedError()
 
+    def add_assignee(self, *assignees: str) -> None:
+        """
+        Assign users to an issue.
+
+        :param assignees: [str]
+        """
+        raise NotImplementedError()
+
 
 class PRStatus(IntEnum):
     open = 1

ogr/services/github/issue.py

@@ -164,3 +164,6 @@ def close(self) -> "Issue":
     def add_label(self, *labels: str) -> None:
         for label in labels:
             self._raw_issue.add_to_labels(label)
+
+    def add_assignee(self, *assignees: str) -> None:
+        self._raw_issue.edit(assignees=list(assignees))

ogr/services/github/service.py

@@ -189,7 +189,7 @@ def project_create(
 
         new_repo = owner.create_repo(
             name=repo,
-            description=description if description else github.GithubObject.NotSet,
+            description=description if description else github.GithubObject.NotSet,  # type: ignore
         )
         return GithubProject(
             repo=repo,

ogr/services/gitlab/issue.py

@@ -67,7 +67,10 @@ def url(self) -> str:
 
     @property
     def assignees(self) -> list:
-        return self._raw_issue.assignees
+        try:
+            return self._raw_issue.assignees
+        except AttributeError:
+            return None  # if issue has no assignees, the attribute is not present
 
     @property
     def description(self) -> str:
@@ -171,3 +174,18 @@ def add_label(self, *labels: str) -> None:
         for label in labels:
             self._raw_issue.labels.append(label)
         self._raw_issue.save()
+
+    def add_assignee(self, *assignees: str) -> None:
+        assignee_ids = self._raw_issue.__dict__.get("assignee_ids") or []
+        for assignee in assignees:
+            users = self.project.service.gitlab_instance.users.list(  # type: ignore
+                username=assignee
+            )
+            if not users:
+                raise GitlabAPIException(f"Unable to find '{assignee}' username")
+            uid = str(users[0].id)
+            if uid not in assignee_ids:
+                assignee_ids.append(str(users[0].id))
+
+        self._raw_issue.assignee_ids = assignee_ids
+        self._raw_issue.save()

ogr/services/pagure/issue.py

@@ -74,7 +74,10 @@ def url(self) -> str:
     @property
     def assignee(self) -> str:
         self.__update()
-        return self._raw_issue["assignee"]["name"]
+        try:
+            return self._raw_issue["assignee"]["name"]
+        except Exception:
+            return None
 
     @property
     def description(self) -> str:
@@ -203,3 +206,11 @@ def close(self) -> "PagureIssue":
         )
         self.__dirty = True
         return self
+
+    def add_assignee(self, *assignees: str) -> None:
+        if len(assignees) > 1:
+            raise PagureAPIException("Pagure does not support multiple assignees")
+        payload = {"assignee": assignees[0]}
+        self.project._call_project_api(
+            "issue", str(self.id), "assign", data=payload, method="POST"
+        )

tests/integration/github/test_issues.py

@@ -110,6 +110,22 @@ def test_issue_labels(self):
         assert labels[0].name == "test_lb1"
         assert labels[1].name == "test_lb2"
 
+    def test_issue_assignees(self):
+        """
+        Remove the assignees from this issue before regenerating the response files:
+        https://github.com/packit/ogr/issues/4
+        """
+        project = self.service.get_project(repo="ogr", namespace="KPostOffice")
+        issue = project.get_issue(4)
+        print(self.service.user.get_username())
+        assignees = issue.assignees
+
+        assert not assignees
+        issue.add_assignee("KPostOffice")
+        assignees = project.get_issue(4).assignees
+        assert len(assignees) == 1
+        assert assignees[0].login == "KPostOffice"
+
     def test_list_contains_only_issues(self):
         issue_list_all = self.ogr_project.get_issue_list(status=IssueStatus.all)
         issue_ids = [issue.id for issue in issue_list_all]