def request(self, method, URL, headers=None, body=""):
"""sending request to server"""
message = ClientMessage(method, URL, headers, body)
message.headers['Cookie'] = str(self.cookieJar)
self.history.add(URL)
try:
self.socket = TCPSocket()
except:
raise Exception("Cannot initiate socket correctly")
try:
self.socket.connect(self.dest)
data = str(message).encode()
while True:
sent = self.socket.sendall(data)
if sent is None:
break
else:
self.socket.connect(self.dest)
except:
raise Exception("connection failed")
try:
response = ServerMessage(self.socket)
except:
raise Exception("empty socket")
self.add_new_cookies(response)
try:
self.socket.close()
except:
raise Exception("Socket cannot close correctly")
return response
def __init__(self, dest):
try:
self.socket = TCPSocket()
except:
raise Exception("Cannot initiate socket correctly")
self.history = set()
self.cookieJar = CookieJar()
self.dest = dest
class MyCurl:
"""Curl to connect server and client"""
def __init__(self, dest):
try:
self.socket = TCPSocket()
except:
raise Exception("Cannot initiate socket correctly")
# self.socket.connect(dest)
self.history = set()
self.cookieJar = CookieJar()
self.dest = dest
def request(self, method, URL, headers=None, body=""):
"""sending request to server"""
message = ClientMessage(method, URL, headers, body)
message.headers['Cookie'] = str(self.cookieJar)
self.history.add(URL)
# try:
# self.socket = TCPSocket()
# except:
# raise Exception("Cannot initiate socket correctly")
try:
self.socket.connect(self.dest)
self.socket.sendall(str(message).encode())
except:
raise Exception("connection failed")
try:
response = ServerMessage(self.socket)
except:
raise Exception("empty socket")
self.add_new_cookies(response)
# try:
# self.socket.close()
# except:
# raise Exception("Socket cannot close correctly")
return response
def get(self, URL, headers={}):
"""sending get request"""
return self.request("GET", URL, headers)
def post(self, URL, headers={}, body=""):
"""sending post request"""
return self.request("POST", URL, headers, body)
def add_new_cookies(self, message):
"""add new coockies to the cookie jar"""
jar = message.cookieJar.getAll()
for key in jar:
self.cookieJar.add_cookie(key, jar[key])
def is_visited_or_Not(self, link):
"""check if the link has been visited"""
return link in self.history
def get_cookie(self, str):
"""get the cookie"""
return self.cookieJar.get_cookie(str)