summaryrefslogtreecommitdiff
path: root/project1/proj1_s4498062/webhttp/parser.py
blob: 7dbb3b5726a00a42e86449de47a7d8b66d5b8de2 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
"""HTTP response and request parsers

This module contains parses for HTTP response and HTTP requests.
"""

import webhttp.message


class RequestParser:
    """Class that parses a HTTP request"""

    def __init__(self):
        """Initialize the RequestParser"""
        self.buff = ''
        
    def parse_requests(self, buff):
        """Parse requests in a buffer

        Args:
            buff (str): the buffer contents received from socket

        Returns:
            list of webhttp.Request
        """
        self.buff += buff
        requests = self.get_requests()
        
        for request in requests:
            http_request = webhttp.message.Request()
            http_request.parse(request)
            yield http_request

    def get_requests(self):
        """Split multiple requests from buffer

        Returns:
            list of str
        """
        requests = self.buff.split('\r\n\r\n')
        self.buff = requests[-1]
        for req in requests[:-1]:
            yield req.lstrip() + '\r\n\r\n'


class ResponseParser:
    """Class that parses a HTTP response"""
    def __init__(self):
        """Initialize the ResponseParser"""
        pass

    def parse_response(self, buff):
        """Parse responses in buffer

        Args:
            buff (str): the buffer contents received from socket

        Returns:
            webhttp.Response
        """
        response = webhttp.message.Response()
        response.parse(buff)
        return response