Skip to content

Commit

Permalink
Add class Artifact (#2313) (#2319)
Browse files Browse the repository at this point in the history
Co-authored-by: Aleksei Fedotov <aleksei@fedotov.email>
  • Loading branch information
lexa and Aleksei Fedotov committed Oct 25, 2022
1 parent 17b44d8 commit 437ff84
Show file tree
Hide file tree
Showing 12 changed files with 432 additions and 0 deletions.
178 changes: 178 additions & 0 deletions 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"]
)
30 changes: 30 additions & 0 deletions 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: ...
29 changes: 29 additions & 0 deletions github/Repository.py
Expand Up @@ -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/ #
Expand All @@ -91,6 +92,7 @@

from deprecated import deprecated

import github.Artifact
import github.Branch
import github.CheckRun
import github.CheckSuite
Expand Down Expand Up @@ -3703,6 +3705,33 @@ def get_check_run(self, check_run_id):
)
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
Expand Down
3 changes: 3 additions & 0 deletions 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
Expand Down Expand Up @@ -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: ...
Expand Down
11 changes: 11 additions & 0 deletions 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/ #
Expand All @@ -22,6 +23,7 @@

from collections import namedtuple

import github.Artifact
import github.GithubObject
import github.PullRequest

Expand Down Expand Up @@ -170,6 +172,15 @@ def artifacts_url(self):
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):
"""
Expand Down
60 changes: 60 additions & 0 deletions tests/Artifact.py
@@ -0,0 +1,60 @@
#!/usr/bin/env python3

import github

from . import Framework


class Artifact(Framework.TestCase):
def setUp(self):
super().setUp()
self.repo = self.g.get_repo("github/vscode-codeql")

def testGetArtifactsFromWorkflow(self):
artifact = self.repo.get_workflow_run(160995070).get_artifacts()[0]

self.assertEqual(artifact.name, "vscode-codeql-extension")
self.assertTrue(artifact.expired)
self.assertEqual(
repr(artifact), 'Artifact(name="vscode-codeql-extension", id=10495898)'
)

def testGetSingleArtifactFromRepo(self):
artifact = self.repo.get_artifact(378970214)

self.assertEqual(artifact.name, "vscode-codeql-extension")
self.assertFalse(artifact.expired)
self.assertEqual(
repr(artifact), 'Artifact(name="vscode-codeql-extension", id=378970214)'
)

def testGetArtifactsFromRepo(self):
artifact_id = 378970214
artifacts = self.repo.get_artifacts()
for item in artifacts:
if item.id == artifact_id:
artifact = item
break
else:
assert False, f"No artifact {artifact_id} is found"

self.assertEqual(
repr(artifact),
f'Artifact(name="vscode-codeql-extension", id={artifact_id})',
)

def testGetNonexistentArtifact(self):
artifact_id = 396724437
repo_name = "lexa/PyGithub"
repo = self.g.get_repo(repo_name)
with self.assertRaises(github.GithubException):
repo.get_artifact(artifact_id)

def testDelete(self):
artifact_id = 396724439
repo_name = "lexa/PyGithub"
repo = self.g.get_repo(repo_name)
artifact = repo.get_artifact(artifact_id)
self.assertTrue(artifact.delete())
with self.assertRaises(github.GithubException):
repo.get_artifact(artifact_id)

0 comments on commit 437ff84

Please sign in to comment.