Add class Artifact (#2313) (#2319)

Co-authored-by: Aleksei Fedotov <aleksei@fedotov.email>

github/Artifact.py

@@ -0,0 +1,178 @@
+############################ Copyrights and license ############################
+#                                                                              #
+# Copyright 2022 Aleksei Fedotov <aleksei@fedotov.email>                       #
+#                                                                              #
+# This file is part of PyGithub.                                               #
+# http://pygithub.readthedocs.io/                                              #
+#                                                                              #
+# PyGithub is free software: you can redistribute it and/or modify it under    #
+# the terms of the GNU Lesser General Public License as published by the Free  #
+# Software Foundation, either version 3 of the License, or (at your option)    #
+# any later version.                                                           #
+#                                                                              #
+# PyGithub 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 Lesser General Public License for more #
+# details.                                                                     #
+#                                                                              #
+# You should have received a copy of the GNU Lesser General Public License     #
+# along with PyGithub. If not, see <http://www.gnu.org/licenses/>.             #
+#                                                                              #
+################################################################################
+
+import github.GithubObject
+import github.WorkflowRun
+
+
+class Artifact(github.GithubObject.NonCompletableGithubObject):
+    """
+    This class represents an Artifact of Github Run
+    """
+
+    def __repr__(self):
+        return self.get__repr__({"name": self._name.value, "id": self._id.value})
+
+    @property
+    def archive_download_url(self):
+        """
+        :type: string
+        """
+        return self._archive_download_url.value
+
+    @property
+    def created_at(self):
+        """
+        :type: datetime.datetime
+        """
+        return self._created_at.value
+
+    @property
+    def expired(self):
+        """
+        :type: bool
+        """
+        return self._expired.value
+
+    @property
+    def expires_at(self):
+        """
+        :type: datetime.datetime
+        """
+        return self._expires_at.value
+
+    @property
+    def head_sha(self):
+        """
+        :type: string
+        """
+        return self._head_sha.value
+
+    @property
+    def id(self):
+        """
+        :type: string
+        """
+        return self._id.value
+
+    @property
+    def name(self):
+        """
+        :type: string
+        """
+        return self._name.value
+
+    @property
+    def node_id(self):
+        """
+        :type: string
+        """
+        return self._node_id.value
+
+    @property
+    def size_in_bytes(self):
+        """
+        :type: integer
+        """
+        return self._size_in_bytes.value
+
+    @property
+    def updated_at(self):
+        """
+        :type: datetime.datetime
+        """
+        return self._updated_at.value
+
+    @property
+    def url(self):
+        """
+        :type: string
+        """
+        return self._url.value
+
+    @property
+    def workflow_run(self):
+        """
+        :type: :class:``
+        """
+        return self._workflow_run.value
+
+    def delete(self) -> bool:
+        """
+        :calls: `DELETE /repos/{owner}/{repo}/actions/artifacts/{artifact_id} <https://docs.github.com/en/rest/actions/artifacts#delete-an-artifact>`_
+        :rtype: bool
+        """
+        status, headers, data = self._requester.requestBlob("DELETE", self.url)
+        return status == 204
+
+    def _initAttributes(self):
+        self._archive_download_url = github.GithubObject.NotSet
+        self._created_at = github.GithubObject.NotSet
+        self._expired = github.GithubObject.NotSet
+        self._expires_at = github.GithubObject.NotSet
+        self._head_sha = github.GithubObject.NotSet
+        self._id = github.GithubObject.NotSet
+        self._name = github.GithubObject.NotSet
+        self._node_id = github.GithubObject.NotSet
+        self._size_in_bytes = github.GithubObject.NotSet
+        self._updated_at = github.GithubObject.NotSet
+        self._url = github.GithubObject.NotSet
+        self._workflow_run = github.GithubObject.NotSet
+
+    def _useAttributes(self, attributes):
+        if "archive_download_url" in attributes:  # pragma no branch
+            self._archive_download_url = self._makeStringAttribute(
+                attributes["archive_download_url"]
+            )
+        if "created_at" in attributes:  # pragma no branch
+            assert attributes["created_at"] is None or isinstance(
+                attributes["created_at"], (str,)
+            ), attributes["created_at"]
+            self._created_at = self._makeDatetimeAttribute(attributes["created_at"])
+        if "expired" in attributes:  # pragma no branch
+            self._expired = self._makeBoolAttribute(attributes["expired"])
+        if "expires_at" in attributes:  # pragma no branch
+            assert attributes["expires_at"] is None or isinstance(
+                attributes["expires_at"], (str,)
+            ), attributes["expires_at"]
+            self._expires_at = self._makeDatetimeAttribute(attributes["expires_at"])
+        if "head_sha" in attributes:  # pragma no branch
+            self._head_sha = self._makeStringAttribute(attributes["head_sha"])
+        if "id" in attributes:  # pragma no branch
+            self._id = self._makeIntAttribute(attributes["id"])
+        if "name" in attributes:  # pragma no branch
+            self._name = self._makeStringAttribute(attributes["name"])
+        if "node_id" in attributes:  # pragma no branch
+            self._node_id = self._makeStringAttribute(attributes["node_id"])
+        if "size_in_bytes" in attributes:  # pragma no branch
+            self._size_in_bytes = self._makeIntAttribute(attributes["size_in_bytes"])
+        if "updated_at" in attributes:  # pragma no branch
+            assert attributes["updated_at"] is None or isinstance(
+                attributes["updated_at"], (str,)
+            ), attributes["updated_at"]
+            self._updated_at = self._makeDatetimeAttribute(attributes["updated_at"])
+        if "url" in attributes:  # pragma no branch
+            self._url = self._makeStringAttribute(attributes["url"])
+        if "workflow_run" in attributes:  # pragma no branch
+            self._workflow_run = self._makeClassAttribute(
+                github.WorkflowRun.WorkflowRun, attributes["workflow_run"]
+            )

github/Artifact.pyi

@@ -0,0 +1,30 @@
+from github.WorkflowRun import WorkflowRun
+from datetime import datetime
+from github.GithubObject import NonCompletableGithubObject as NonCompletableGithubObject
+
+class Artifact(NonCompletableGithubObject):
+    @property
+    def archive_download_url(self) -> str: ...
+    @property
+    def created_at(self) -> datetime: ...
+    def delete(self) -> bool: ...
+    @property
+    def expired(self) -> bool: ...
+    @property
+    def expires_at(self) -> datetime: ...
+    @property
+    def head_sha(self) -> str: ...
+    @property
+    def id(self) -> int: ...
+    @property
+    def name(self) -> str: ...
+    @property
+    def node_id(self) -> str: ...
+    @property
+    def size_in_bytes(self) -> int: ...
+    @property
+    def updated_at(self) -> datetime: ...
+    @property
+    def url(self) -> str: ...
+    @property
+    def workflow_run(self) -> WorkflowRun: ...

github/Repository.py

@@ -65,6 +65,7 @@
 # Copyright 2018 Yves Zumbach <yzumbach@andrew.cmu.edu>                        #
 # Copyright 2018 Leying Chen <leyingc@andrew.cmu.edu>                          #
 # Copyright 2020 Pascal Hofmann <mail@pascalhofmann.de>                        #
+# Copyright 2022 Aleksei Fedotov <aleksei@fedotov.email>                       #
 #                                                                              #
 # This file is part of PyGithub.                                               #
 # http://pygithub.readthedocs.io/                                              #
@@ -91,6 +92,7 @@ from base64 import b64encode
 
 from deprecated import deprecated
 
+import github.Artifact
 import github.Branch
 import github.CheckRun
 import github.CheckSuite
@@ -3703,6 +3705,33 @@ class Repository(github.GithubObject.CompletableGithubObject):
         )
         return github.CheckRun.CheckRun(self._requester, headers, data, completed=True)
 
