utils.py

"""
Misc functions for headshot.py
"""
from modules.console_beautifier import ConsoleOutputBeautifier


def formatted_request(method, host, header, payload):
    return  """

---- REQUEST ----

{} / HTTP/1.1
Host: {}
{}: {}

---- RESPONSE ----
    """.format(method, host, header, payload)


def line_start(fn):
    """
    line start decorator
    """
    def wrapper(*args, **kwargs):
        print "[+]"
        fn(*args, **kwargs)
    return wrapper


def response_description(method, host, resp_size, resp):
    message = "{}[+] Sending {} request to {}\t\t  received {} {} with {} bytes of response {}"

    if resp.status_code != 200:
        return "{}[-] Sending {} request to {}\t\t  response size is {}; HTTP respone status is: {} {}{}".format(ConsoleOutputBeautifier.getColor('red'), method, host, resp_size, resp.status_code, resp.reason, ConsoleOutputBeautifier.getColor('white'))
    else:
        return message.format(
            ConsoleOutputBeautifier.getColor(
                'green'), method, host, resp.status_code,
            resp.reason, resp.headers.get(
                'content-length'),
            ConsoleOutputBeautifier.getColor('white'))