From df8fde4d786e32ed7e8313d63e9a559d5f6137d3 Mon Sep 17 00:00:00 2001 From: James Cammarata Date: Mon, 24 Jul 2017 00:04:11 -0500 Subject: [PATCH] Add cookie parsing to fetch_url/open_url This patch adds cookie parsing to the fetch_url/open_url module_utils method. The overall result will still contain the key `set_cookie`, however an additional key (`cookies`) will also be present. This new field is a dictionary of values. Overall, this should make looking for individual cookies in the response much easier, as currently the `set_cookie` field is an amalgamation of the returned set-cookie headers and can be somewhat difficult to parse. --- CHANGELOG.md | 3 +++ lib/ansible/module_utils/urls.py | 17 +++++++++++++++-- 2 files changed, 18 insertions(+), 2 deletions(-) diff --git a/CHANGELOG.md b/CHANGELOG.md index 36908b4b32..9869cef7b8 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -88,6 +88,9 @@ Ansible Changes By Release parameters to deal with multiple operators. For instance, mode='u=rw+x-X' to set the execute bit on directories, remove it from filea, and set read-write on both is now supported +* Added better cookie parsing to fetch_url/open_url. Cookies are now in a dictionary named `cookies` + in the fetch_url result. Anything using `open_url` directly can pass a cookie object as a named arg + (`cookies`), and then parse/format the cookies in the result. #### New Callbacks: - profile_roles diff --git a/lib/ansible/module_utils/urls.py b/lib/ansible/module_utils/urls.py index 661e4bb159..b9c21704e8 100644 --- a/lib/ansible/module_utils/urls.py +++ b/lib/ansible/module_utils/urls.py @@ -111,6 +111,7 @@ except ImportError: # Python 3 import http.client as httplib +import ansible.module_utils.six.moves.http_cookiejar as cookiejar import ansible.module_utils.six.moves.urllib.request as urllib_request import ansible.module_utils.six.moves.urllib.error as urllib_error from ansible.module_utils.basic import get_distribution, get_exception @@ -813,7 +814,7 @@ def open_url(url, data=None, headers=None, method=None, use_proxy=True, force=False, last_mod_time=None, timeout=10, validate_certs=True, url_username=None, url_password=None, http_agent=None, force_basic_auth=False, follow_redirects='urllib2', - client_cert=None, client_key=None): + client_cert=None, client_key=None, cookies=None): ''' Sends a request via HTTP(S) or FTP using urllib2 (Python2) or urllib (Python3) @@ -907,6 +908,10 @@ def open_url(url, data=None, headers=None, method=None, use_proxy=True, handlers.append(RedirectHandlerFactory(follow_redirects, validate_certs)) + # add some nicer cookie handling + if cookies is not None: + handlers.append(urllib_request.HTTPCookieProcessor(cookies)) + opener = urllib_request.build_opener(*handlers) urllib_request.install_opener(opener) @@ -1028,6 +1033,8 @@ def fetch_url(module, url, data=None, headers=None, method=None, client_cert = module.params.get('client_cert') client_key = module.params.get('client_key') + cookies = cookiejar.LWPCookieJar() + r = None info = dict(url=url) try: @@ -1036,8 +1043,14 @@ def fetch_url(module, url, data=None, headers=None, method=None, validate_certs=validate_certs, url_username=username, url_password=password, http_agent=http_agent, force_basic_auth=force_basic_auth, follow_redirects=follow_redirects, client_cert=client_cert, - client_key=client_key) + client_key=client_key, cookies=cookies) info.update(r.info()) + # parse the cookies into a nice dictionary + cookie_dict = dict() + for cookie in cookies: + cookie_dict[cookie.name] = cookie.value + info['cookies'] = cookie_dict + # finally update the result with a message about the fetch info.update(dict(msg="OK (%s bytes)" % r.headers.get('Content-Length', 'unknown'), url=r.geturl(), status=r.code)) except NoSSLError: e = get_exception()