Port campifre to fetch_url

9 years ago · d6b0cf7bc4
parent 79173ac18d
commit d6b0cf7bc4
1 changed files with 27 additions and 45 deletions
--- a/notification/campfire.py
+++ b/notification/campfire.py
@ -42,7 +42,7 @@ options:
              "vuvuzela", "what", "whoomp", "yeah", "yodel"]

 # informational: requirements for nodes
-requirements: [ urllib2, cgi ]
+requirements: [ ]
 author: "Adam Garside (@fabulops)"
 '''

@ -53,19 +53,10 @@ EXAMPLES = '''
        msg="Task completed ... with feeling."
 '''

+import cgi

 def main():

-    try:
-        import urllib2
-    except ImportError:
-        module.fail_json(msg="urllib2 is required")
-
-    try:
-        import cgi
-    except ImportError:
-        module.fail_json(msg="cgi is required")
-
    module = AnsibleModule(
        argument_spec=dict(
            subscription=dict(required=True),
@ -102,42 +93,33 @@ def main():
    MSTR = "<message><body>%s</body></message>"
    AGENT = "Ansible/1.2"

-    try:
-
-        # Setup basic auth using token as the username
-        pm = urllib2.HTTPPasswordMgrWithDefaultRealm()
-        pm.add_password(None, URI, token, 'X')
-
-        # Setup Handler and define the opener for the request
-        handler = urllib2.HTTPBasicAuthHandler(pm)
-        opener = urllib2.build_opener(handler)
+    # Hack to add basic auth username and password the way fetch_url expects
+    module.params['username'] = token
+    module.params['password'] = 'X'

    target_url = '%s/room/%s/speak.xml' % (URI, room)
+    headers = {'Content-Type': 'application/xml',
+            'User-agent': AGENT}

    # Send some audible notification if requested
    if notify:
-            req = urllib2.Request(target_url, NSTR % cgi.escape(notify))
-            req.add_header('Content-Type', 'application/xml')
-            req.add_header('User-agent', AGENT)
-            response = opener.open(req)
+        response, info = fetch_url(module, target_url, data=NSTR % cgi.escape(notify), headers=headers)
+    if info['status'] != 200:
+        module.fail_json(msg="unable to send msg: '%s', campfire api"
+                            " returned error code: '%s'" %
+                             (notify, info['status']))

    # Send the message
-        req = urllib2.Request(target_url, MSTR % cgi.escape(msg))
-        req.add_header('Content-Type', 'application/xml')
-        req.add_header('User-agent', AGENT)
-        response = opener.open(req)
-
-    except urllib2.HTTPError, e:
-        if not (200 <= e.code < 300):
+    response, info = fetch_url(module, target_url, data=MSTR %cgi.escape(msg), headers=headers)
+    if info['status'] != 200:
        module.fail_json(msg="unable to send msg: '%s', campfire api"
                            " returned error code: '%s'" %
-                                 (msg, e.code))
-
-    except Exception, e:
-        module.fail_json(msg="unable to send msg: %s" % msg)
+                             (msg, info['status']))

    module.exit_json(changed=True, room=room, msg=msg, notify=notify)

 # import module snippets
 from ansible.module_utils.basic import *
+from ansible.module_utils.urls import *
+if __name__ == '__main__':
    main()