From 14bd9ebb455ab1fd8a08868b37d4b271658b0dee Mon Sep 17 00:00:00 2001
From: Kenneth Reitz <me@kennethreitz.com>
Date: Mon, 26 Sep 2011 01:33:18 -0400
Subject: [PATCH] yay cookies

---
 requests/api.py    |  2 +-
 requests/models.py | 42 ++++++++++++++++++++++++++++++++++++++++--
 requests/utils.py  |  4 ++--
 3 files changed, 43 insertions(+), 5 deletions(-)

diff --git a/requests/api.py b/requests/api.py
index 653f82d..5e5bc29 100644
--- a/requests/api.py
+++ b/requests/api.py
@@ -49,7 +49,7 @@ def request(method, url,
     if cookies is None:
         cookies = {}
 
-    cookies = cookiejar_from_dict(cookies)
+    # cookies = cookiejar_from_dict(cookies)
 
     # Expand header values
     if headers:
diff --git a/requests/models.py b/requests/models.py
index 406abd5..101ecb4 100644
--- a/requests/models.py
+++ b/requests/models.py
@@ -8,6 +8,7 @@ requests.models
 
 import urllib
 import zlib
+from Cookie import SimpleCookie
 from urlparse import urlparse, urlunparse, urljoin
 
 from .packages import urllib3
@@ -133,6 +134,22 @@ class Request(object):
             # Make headers case-insensitive.
             response.headers = CaseInsensitiveDict(getattr(resp, 'headers', None))
 
+            # Start off with our local cookies.
+            cookies = self.cookies or dict()
+
+            # Add new cookies from the server.
+            if 'set-cookie' in response.headers:
+                cookie_header = response.headers['set-cookie']
+
+                c = SimpleCookie()
+                c.load(cookie_header)
+
+                for k,v in c.items():
+                    cookies.update({k: v.value})
+
+            # Save cookies in Response.
+            response.cookies = cookies
+
             # Save original resopnse for later.
             response.raw = resp
 
@@ -240,6 +257,7 @@ class Request(object):
         # Build the final URL.
         url = build_url(self.url, self.params)
 
+        # Nottin' on you.
         body = None
         content_type = None
 
@@ -296,6 +314,20 @@ class Request(object):
                     # Part of a connection pool, so no fancy stuff. Sorry!
                     do_block = True
 
+                if self.cookies:
+                    # Skip if 'cookie' header is explicitly set.
+                    if 'cookie' not in self.headers:
+
+                        # Simple cookie with our dict.
+                        c = SimpleCookie()
+                        c.load(self.cookies)
+
+                        # Turn it into a header.
+                        cookie_header = c.output(header='').strip()
+
+                        # Attach Cookie header to request.
+                        self.headers['Cookie'] = cookie_header
+
                 # Create the connection.
                 r = connection.urlopen(
                     method=self.method,
@@ -313,7 +345,14 @@ class Request(object):
                     self._pools = pools
 
                 # Extract cookies.
-                # if self.cookiejar is not None:
+                if self.cookies is not None:
+                    pass
+                    # cookies = cookiejar_from_dict(self.cookies)
+#                     >>> C = Cookie.SimpleCookie()
+# >>> C["rocky"] = "road"
+# >>> C["rocky"]["path"] = "/cookie"
+# >>> print C.output(header="Cookie:")
+# Cookie: rocky=road; Path=/cookie
                     # self.cookiejar.extract_cookies(resp, req)
 
             # except (urllib2.HTTPError, urllib2.URLError), why:
@@ -332,7 +371,6 @@ class Request(object):
                 self._build_response(r)
                 self.response.ok = True
 
-
         self.sent = self.response.ok
 
         return self.sent
diff --git a/requests/utils.py b/requests/utils.py
index 30ab901..c422ac5 100644
--- a/requests/utils.py
+++ b/requests/utils.py
@@ -117,7 +117,7 @@ def header_expand(headers):
     return ''.join(collector)
 
 
-def dict_from_cookiejar(cj):
+def dict_from_cookiejar(cookies):
     """Returns a key/value dictionary from a CookieJar.
 
     :param cj: CookieJar object to extract cookies from.
@@ -125,7 +125,7 @@ def dict_from_cookiejar(cj):
 
     cookie_dict = {}
 
-    for _, cookies in cj._cookies.items():
+    for _, cookies in cookies.items():
         for _, cookies in cookies.items():
             for cookie in cookies.values():
                 cookie_dict[cookie.name] = cookie.value
-- 
2.34.1