66 lines
1.8 KiB
Python
66 lines
1.8 KiB
Python
"""
|
|
Everything specific to the Github API (via PyGithub - https://pygithub.readthedocs.io/en/latest/index.html).
|
|
"""
|
|
|
|
from github import Github, GithubException
|
|
|
|
|
|
def normalize_repo_name(repo):
|
|
"""
|
|
Bring repo to style xxx/yyy
|
|
"""
|
|
prefix = 'https://github.com/'
|
|
if repo.startswith(prefix):
|
|
repo = repo[len(prefix):]
|
|
suffix = '.git'
|
|
if repo.endswith(suffix):
|
|
repo = repo[:-len(suffix)]
|
|
return repo
|
|
|
|
|
|
def repo_get_contributors(repo):
|
|
contributors = []
|
|
c = repo.get_contributors()
|
|
for i in range(c.totalCount):
|
|
contributors.append(c[i])
|
|
return contributors
|
|
|
|
|
|
def retrieve_repo_info(repos, token=None):
|
|
"""
|
|
For a list of Github repos, retrieves repo information.
|
|
|
|
Repos must be have the style xxx/yyy example: "PyGithub/PyGithub"
|
|
"""
|
|
single_repo = isinstance(repos, str)
|
|
if single_repo:
|
|
repos = (repos,)
|
|
result = []
|
|
if token:
|
|
g = Github(token)
|
|
else:
|
|
g = Github()
|
|
for repo in repos:
|
|
repo = normalize_repo_name(repo)
|
|
try:
|
|
# get repo
|
|
r = g.get_repo(repo)
|
|
except GithubException as e:
|
|
raise RuntimeError(e) # TODO what to do if repo does not exist?
|
|
e = {'archived': r.archived, 'contributors': repo_get_contributors(r), 'created': r.created_at, 'description': r.description,
|
|
'forks': r.forks_count, 'language': r.language, 'last modified': r.last_modified, 'name': r.name,
|
|
'open issues count': r.open_issues_count, 'owner': r.owner, 'stars': r.stargazers_count, 'topics': r.get_topics(), 'repo': repo}
|
|
result.append(e)
|
|
if single_repo:
|
|
result = result[0]
|
|
return result
|
|
|
|
|
|
def get_user(login, token=None):
|
|
if token:
|
|
g = Github(token)
|
|
else:
|
|
g = Github()
|
|
user = g.get_user(login)
|
|
return user
|