+    def get_artifacts(self):
+        """
+        :calls: `GET /repos/{owner}/{repo}/actions/artifacts <https://docs.github.com/en/rest/actions/artifacts#list-artifacts-for-a-repository>`_
+        :rtype: :class:`github.PaginatedList.PaginatedList` of :class:`github.Artifact.Artifact`
+        """
+
+        return github.PaginatedList.PaginatedList(
+            github.Artifact.Artifact,
+            self._requester,
+            f"{self.url}/actions/artifacts",
+            None,
+            list_item="artifacts",
+        )
+
+    def get_artifact(self, artifact_id):
+        """
+        :calls: `GET /repos/{owner}/{repo}/actions/artifacts/{artifact_id} <https://docs.github.com/en/rest/actions/artifacts#get-an-artifact>`_
+        :param artifact_id: int
+        :rtype: :class:`github.Artifact.Artifact`
+        """
+        assert isinstance(artifact_id, int), artifact_id
+        headers, data = self._requester.requestJsonAndCheck(
+            "GET", f"{self.url}/actions/artifacts/{artifact_id}"
+        )
+
+        return github.Artifact.Artifact(self._requester, headers, data, completed=True)
+
     def _initAttributes(self):
         self._allow_merge_commit = github.GithubObject.NotSet
         self._allow_rebase_merge = github.GithubObject.NotSet

github/Repository.pyi

@@ -1,6 +1,7 @@
 from datetime import date, datetime
 from typing import Any, Dict, List, Optional, Union, overload
 
+from github.Artifact import Artifact
 from github.AuthenticatedUser import AuthenticatedUser
 from github.Branch import Branch
 from github.CheckRun import CheckRun
@@ -305,6 +306,8 @@ class Repository(CompletableGithubObject):
     def get_archive_link(
         self, archive_format: str, ref: Union[str, _NotSetType] = ...
     ) -> str: ...
+    def get_artifact(self) -> Artifact: ...
+    def get_artifacts(self) -> PaginatedList[Artifact]: ...
     def get_assignees(self) -> PaginatedList[NamedUser]: ...
     def get_branch(self, branch: str) -> Branch: ...
     def rename_branch(self, branch: Union[str, Branch], new_name: str) -> bool: ...

github/WorkflowRun.py

@@ -1,6 +1,7 @@
 ############################ Copyrights and license ############################
 #                                                                              #
 # Copyright 2020 Steve Kowalik <steven@wedontsleep.org>                        #
+# Copyright 2022 Aleksei Fedotov <aleksei@fedotov.email>                       #
 #                                                                              #
 # This file is part of PyGithub.                                               #
 # http://pygithub.readthedocs.io/                                              #
@@ -22,6 +23,7 @@
 
 from collections import namedtuple
 
+import github.Artifact
 import github.GithubObject
 import github.PullRequest
 
@@ -170,6 +172,15 @@ class WorkflowRun(github.GithubObject.CompletableGithubObject):
         self._completeIfNotSet(self._artifacts_url)
         return self._artifacts_url.value
 
+    def get_artifacts(self):
+        return github.PaginatedList.PaginatedList(
+            github.Artifact.Artifact,
+            self._requester,
+            self._artifacts_url.value,
+            None,
+            list_item="artifacts",
+        )
+
     @property
     def cancel_url(self):
         """