#coding:utf-8

"""
    convert "Tamper Data" -> python dict

    usefull to get unittest data ;)

    firefox "Tamper Data" addon:
    https://addons.mozilla.org/de/firefox/addon/966

    http://jensdiemer.de/de/Blog/147/tamper-data-python-dict/
"""

from pprint import pprint

try:
    from urlparse import parse_qs
except ImportError:
    from cgi import parse_qs


def evaluate_tamperdata(tamperdata):
    tamperdata = tamperdata.strip()
    query_string = tamperdata.split("=",1)[-1] # cut e.g. POSTDATA

    result = {}
    data = parse_qs(query_string, keep_blank_values=True)
    for key, raw_values in data.iteritems():
        values = []
        for value in raw_values:
            if value.isdigit():
                value = int(value)

            values.append(value)

        if len(values)>1:
            result[key] = values
        else:
            result[key] = values[0]

    return result


def evaluate_multipart_tamperdata(tamperdata):
    tamperdata = tamperdata.strip()
    query_string = tamperdata.split("=",1)[-1] # cut e.g. POSTDATA

    result = {}
    for line in query_string.splitlines():
        line = line.strip()
        if not line or line.startswith("---"):
            continue

        if line.startswith("Content-Disposition: form-data; name="):
            key = line[38:-1]
        else:
            result[key] = line

    return result


#-----------------------------------------------------------------------------


tamper_postdata="""
POSTDATA =-----------------------------142200474478686960433644084
Content-Disposition: form-data; name="csrfmiddlewaretoken"

53dba2f54f765f1f5ef5c999c4809233
-----------------------------142200474478686960433644084
Content-Disposition: form-data; name="foo"

bar

"""
pprint(evaluate_multipart_tamperdata(tamper_postdata))


#-----------------------------------------------------------------------------


tamper_postdata="""
POSTDATA=save=save&foo=bar
"""
pprint(evaluate_tamperdata(tamper_postdata))

{'csrfmiddlewaretoken': '53dba2f54f765f1f5ef5c999c4809233', 'foo': 'bar'}
{'foo': 'bar', 'save': 'save'}

deb http://debian.fladi.at/ sid main
deb-src http://debian.fladi.at/ sid main