http_request.py

import re
from http_error import HTTPBadRequestError

class HTTPRequest():
    def __init__(self, data):
        self.set_properties(data)
        self.headers = {}
        self.invalid_request = False
        self.invalid_request_error_message = ""

    def set_properties(self, data):

        lines = data.split(b'\r\n')
        # Check for bare CR in lines
        for line in lines:
            if b'\r' in line:
                raise HTTPBadRequestError

        req_line = lines.pop(0)
        req_line = req_line.split(b' ')
        self.http_method = req_line[0]
        self.path = req_line[1]
        self.http_version = req_line[2]

        headers = {}

        for line in lines:
            if line:
                lst = line.split(b':')
                headers[lst[0]] = lst[1].strip()

        self.headers = headers

    def get_path(self):
        return self.path