BioArchLinux/nvchecker
2
0
Fork 0
mirror of https://github.com/lilydjwg/nvchecker.git synced 2025-03-10 06:14:02 +00:00
Code Issues Projects Releases Packages Wiki Activity

feat: add configurable host to github

Browse source
This commit is contained in:
David Arnold 2024-03-02 10:47:50 +01:00
parent e3e8415251
commit 8c82c3c289
No known key found for this signature in database
GPG key ID: AB15A6AF1101390D
1 changed files with 17 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

33
nvchecker_source/github.py
View file

@ -17,11 +17,11 @@ logger = structlog.get_logger(logger_name=__name__)
ALLOW_REQUEST = None ALLOW_REQUEST = None
RATE_LIMITED_ERROR = False RATE_LIMITED_ERROR = False
GITHUB_URL = 'https://api.github.com/repos/%s/commits' GITHUB_URL = 'https://api.%s/repos/%s/commits'
GITHUB_LATEST_RELEASE = 'https://api.github.com/repos/%s/releases/latest' GITHUB_LATEST_RELEASE = 'https://api.%s/repos/%s/releases/latest'
# https://developer.github.com/v3/git/refs/#get-all-references # https://developer.github.com/v3/git/refs/#get-all-references
GITHUB_MAX_TAG = 'https://api.github.com/repos/%s/git/refs/tags' GITHUB_MAX_TAG = 'https://api.%s/repos/%s/git/refs/tags'
GITHUB_GRAPHQL_URL = 'https://api.github.com/graphql' GITHUB_GRAPHQL_URL = 'https://api.%s/graphql'
async def get_version(name, conf, **kwargs): async def get_version(name, conf, **kwargs):
global RATE_LIMITED_ERROR, ALLOW_REQUEST global RATE_LIMITED_ERROR, ALLOW_REQUEST
@ -78,8 +78,8 @@ QUERY_LATEST_RELEASE_WITH_PRERELEASES = '''
}} }}
''' '''
async def get_latest_tag(key: Tuple[str, str, str]) -> RichResult: async def get_latest_tag(key: Tuple[str, str, str, str]) -> RichResult:
repo, query, token = key host, repo, query, token = key
owner, reponame = repo.split('/') owner, reponame = repo.split('/')
headers = { headers = {
'Authorization': f'bearer {token}', 'Authorization': f'bearer {token}',
@ -92,7 +92,7 @@ async def get_latest_tag(key: Tuple[str, str, str]) -> RichResult:
) )
res = await session.post( res = await session.post(
GITHUB_GRAPHQL_URL, GITHUB_GRAPHQL_URL % host,
headers = headers, headers = headers,
json = {'query': q}, json = {'query': q},
) )
@ -108,8 +108,8 @@ async def get_latest_tag(key: Tuple[str, str, str]) -> RichResult:
url = f'https://github.com/{repo}/releases/tag/{version}', url = f'https://github.com/{repo}/releases/tag/{version}',
) )
async def get_latest_release_with_prereleases(key: Tuple[str, str]) -> RichResult: async def get_latest_release_with_prereleases(key: Tuple[str, str, str]) -> RichResult:
repo, token = key host, repo, token = key
owner, reponame = repo.split('/') owner, reponame = repo.split('/')
headers = { headers = {
'Authorization': f'bearer {token}', 'Authorization': f'bearer {token}',
@ -121,7 +121,7 @@ async def get_latest_release_with_prereleases(key: Tuple[str, str]) -> RichResul
) )
res = await session.post( res = await session.post(
GITHUB_GRAPHQL_URL, GITHUB_GRAPHQL_URL % host,
headers = headers, headers = headers,
json = {'query': q}, json = {'query': q},
) )
@ -142,12 +142,13 @@ async def get_version_real(
**kwargs, **kwargs,
) -> VersionResult: ) -> VersionResult:
repo = conf['github'] repo = conf['github']
host = conf.get('host', "github.com")
# Load token from config # Load token from config
token = conf.get('token') token = conf.get('token')
# Load token from keyman # Load token from keyman
if token is None: if token is None:
token = keymanager.get_key('github') token = keymanager.get_key(host.lower(), 'github')
use_latest_tag = conf.get('use_latest_tag', False) use_latest_tag = conf.get('use_latest_tag', False)
if use_latest_tag: if use_latest_tag:
@ -155,7 +156,7 @@ async def get_version_real(
raise GetVersionError('token not given but it is required') raise GetVersionError('token not given but it is required')
query = conf.get('query', '') query = conf.get('query', '')
return await cache.get((repo, query, token), get_latest_tag) # type: ignore return await cache.get((host, repo, query, token), get_latest_tag) # type: ignore
use_latest_release = conf.get('use_latest_release', False) use_latest_release = conf.get('use_latest_release', False)
include_prereleases = conf.get('include_prereleases', False) include_prereleases = conf.get('include_prereleases', False)
@ -163,17 +164,17 @@ async def get_version_real(
if not token: if not token:
raise GetVersionError('token not given but it is required') raise GetVersionError('token not given but it is required')
return await cache.get((repo, token), get_latest_release_with_prereleases) # type: ignore return await cache.get((host, repo, token), get_latest_release_with_prereleases) # type: ignore
br = conf.get('branch') br = conf.get('branch')
path = conf.get('path') path = conf.get('path')
use_max_tag = conf.get('use_max_tag', False) use_max_tag = conf.get('use_max_tag', False)
if use_latest_release: if use_latest_release:
url = GITHUB_LATEST_RELEASE % repo url = GITHUB_LATEST_RELEASE % (host, repo)
elif use_max_tag: elif use_max_tag:
url = GITHUB_MAX_TAG % repo url = GITHUB_MAX_TAG % (host, repo)
else: else:
url = GITHUB_URL % repo url = GITHUB_URL % (host, repo)
parameters = {} parameters = {}
if br: if br:
parameters['sha'] = br parameters['sha'] = br