99 lines
2.4 KiB
Python
99 lines
2.4 KiB
Python
import logging
|
|
from abc import ABC, abstractmethod
|
|
from typing import Dict, Tuple
|
|
from urllib.parse import urlparse
|
|
|
|
from client.httpclient import FORMAT, HTTPClient
|
|
from httplib import parser
|
|
from httplib.exceptions import InvalidResponse, InvalidStatusLine, UnsupportedEncoding
|
|
from httplib.message import Message
|
|
from httplib.retriever import PreambleRetriever
|
|
|
|
|
|
def create(method: str, message: Message):
|
|
|
|
if method == "GET":
|
|
return GetCommand(url, port)
|
|
elif method == "HEAD":
|
|
return HeadCommand(url, port)
|
|
elif method == "POST":
|
|
return PostCommand(url, port)
|
|
elif method == "PUT":
|
|
return PutCommand(url, port)
|
|
else:
|
|
raise ValueError()
|
|
|
|
|
|
class AbstractCommand(ABC):
|
|
path: str
|
|
headers: Dict[str, str]
|
|
|
|
def __init(self):
|
|
pass
|
|
|
|
@property
|
|
@abstractmethod
|
|
def command(self):
|
|
pass
|
|
|
|
|
|
class AbstractWithBodyCommand(AbstractCommand, ABC):
|
|
|
|
def _build_message(self, message: str) -> bytes:
|
|
body = input(f"Enter {self.command} data: ").encode(FORMAT)
|
|
print()
|
|
|
|
message += "Content-Type: text/plain\r\n"
|
|
message += f"Content-Length: {len(body)}\r\n"
|
|
message += "\r\n"
|
|
message = message.encode(FORMAT)
|
|
message += body
|
|
message += b"\r\n"
|
|
|
|
return message
|
|
|
|
|
|
class HeadCommand(AbstractCommand):
|
|
@property
|
|
def command(self):
|
|
return "HEAD"
|
|
|
|
|
|
class GetCommand(AbstractCommand):
|
|
|
|
def __init__(self, uri: str, port, dir=None):
|
|
super().__init__(uri, port)
|
|
self.dir = dir
|
|
self.filename = None
|
|
|
|
@property
|
|
def command(self):
|
|
return "GET"
|
|
|
|
def _get_preamble(self, retriever):
|
|
lines = retriever.retrieve()
|
|
(version, status, msg) = parser.parse_status_line(next(lines))
|
|
headers = parser.parse_headers(lines)
|
|
|
|
logging.debug("---response begin---\r\n%s--- response end---", "".join(retriever.buffer))
|
|
|
|
return Message(version, status, msg, headers)
|
|
|
|
def _await_response(self, client, retriever):
|
|
msg = self._get_preamble(retriever)
|
|
|
|
from client import response_handler
|
|
self.filename = response_handler.handle(client, msg, self, self.dir)
|
|
|
|
|
|
class PostCommand(AbstractWithBodyCommand):
|
|
@property
|
|
def command(self):
|
|
return "POST"
|
|
|
|
|
|
class PutCommand(AbstractWithBodyCommand):
|
|
@property
|
|
def command(self):
|
|
return "PUT"
|