add Request attribute to Response

This commit is contained in:
Hedy Li 2021-07-26 10:18:33 +08:00
parent f8dcf4642c
commit d6baf888f1
Signed by: hedy
GPG Key ID: B51B5A8D1B176372
2 changed files with 4 additions and 3 deletions

View File

@ -1,6 +1,6 @@
[tool.poetry]
name = "spartan-py"
version = "0.2.6"
version = "0.2.7"
description = "Library for spartan protocol"
authors = ["Hedy Li <hedy@tilde.cafe>"]
packages = [

View File

@ -24,13 +24,14 @@ class Request:
path = self.path.encode("ascii")
sock.send(b"%s %s %d\r\n" % (host, path, len(self.data)))
sock.send(self.data.encode("ascii"))
return Response(sock)
return Response(sock, self)
class Response:
def __init__(self, socket: socket.socket):
def __init__(self, socket: socket.socket, request: Request = None):
self.socket = socket
self.file = self.socket.makefile(mode="rb")
self.request = request
try:
status, meta = self.file.readline(4096).split(maxsplit=1)