mirror of
https://github.com/lilydjwg/nvchecker.git
synced 2025-03-10 06:14:02 +00:00
83 lines
2.1 KiB
Python
83 lines
2.1 KiB
Python
# MIT licensed
|
|
# Copyright (c) 2013-2020 lilydjwg <lilydjwg@gmail.com>, et al.
|
|
|
|
import json
|
|
from urllib.parse import urlencode
|
|
|
|
from tornado.httpclient import AsyncHTTPClient, HTTPRequest, HTTPResponse
|
|
from tornado.httpclient import HTTPError
|
|
|
|
try:
|
|
import pycurl
|
|
AsyncHTTPClient.configure("tornado.curl_httpclient.CurlAsyncHTTPClient", max_clients=20)
|
|
except ImportError:
|
|
pycurl = None
|
|
|
|
from .httpclient import DEFAULT_USER_AGENT
|
|
|
|
__all__ = ['session', 'HTTPError', 'NetworkErrors']
|
|
|
|
HTTP2_AVAILABLE = None if pycurl else False
|
|
|
|
def try_use_http2(curl):
|
|
global HTTP2_AVAILABLE
|
|
if HTTP2_AVAILABLE is None:
|
|
try:
|
|
curl.setopt(pycurl.HTTP_VERSION, 4)
|
|
HTTP2_AVAILABLE = True
|
|
except pycurl.error:
|
|
HTTP2_AVAILABLE = False
|
|
elif HTTP2_AVAILABLE:
|
|
curl.setopt(pycurl.HTTP_VERSION, 4)
|
|
|
|
class Session:
|
|
def post(self, url, **kwargs):
|
|
j = kwargs.pop('json', None)
|
|
if j:
|
|
kwargs['body'] = json.dumps(j)
|
|
return self.get(url, method='POST', **kwargs)
|
|
|
|
def get(self, url, **kwargs):
|
|
kwargs['prepare_curl_callback'] = try_use_http2
|
|
|
|
proxy = kwargs.get('proxy')
|
|
if proxy:
|
|
del kwargs['proxy']
|
|
elif hasattr(self, 'nv_config') and self.nv_config.get('proxy'):
|
|
proxy = self.nv_config.get('proxy')
|
|
if proxy:
|
|
host, port = proxy.rsplit(':', 1)
|
|
kwargs['proxy_host'] = host
|
|
kwargs['proxy_port'] = int(port)
|
|
|
|
params = kwargs.get('params')
|
|
if params:
|
|
del kwargs['params']
|
|
q = urlencode(params)
|
|
url += '?' + q
|
|
|
|
kwargs.setdefault("headers", {}).setdefault('User-Agent', DEFAULT_USER_AGENT)
|
|
r = HTTPRequest(url, **kwargs)
|
|
return ResponseManager(r)
|
|
|
|
class ResponseManager:
|
|
def __init__(self, req):
|
|
self.req = req
|
|
|
|
async def __aenter__(self):
|
|
return await AsyncHTTPClient().fetch(self.req)
|
|
|
|
async def __aexit__(self, exc_type, exc, tb):
|
|
pass
|
|
|
|
async def json_response(self, **kwargs):
|
|
return json.loads(self.body.decode('utf-8'))
|
|
|
|
async def read(self):
|
|
return self.body
|
|
|
|
HTTPResponse.json = json_response
|
|
HTTPResponse.read = read
|
|
session = Session()
|
|
|
|
NetworkErrors = ()
|