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

aa

Browse source
This commit is contained in:
envolution 2024-11-19 23:01:23 -05:00
parent ec556f6957
commit 74fdf91102
1 changed files with 12 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
nvchecker_source/github-test.py
View file

@ -6,7 +6,7 @@ import asyncio
import json # Added for JSON handling import json # Added for JSON handling
import structlog import structlog
http_client = None
from nvchecker.api import ( from nvchecker.api import (
VersionResult, Entry, AsyncCache, KeyManager, VersionResult, Entry, AsyncCache, KeyManager,
HTTPError, session, RichResult, GetVersionError, HTTPError, session, RichResult, GetVersionError,
@ -21,6 +21,14 @@ 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
global http_client
# Initialize the HTTP client if not already done
if http_client is None:
if asyncio.iscoroutine(session):
http_client = await session
http_client = session
if RATE_LIMITED_ERROR: if RATE_LIMITED_ERROR:
raise RuntimeError('rate limited') raise RuntimeError('rate limited')
@ -115,7 +123,7 @@ async def get_latest_tag(key: Tuple[str, str, str, str]) -> RichResult:
# Make GraphQL query # Make GraphQL query
query_vars = QUERY_GITHUB.replace("$owner", owner).replace("$name", reponame) query_vars = QUERY_GITHUB.replace("$owner", owner).replace("$name", reponame)
async with session.post( async with http_client.post(
GITHUB_GRAPHQL_URL % host, GITHUB_GRAPHQL_URL % host,
headers=headers, headers=headers,
json={'query': query_vars} json={'query': query_vars}
@ -148,7 +156,7 @@ async def get_latest_release_with_prereleases(key: Tuple[str, str, str, str]) ->
# Make GraphQL query # Make GraphQL query
query_vars = QUERY_GITHUB.replace("$owner", owner).replace("$name", reponame) query_vars = QUERY_GITHUB.replace("$owner", owner).replace("$name", reponame)
async with session.post( async with http_client.post(
GITHUB_GRAPHQL_URL % host, GITHUB_GRAPHQL_URL % host,
headers=headers, headers=headers,
json={'query': query_vars} json={'query': query_vars}
@ -205,7 +213,7 @@ async def get_version_real(
# Make GraphQL query # Make GraphQL query
query_vars = QUERY_GITHUB.replace("$owner", owner).replace("$name", reponame) query_vars = QUERY_GITHUB.replace("$owner", owner).replace("$name", reponame)
async with session.post( async with http_client.post(
GITHUB_GRAPHQL_URL % host, GITHUB_GRAPHQL_URL % host,
headers=headers, headers=headers,
json={'query': query_vars} json={'query': query_vars}