1 files changed, 59 insertions, 2 deletions
diff --git a/netlib/utils.py b/netlib/utils.py
index 35ea0ec7..31dcd622 100644
--- a/netlib/utils.py
+++ b/netlib/utils.py
@@ -4,6 +4,7 @@ import cgi
 import urllib
 import urlparse
 import string
+import re
 
 
 def isascii(s):
@@ -118,6 +119,7 @@ def pretty_size(size):
 
 
 class Data(object):
+
     def __init__(self, name):
         m = __import__(name)
         dirname, _ = os.path.split(m.__file__)
@@ -136,8 +138,6 @@ class Data(object):
         return fullpath
 
 
-
-
 def is_valid_port(port):
     if not 0 <= port <= 65535:
         return False
@@ -220,6 +220,7 @@ def hostport(scheme, host, port):
     else:
         return "%s:%s" % (host, port)
 
+
 def unparse_url(scheme, host, port, path=""):
     """
         Returns a URL string, constructed from the specified compnents.
@@ -234,8 +235,64 @@ def urlencode(s):
     s = [tuple(i) for i in s]
     return urllib.urlencode(s, False)
 
+
 def urldecode(s):
     """
         Takes a urlencoded string and returns a list of (key, value) tuples.
     """
     return cgi.parse_qsl(s, keep_blank_values=True)
+
+
+def parse_content_type(c):
+    """
+        A simple parser for content-type values. Returns a (type, subtype,
+        parameters) tuple, where type and subtype are strings, and parameters
+        is a dict. If the string could not be parsed, return None.
+
+        E.g. the following string:
+
+            text/html; charset=UTF-8
+
+        Returns:
+
+            ("text", "html", {"charset": "UTF-8"})
+    """
+    parts = c.split(";", 1)
+    ts = parts[0].split("/", 1)
+    if len(ts) != 2:
+        return None
+    d = {}
+    if len(parts) == 2:
+        for i in parts[1].split(";"):
+            clause = i.split("=", 1)
+            if len(clause) == 2:
+                d[clause[0].strip()] = clause[1].strip()
+    return ts[0].lower(), ts[1].lower(), d
+
+
+def multipartdecode(hdrs, content):
+    """
+        Takes a multipart boundary encoded string and returns list of (key, value) tuples.
+    """
+    v = hdrs.get_first("content-type")
+    if v:
+        v = parse_content_type(v)
+        if not v:
+            return []
+        boundary = v[2].get("boundary")
+        if not boundary:
+            return []
+
+        rx = re.compile(r'\bname="([^"]+)"')
+        r = []
+
+        for i in content.split("--" + boundary):
+            parts = i.splitlines()
+            if len(parts) > 1 and parts[0][0:2] != "--":
+                match = rx.search(parts[1])
+                if match:
+                    key = match.group(1)
+                    value = "".join(parts[3 + parts[2:].index(""):])
+                    r.append((key, value))
+        return r
+    return []