コード例 #1
0
 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"
コード例 #2
0
    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)
コード例 #3
0
ファイル: response.py プロジェクト: SoblaznitelniTsa/VestesPy
	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"
コード例 #4
0
ファイル: response.py プロジェクト: SoblaznitelniTsa/VestesPy
	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)
コード例 #5
0
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
コード例 #6
0
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
コード例 #7
0
ファイル: response.py プロジェクト: SoblaznitelniTsa/VestesPy
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