Skip to content

Commit 7175675

Browse files
committed
Refactor GitHub client to accept Repository
1 parent 75b4383 commit 7175675

File tree

1 file changed

+15
-15
lines changed

1 file changed

+15
-15
lines changed

src/labels/github.py

Lines changed: 15 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -56,17 +56,17 @@ def __init__(
5656
self.session = requests.Session()
5757
self.session.auth = auth
5858

59-
def list_labels(self, owner: str, repo: str) -> typing.List[Label]:
59+
def list_labels(self, repo: Repository) -> typing.List[Label]:
6060
"""Return the list of Labels from the repository.
6161
6262
GitHub API docs:
6363
https://developer.github.com/v3/issues/labels/#list-all-labels-for-this-repository
6464
"""
6565
logger = logging.getLogger("labels")
66-
logger.debug(f"Requesting labels for {owner}/{repo}")
66+
logger.debug(f"Requesting labels for {repo.owner}/{repo.name}")
6767

6868
response = self.session.get(
69-
f"{self.base_url}/repos/{owner}/{repo}/labels",
69+
f"{self.base_url}/repos/{repo.owner}/{repo.name}/labels",
7070
headers={"Accept": "application/vnd.github.symmetra-preview+json"},
7171
)
7272

@@ -79,17 +79,17 @@ def list_labels(self, owner: str, repo: str) -> typing.List[Label]:
7979

8080
return [Label(**data) for data in response.json()]
8181

82-
def get_label(self, owner: str, repo: str, *, name: str) -> Label:
82+
def get_label(self, repo: Repository, *, name: str) -> Label:
8383
"""Return a single Label from the repository.
8484
8585
GitHub API docs:
8686
https://developer.github.com/v3/issues/labels/#get-a-single-label
8787
"""
8888
logger = logging.getLogger("labels")
89-
logger.debug(f"Requesting label '{name}' for {owner}/{repo}")
89+
logger.debug(f"Requesting label '{name}' for {repo.owner}/{repo.name}")
9090

9191
response = self.session.get(
92-
f"{self.base_url}/repos/{owner}/{repo}/labels/{name}",
92+
f"{self.base_url}/repos/{repo.owner}/{repo.name}/labels/{name}",
9393
headers={"Accept": "application/vnd.github.symmetra-preview+json"},
9494
)
9595

@@ -102,17 +102,17 @@ def get_label(self, owner: str, repo: str, *, name: str) -> Label:
102102

103103
return Label(**response.json())
104104

105-
def create_label(self, owner: str, repo: str, *, label: Label) -> Label:
105+
def create_label(self, repo: Repository, *, label: Label) -> Label:
106106
"""Create a new Label for the repository.
107107
108108
GitHub API docs:
109109
https://developer.github.com/v3/issues/labels/#create-a-label
110110
"""
111111
logger = logging.getLogger("labels")
112-
logger.debug(f"Creating label '{label.name}' for {owner}/{repo}")
112+
logger.debug(f"Creating label '{label.name}' for {repo.owner}/{repo.name}")
113113

114114
response = self.session.post(
115-
f"{self.base_url}/repos/{owner}/{repo}/labels",
115+
f"{self.base_url}/repos/{repo.owner}/{repo.name}/labels",
116116
headers={"Accept": "application/vnd.github.symmetra-preview+json"},
117117
json=label.params_dict,
118118
)
@@ -126,17 +126,17 @@ def create_label(self, owner: str, repo: str, *, label: Label) -> Label:
126126

127127
return Label(**response.json())
128128

129-
def edit_label(self, owner: str, repo: str, *, name: str, label: Label) -> Label:
129+
def edit_label(self, repo: Repository, *, name: str, label: Label) -> Label:
130130
"""Update a GitHub issue label.
131131
132132
GitHub API docs:
133133
https://developer.github.com/v3/issues/labels/#update-a-label
134134
"""
135135
logger = logging.getLogger("labels")
136-
logger.debug(f"Editing label '{name}' for {owner}/{repo}")
136+
logger.debug(f"Editing label '{name}' for {repo.owner}/{repo.name}")
137137

138138
response = self.session.patch(
139-
f"{self.base_url}/repos/{owner}/{repo}/labels/{name}",
139+
f"{self.base_url}/repos/{repo.owner}/{repo.name}/labels/{name}",
140140
headers={"Accept": "application/vnd.github.symmetra-preview+json"},
141141
json=label.params_dict,
142142
)
@@ -150,17 +150,17 @@ def edit_label(self, owner: str, repo: str, *, name: str, label: Label) -> Label
150150

151151
return Label(**response.json())
152152

153-
def delete_label(self, owner: str, repo: str, *, name: str) -> None:
153+
def delete_label(self, repo: Repository, *, name: str) -> None:
154154
"""Delete a GitHub issue label.
155155
156156
GitHub API docs:
157157
https://developer.github.com/v3/issues/labels/#delete-a-label
158158
"""
159159
logger = logging.getLogger("labels")
160-
logger.debug(f"Deleting label '{name}' for {owner}/{repo}")
160+
logger.debug(f"Deleting label '{name}' for {repo.owner}/{repo.name}")
161161

162162
response = self.session.delete(
163-
f"{self.base_url}/repos/{owner}/{repo}/labels/{name}"
163+
f"{self.base_url}/repos/{repo.owner}/{repo.name}/labels/{name}"
164164
)
165165

166166
if response.status_code != 204:

0 commit comments

Comments
 (0)