-
Notifications
You must be signed in to change notification settings - Fork 0
/
DataManager.py
105 lines (92 loc) · 3.18 KB
/
DataManager.py
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
from typing import List
import requests
gitHubApiGraphQLQuery = """
query userInfo
{
viewer
{
name
login
contributionsCollection
{
totalCommitContributions
}
repositoriesContributedTo(first: 1, contributionTypes: [COMMIT, ISSUE, PULL_REQUEST, REPOSITORY])
{
totalCount
}
pullRequests(first: 1)
{
totalCount
}
issues(first: 1)
{
totalCount
}
repositories(first: 100, ownerAffiliations: OWNER, isFork: false, orderBy: {direction: DESC, field: STARGAZERS})
{
totalCount
nodes
{
stargazers
{
totalCount
}
name
languages(first: 10, orderBy: {field: SIZE, direction: DESC})
{
edges
{
size
node
{
name
}
}
}
}
}
}
}
"""
class GitHubStatsData:
def __init__(self, gitHubLogin: str, gitHubToken: str):
self.m_token = gitHubToken
self.m_login = gitHubLogin
self.downloadData()
def downloadData(self):
url = "https://api.github.com/graphql"
headers = {"Authorization": f"Bearer {self.m_token}"}
request = requests.post(url, json={"query": gitHubApiGraphQLQuery}, headers=headers)
requestJson = request.json()["data"]["viewer"]
self.m_pullRequestsCount = requestJson["pullRequests"]["totalCount"]
self.m_issuesCount = requestJson["issues"]["totalCount"]
self.m_contributedToCount = requestJson["repositoriesContributedTo"]["totalCount"]
self.m_starsCount = 0
self.m_languages = {}
for node in requestJson["repositories"]["nodes"]:
self.m_starsCount += node["stargazers"]["totalCount"]
languages = node["languages"]["edges"]
for language in languages:
size = language["size"]
name = language["node"]["name"]
if name not in self.m_languages:
self.m_languages[name] = 0
self.m_languages[name] += size
self.m_languages = sorted(self.m_languages.items(), key=lambda x:x[1], reverse=True)
self.m_languages = dict(self.m_languages)
url = f"https://api.github.com/search/commits?q=author:{self.m_login}"
request = requests.get(url)
self.m_commitsCount = request.json()["total_count"]
def computeMostUsedLanguagesByPercent(self, ignoredLanguages: List[str] = {}):
totalSize = 0
mostUsedLanguages = {}
for name, size in self.m_languages.items():
if name.lower() in map(str.lower, ignoredLanguages):
continue
mostUsedLanguages[name] = size
totalSize += size
for name, size in mostUsedLanguages.items():
percent = size / totalSize
mostUsedLanguages[name] = percent * 100
return mostUsedLanguages