def __init__(self, conn, server, request=None):
self._buffer = b""
self.request = request
self.alive = True
self.connection = conn
self.server = server
self.headers = Headers()
self.status = 200
self.protocol = b"HTTP/1.1"
self.charset = "utf-8"
def send_error(self, code, body=None, headers={}):
self.check_self()
if body is not None:
if isinstance(body, str):
body = body.encode("utf-8")
if not isinstance(body, bytes):
raise TypeError(
"res.send_error accepts only str or bytes or None as body!"
)
self.status = code
self.headers = Headers()
for key, value in headers:
self.headers[key] = value
self.send_all(body, buffer=False)
def __init__(self, conn, server, request=None): self._buffer = b"" self.request = request self.alive = True self.connection = conn self.server = server self.headers = Headers() self.status = 200 self.protocol = b"HTTP/1.1" self.charset = "utf-8"
def send_error(self, code, body=None, headers={}):
self.check_self()
if body is not None:
if isinstance(body, str):
body = body.encode("utf-8")
if not isinstance(body, bytes):
raise TypeError("res.send_error accepts only str or bytes or None as body!")
self.status = code
self.headers = Headers()
for key, value in headers:
self.headers[key] = value
self.send_all(body, buffer=False)
def parse_headers(raw):
data = raw.strip().split(CRLF)
head = data.pop(0).split()
method = validators.method(head[0])
url = validators.url(head[1])
query = url[1]
url = url[0]
protocol = validators.protocol(head[2])
headers = Headers()
for line in data:
try:
key, value = line.split(HTTP_HEADER_SEPARATOR)
except Exception:
continue
headers[key.strip()] = value.strip()
return headers, method, url, query, protocol
class Response:
def __init__(self, conn, server, request=None):
self._buffer = b""
self.request = request
self.alive = True
self.connection = conn
self.server = server
self.headers = Headers()
self.status = 200
self.protocol = b"HTTP/1.1"
self.charset = "utf-8"
def check_self(self):
if not self.alive:
raise Exception("Already used that response instance!")
def combine_headers(self):
if not hasattr(self, "_cached_headers"):
if self.status not in http_status.name:
self.status = 500
status = str(self.status) + " " + http_status.name[self.status]
status = status.encode(self.charset)
response = [self.protocol + b" " + status]
for value in self.headers.values():
key = str(value[1]).encode(self.charset)
val = str(value[0]).encode(self.charset)
response.append(key + b":" + val)
response.append(CRLF)
response = CRLF.join(response)
self._cached_headers = response
return self._cached_headers
def send_all(self, data=None, buffer=True):
self.check_self()
if data is not None:
if isinstance(data, str):
data = data.encode(self.charset)
if not isinstance(data, bytes):
raise TypeError(
"res.send_all accepts only None, str or bytes!")
else:
data = b""
if not buffer:
if data is not None:
self.headers["Content-Length"] = len(data)
else:
self.headers["Content-Length"] = 0
response = self.combine_headers()
if data is not None:
response += data
self.send(response)
self.alive = False
else:
self._buffer = data
def send_stream(self, iterable, total):
self.check_self()
total = int(total)
self.headers["Content-Length"] = total
headers = self.combine_headers()
self.send(headers)
current = 0
for data in iterable:
if isinstance(data, str):
data = data.encode(self.charset)
if not isinstance(data, bytes):
raise TypeError(
"res.send_stream accepts only a stream of str or bytes!")
data_length = len(data)
if current + len(data) > total:
data_length = total - current
data = data[:data_length]
self.send(data)
current += data_length
if current == total:
break
self.alive = False
def send_chunked(self, iterable):
self.check_self()
self.headers["Transfer-Encoding"] = "chunked"
headers = self.combine_headers()
self.send(headers)
for data in iterable:
if isinstance(data, str):
data = data.encode(self.charset)
if not isinstance(data, bytes):
raise TypeError(
"res.send_stream accepts only a stream of str or bytes!")
data_length = ("%X" % len(data)).encode("ascii")
data = data_length + CRLF + data + CRLF
self.send(data)
self.send(b"0" + CRLF + CRLF)
self.alive = False
def send_error(self, code, body=None, headers={}):
self.check_self()
if body is not None:
if isinstance(body, str):
body = body.encode("utf-8")
if not isinstance(body, bytes):
raise TypeError(
"res.send_error accepts only str or bytes or None as body!"
)
self.status = code
self.headers = Headers()
for key, value in headers:
self.headers[key] = value
self.send_all(body, buffer=False)
def send(self, data):
try:
self.connection.send(data)
except Exception:
pass
class Response:
def __init__(self, conn, server, request=None):
self._buffer = b""
self.request = request
self.alive = True
self.connection = conn
self.server = server
self.headers = Headers()
self.status = 200
self.protocol = b"HTTP/1.1"
self.charset = "utf-8"
def check_self(self):
if not self.alive:
raise Exception("Already used that response instance!")
def combine_headers(self):
if not hasattr(self, "_cached_headers"):
if self.status not in http_status.name:
self.status = 500
status = str(self.status) + " " + http_status.name[self.status]
status = status.encode(self.charset)
response = [self.protocol + b" " + status]
for value in self.headers.values():
key = str(value[1]).encode(self.charset)
val = str(value[0]).encode(self.charset)
response.append(key+b":"+val)
response.append(CRLF)
response = CRLF.join(response)
self._cached_headers = response
return self._cached_headers
def send_all(self, data=None, buffer=True):
self.check_self()
if data is not None:
if isinstance(data, str):
data = data.encode(self.charset)
if not isinstance(data, bytes):
raise TypeError("res.send_all accepts only None, str or bytes!")
else:
data = b""
if not buffer:
if data is not None:
self.headers["Content-Length"] = len(data)
else:
self.headers["Content-Length"] = 0
response = self.combine_headers()
if data is not None:
response += data
self.send(response)
self.alive = False
else:
self._buffer = data
def send_stream(self, iterable, total):
self.check_self()
total = int(total)
self.headers["Content-Length"] = total
headers = self.combine_headers()
self.send(headers)
current = 0
for data in iterable:
if isinstance(data, str):
data = data.encode(self.charset)
if not isinstance(data, bytes):
raise TypeError("res.send_stream accepts only a stream of str or bytes!")
data_length = len(data)
if current + len(data) > total:
data_length = total - current
data = data[:data_length]
self.send(data)
current += data_length
if current == total:
break
self.alive = False
def send_chunked(self, iterable):
self.check_self()
self.headers["Transfer-Encoding"] = "chunked"
headers = self.combine_headers()
self.send(headers)
for data in iterable:
if isinstance(data, str):
data = data.encode(self.charset)
if not isinstance(data, bytes):
raise TypeError("res.send_stream accepts only a stream of str or bytes!")
data_length = ("%X" % len(data)).encode("ascii")
data = data_length + CRLF + data + CRLF
self.send(data)
self.send(b"0"+CRLF+CRLF)
self.alive = False
def send_error(self, code, body=None, headers={}):
self.check_self()
if body is not None:
if isinstance(body, str):
body = body.encode("utf-8")
if not isinstance(body, bytes):
raise TypeError("res.send_error accepts only str or bytes or None as body!")
self.status = code
self.headers = Headers()
for key, value in headers:
self.headers[key] = value
self.send_all(body, buffer=False)
def send(self, data):
try:
self.connection.send(data)
except Exception:
pass