GithubClient/client.py at master · bendevera/GithubClient · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
import json
import os
from typing import Dict, List, Tuple

import requests
from dotenv import load_dotenv

load_dotenv()


class GithubClient:
    """ GitHub client to get pull request data. """

    base_url = "https://api.github.com"

    def __init__(self, token: str, owner: str, repo: str):
        self.headers = self.__get_headers(token)
        self.owner = owner
        self.repo = repo

    def __get_headers(self, token: str) -> Dict:
        return {'Authorization': 'token %s' % token}

    def list_pull_requests(self, params: Dict = {"state": "open"}) -> List:
        url = "%s/repos/%s/%s/pulls" % (
            self.base_url, self.owner, self.repo)
        response = requests.get(url, headers=self.headers, params=params)
        return response.json()

    def get_pull_commits(self, number: int) -> List:
        url = "%s/repos/%s/%s/pulls/%s/commits" % (
            self.base_url, self.owner, self.repo, number)
        response = requests.get(url, headers=self.headers)
        return response.json()

    def get_pull_files(self, number: int) -> List:
        url = "%s/repos/%s/%s/pulls/%s/files" % (
            self.base_url, self.owner, self.repo, number)
        response = requests.get(url, headers=self.headers)
        return response.json()

    def get_pull_comments(self, number: int) -> List:
        url = "%s/repos/%s/%s/pulls/%s/comments" % (
            self.base_url, self.owner, self.repo, number)
        response = requests.get(url, headers=self.headers)
        return response.json()

    def create_pull(
            self,
            title: str,
            head: str,
            base: str,
            body: str = "",
            maintainer_can_modify: bool = True,
            draft: bool = False) -> Tuple[bool, Dict]:
        params = {
            "title": title,
            "head": self.owner + ":" + head,
            "base": base,
            "body": body,
            "maintainer_can_modify": maintainer_can_modify,
            "draft": draft
        }
        url = "%s/repos/%s/%s/pulls" % (
            self.base_url, self.owner, self.repo)
        response = requests.post(url, headers=self.headers, json=params)
        return response.status_code == 201, response.json()

    def update_pull(self, number: int, params: Dict) -> Tuple[bool, Dict]:
        url = "%s/repos/%s/%s/pulls/%s" % (
            self.base_url, self.owner, self.repo, number)
        response = requests.patch(url, headers=self.headers, json=params)
        return response.status_code == 200, response.json()

    def is_merged(self, number: int) -> bool:
        url = "%s/repos/%s/%s/pulls/%s/merge" % (
            self.base_url, self.owner, self.repo, number)
        response = requests.get(url, headers=self.headers)
        return response.status_code == 204

    def merge_pull(self, number: int, params: Dict = {}) -> Tuple[bool, str]:
        if params == {}:
            params = {"commit_title": "merging pull id %s" % (number)}
        url = "%s/repos/%s/%s/pulls/%s/merge" % (
            self.base_url, self.owner, self.repo, number)
        response = requests.put(url, headers=self.headers, params=params)
        return response.status_code == 200, response.json()["message"]


def main(token: str, owner: str, repo: str):
    client = GithubClient(token, owner, repo)
    pull_requests = client.list_pull_requests()
    data = []
    for pull in pull_requests:
        commits = client.get_pull_commits(pull["number"])
        files = client.get_pull_files(pull["number"])
        comments = client.get_pull_comments(pull["number"])
        data.append((pull["number"], commits, files, comments))
        print(json.dumps(comments, indent=2))


if __name__ == "__main__":
    TEST_OWNER = "brendajin"
    TEST_REPO = "bendevera"
    GITHUB_TOKEN = str(os.getenv("GITHUB_TOKEN"))
    main(GITHUB_TOKEN, TEST_OWNER, TEST_REPO)