From 521d06019a5911b46d62efa9603802851a194a0a Mon Sep 17 00:00:00 2001 From: yumoqing Date: Mon, 25 May 2020 17:46:50 +0800 Subject: [PATCH] bugfix --- http_client.py | 120 +++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 120 insertions(+) create mode 100644 http_client.py diff --git a/http_client.py b/http_client.py new file mode 100644 index 0000000..0145b1c --- /dev/null +++ b/http_client.py @@ -0,0 +1,120 @@ +import requests + +class NeedLogin(Exception): + pass + +class InsufficientPrivilege(Exception): + pass + +class HTTPError(Exception): + def __init__(self,resp_code): + self.resp_code = resp_code + Exception.__init__(self) + + def __expr__(self): + return f'Exception:return code={self.resp_code}' + + def __str__(self): + return f'Exception:return code={self.resp_code}' + +hostsessions = {} + +class HttpClient: + def __init__(self): + self.s = requests.Session() + + def url2domain(self,url): + parts = url.split('/')[:3] + pre = '/'.join(parts) + return pre + + def _webcall(self,url,method="GET", + params={}, + files={}, + headers={}, + stream=False): + domain = self.url2domain(url) + sessionid = hostsessions.get(domain,None) + if sessionid: + headers.update({'session':sessionid}) + + if method in ['GET']: + req = requests.Request(method,url, + params=params,headers=headers) + else: + req = requests.Request(method,url, + data=params,files=files,headers=headers) + prepped = self.s.prepare_request(req) + resp = self.s.send(prepped) + if resp.status_code == 200: + h = resp.headers.get('Set-Cookie',None) + if h: + sessionid = h.split(';')[0] + hostsessions[domain] = sessionid + + if resp.status_code == 401: + print('NeedLogin:',url) + raise NeedLogin + + if resp.status_code == 403: + raise InsufficientPrivilege + + if resp.status_code != 200: + print('Error', url, method, + params, resp.status_code, + type(resp.status_code)) + raise HTTPError(resp.status_code) + return resp + + def webcall(self,url,method="GET", + params={}, + files={}, + headers={}, + stream=False): + resp = self._webcall(url,method=method, + params=params, + files=files, + headers=headers, + stream=stream) + if stream: + return resp + try: + data = resp.json() + if type(data) != type({}): + return data + status = data.get('status',None) + if status is None: + return data + if status == 'OK': + return data['data'] + return data + except: + return resp.text + + def __call__(self,url,method="GET", + params={}, + headers={}, + files={}, + stream=False): + return self.webcall(url, method=method, + params=params, files=files, headers=headers) + + def get(self, url, params={}, headers={}): + return self.__call__(url,method='GET',params=params, + headers=headers) + def post(self, url, params={}, headers={}, files={}): + return self.__call__(url,method='POST',params=params, files=files, + headers=headers) + + def put(self, url, params={}, headers={}): + return self.__call__(url,method='PUT',params=params, + headers=headers) + + def delete(self, url, params={}, headers={}): + return self.__call__(url,method='DELETE',params=params, + headers=headers) + + def option(self, url, params={}, headers={}): + return self.__call__(url,method='OPTION',params=params, + headers=headers) +