Skip to content

Commit

Permalink
Add class Artifact (PyGithub#2313)
Browse files Browse the repository at this point in the history
Github workflow runs produce downloadable artifacts, add a class to
represent them.
  • Loading branch information
Aleksei Fedotov committed Oct 13, 2022
1 parent 7d3c3f0 commit 77ace0a
Show file tree
Hide file tree
Showing 5 changed files with 263 additions and 0 deletions.
170 changes: 170 additions & 0 deletions github/Artifact.py
@@ -0,0 +1,170 @@
############################ 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 _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"]
)
28 changes: 28 additions & 0 deletions github/Artifact.pyi
@@ -0,0 +1,28 @@
import github.WorkflowRun
from datetime import datetime

class Artifact(github.GithubObject.NonCompletableGithubObject):
@property
def archive_download_url(self) -> str: ...
@property
def created_at(self) -> datetime: ...
@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: ...
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
21 changes: 21 additions & 0 deletions tests/Artifact.py
@@ -0,0 +1,21 @@
#!/usr/bin/env python3

from . import Framework


class Artifact(Framework.TestCase):
def setUp(self):
super().setUp()
self.artifact = (
self.g.get_repo("github/vscode-codeql")
.get_workflow_run(160995070)
.get_artifacts()[0]
)

def testAttributes(self):
self.assertEqual(self.artifact.name, "vscode-codeql-extension")

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

0 comments on commit 77ace0a

Please sign in to comment.