forked from pyvec/python.cz
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathgithub.py
More file actions
169 lines (131 loc) · 4.9 KB
/
Copy pathgithub.py
File metadata and controls
169 lines (131 loc) · 4.9 KB
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
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
import os
import itertools
import warnings
from datetime import datetime
import requests
from cachelib import FileSystemCache
from pythoncz import app
__all__ = ('get_issues',)
get_issues_graphql_filename = (
os.path.join(os.path.dirname(__file__), 'github_get_issues.graphql')
)
with open(get_issues_graphql_filename) as f:
GET_ISSUES_GRAPHQL = f.read()
SIX_HOURS_AS_SECONDS = 21600
cache = FileSystemCache(app.config['CACHE_DIR'],
default_timeout=SIX_HOURS_AS_SECONDS)
def get_issues(org_names):
issues = cache.get('github-issues')
if issues is None:
session = _create_api_session()
issues = itertools.chain(*(
_get_issues_for_org(session, org_name) for org_name in org_names)
)
issues = _sort_issues(issues)
cache.set('github-issues', issues)
return issues
def _create_api_session():
user_agent = ('pythoncz/{now.year}-{now.month} '
'(+https://python.cz)').format(now=datetime.now())
session = requests.Session()
token = app.config['GITHUB_TOKEN']
if not token:
raise Exception('GITHUB_TOKEN not configured')
session.headers.update({
'User-Agent': user_agent,
'Authorization': 'token {}'.format(token),
})
return session
def _get_issues_for_org(session, org_name):
json = _request_api(session, GET_ISSUES_GRAPHQL, {'org_name': org_name})
try:
organization = json['data']['organization']
repositories = _get_nodes(organization, 'repositories')
for repository in repositories:
if repository['isPrivate']:
continue
for issue in _get_nodes(repository, 'issues'):
yield _format_issue(org_name, repository, issue)
for pull_request in _get_nodes(repository, 'pullRequests'):
yield _format_issue(org_name, repository, pull_request,
is_pull_request=True)
except KeyError as e:
raise ValueError(
'Unexpected structure of the GitHub API response: {}'.format(e)
)
def _request_api(session, query, variables):
res = session.post('https://api.github.com/graphql', json={
'query': query,
'variables': variables,
})
try:
json = res.json()
except ValueError as e:
res.raise_for_status()
raise ValueError(
'Unexpected structure of the GitHub API response: {}'.format(e)
)
if json.get('errors'):
message = '; '.join(error['message'] for error in json['errors'])
raise requests.HTTPError(message, response=res)
res.raise_for_status()
return json
def _format_issue(org_name, repository, issue, is_pull_request=False):
author = issue['author'] or {
'login': None,
'url': 'https://github.com/ghost',
}
labels = [label['name'] for label in _get_nodes(issue, 'labels')]
return {
'title': issue['title'],
'html_url': issue['url'],
'updated_at': issue['updatedAt'],
'user': {
'login': author['login'],
'html_url': author['url'],
},
'is_pull_request': is_pull_request,
'repository_name': repository['name'],
'repository_url_html': repository['url'],
'organization_name': org_name,
'comments': issue['comments']['totalCount'],
'participants': issue['participants']['totalCount'],
'votes': _calculate_votes(issue),
'labels': labels,
'coach': 'coach' in labels,
'sprint-idea': 'sprint-idea' in labels,
}
def _calculate_votes(issue):
"""Get votes per issue from counts of Reactions
A THUMBS_DOWN and CONFUSED counts as "-1", the rest as "+1".
"""
# First, count *all* reactions as "+1"
score = issue['reactions']['totalCount']
# Then change the negative ones from "+1" to "-1"
# (subtract 2 for each negative reaction)
score -= issue['thumbs_down']['totalCount'] * 2
score -= issue['confuseds']['totalCount'] * 2
return score
def _get_nodes(node, connection_name):
connection_nodes = node[connection_name]['nodes']
if 'totalCount' in node[connection_name]:
connection_nodes_count = len(connection_nodes)
connection_nodes_total_count = node[connection_name]['totalCount']
if connection_nodes_count < connection_nodes_total_count:
warnings.warn((
"The '{}' contain {} nodes in total, but only {} was fetched"
).format(
connection_name,
connection_nodes_total_count,
connection_nodes_count,
), UserWarning)
return connection_nodes
def _sort_issues(issues):
return sorted(issues, key=_get_issue_sort_key, reverse=True)
def _get_issue_sort_key(issue):
return (
issue['coach'],
issue['votes'],
issue['comments'] + issue['participants'],
issue['updated_at'],
)