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

Add default User-Agent

Browse source 
crates.io requires it since https://github.com/rust-lang/crates.io/pull/1696
This commit is contained in:
Chih-Hsuan Yen 2019-07-05 09:20:36 +08:00
parent 8720d59f79
commit 629e82ac8a
No known key found for this signature in database
GPG key ID: 0453A6CA23C56315
4 changed files with 8 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
nvchecker/source/aiohttp_httpclient.py
View file

@ -4,6 +4,8 @@
import atexit
import asyncio
import aiohttp
from .httpclient import DEFAULT_USER_AGENT
connector = aiohttp.TCPConnector(limit=20)
__all__ = ['session', 'HTTPError', 'NetworkErrors']
@ -19,6 +21,8 @@ class BetterClientSession(aiohttp.ClientSession):
if hasattr(self, "nv_config") and self.nv_config.get("proxy"):
kwargs.setdefault("proxy", self.nv_config.get("proxy"))
kwargs.setdefault("headers", {}).setdefault('User-Agent', DEFAULT_USER_AGENT)
res = await super(BetterClientSession, self)._request(
*args, **kwargs)
if res.status >= 400:

1
nvchecker/source/github.py
View file

@ -46,7 +46,6 @@ async def get_version_real(name, conf, **kwargs):
url += '?' + urlencode(parameters)
headers = {
'Accept': 'application/vnd.github.quicksilver-preview+json',
'User-Agent': 'lilydjwg/nvchecker',
}
if 'NVCHECKER_GITHUB_TOKEN' in os.environ:
headers['Authorization'] = 'token %s' % os.environ['NVCHECKER_GITHUB_TOKEN']

1
nvchecker/source/httpclient.py Normal file
View file

@ -0,0 +1 @@
DEFAULT_USER_AGENT = 'lilydjwg/nvchecker'

3
nvchecker/source/tornado_httpclient.py
View file

@ -15,6 +15,8 @@ try:
except ImportError:
pycurl = None
from .httpclient import DEFAULT_USER_AGENT
__all__ = ['session', 'HTTPError', 'NetworkErrors']
client = AsyncHTTPClient()
@ -51,6 +53,7 @@ class Session:
q = urlencode(params)
url += '?' + q
kwargs.setdefault("headers", {}).setdefault('User-Agent', DEFAULT_USER_AGENT)
r = HTTPRequest(url, **kwargs)
return ResponseManager(r)