From 5bb18810d4a9bad0255628dde7d3d2825ca5113a Mon Sep 17 00:00:00 2001 From: Kenneth Reitz Date: Tue, 16 Aug 2011 22:19:49 -0400 Subject: [PATCH] memory location repr for requests-client --- requests/sessions.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/requests/sessions.py b/requests/sessions.py index 5f62a78f..f8a6883f 100644 --- a/requests/sessions.py +++ b/requests/sessions.py @@ -16,22 +16,26 @@ class Session(object): __attrs__ = ['headers', 'cookies', 'auth', 'timeout', 'proxies'] def __init__(self, **kwargs): + # Set up a CookieJar to be used by default self.cookies = cookielib.FileCookieJar() + # Map args from kwargs to instance-local variables map(lambda k, v: (k in self.__attrs__) and setattr(self, k, v), kwargs.iterkeys(), kwargs.itervalues()) + # Map and wrap requests.api methods self._map_api_methods() def __repr__(self): - return '' % (id(self)) + return '' % (id(self)) def _map_api_methods(self): """Reads each available method from requests.api and decorates them with a wrapper, which inserts any instance-local attributes (from __attrs__) that have been set, combining them with **kwargs. """ + def pass_args(func): def wrapper_func(*args, **kwargs): inst_attrs = dict((k, v) for k, v in self.__dict__.iteritems() @@ -41,6 +45,7 @@ class Session(object): kwargs = dict(inst_attrs.items() + kwargs.items()) return func(*args, **kwargs) return wrapper_func + # Map and decorate each function available in requests.api map(lambda fn: setattr(self, fn, pass_args(getattr(requests.api, fn))), requests.api.__all__)