51 lines
1.4 KiB
Python
51 lines
1.4 KiB
Python
|
"""
|
||
|
|
"""
|
||
|
|
|
||
|
|
import httplib
|
||
|
|
import json
|
||
|
|
|
||
|
|
from errors import *
|
||
|
|
|
||
|
|
class Connection(object):
|
||
|
|
def __init__(self, host='api.fastly.com', secure=True, port=None, root='',
|
||
|
|
timeout=10.0):
|
||
|
|
self.host = host
|
||
|
|
self.secure = secure
|
||
|
|
self.port = port
|
||
|
|
self.root = root
|
||
|
|
self.timeout = timeout
|
||
|
|
|
||
|
|
self.authenticator = None
|
||
|
|
self.http_conn = None
|
||
|
|
|
||
|
|
def request(self, method, path, body=None, headers={}):
|
||
|
|
if not self.port:
|
||
|
|
self.port = 443 if self.secure else 80
|
||
|
|
|
||
|
|
if self.secure:
|
||
|
|
self.http_conn = httplib.HTTPSConnection(self.host, self.port,
|
||
|
|
timeout=self.timeout)
|
||
|
|
else:
|
||
|
|
self.http_conn = httplib.HTTPConnection(self.host, self.port,
|
||
|
|
timeout=self.timeout)
|
||
|
|
|
||
|
|
if self.authenticator:
|
||
|
|
self.authenticator.add_auth(headers)
|
||
|
|
|
||
|
|
self.http_conn.request(method, self.root + path, body, headers=headers)
|
||
|
|
response = self.http_conn.getresponse()
|
||
|
|
body = response.read()
|
||
|
|
try:
|
||
|
|
data = json.loads(body)
|
||
|
|
except ValueError:
|
||
|
|
data = body
|
||
|
|
|
||
|
|
if response.status == 403:
|
||
|
|
raise AuthenticationError()
|
||
|
|
elif response.status == 500:
|
||
|
|
raise InternalServerError()
|
||
|
|
elif response.status == 400:
|
||
|
|
raise BadRequestError()
|
||
|
|
|
||
|
|
return (response, data)
|