ansible/community.general
1
0
Fork 0
mirror of https://github.com/ansible-collections/community.general.git synced 2024-09-14 20:13:21 +02:00
Code Releases Activity

Port typetalk to fetch_url

Browse source
This commit is contained in:
Toshio Kuratomi 2015-07-20 23:47:56 -07:00 committed by Matt Clay
parent 91ed01d73e
commit 6ac750f174
1 changed files with 20 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
lib/ansible/modules/extras/notification/typetalk.py
View file

@ -35,21 +35,28 @@ EXAMPLES = '''
import urllib import urllib
import urllib2
try: try:
import json import json
except ImportError: except ImportError:
json = None try:
import simplejson as json
except ImportError:
json = None
def do_request(url, params, headers={}): def do_request(module, url, params, headers=None):
data = urllib.urlencode(params) data = urllib.urlencode(params)
if headers is None:
headers = dict()
headers = dict(headers, **{ headers = dict(headers, **{
'User-Agent': 'Ansible/typetalk module', 'User-Agent': 'Ansible/typetalk module',
}) })
return urllib2.urlopen(urllib2.Request(url, data, headers)) r, info = fetch_url(module, url, data=data, headers=headers)
if info['status'] != 200:
exc = ConnectionError(info['msg'])
exc.code = info['status']
raise exc
return r
def get_access_token(client_id, client_secret): def get_access_token(client_id, client_secret):
params = { params = {
@ -62,7 +69,7 @@ def get_access_token(client_id, client_secret):
return json.load(res)['access_token'] return json.load(res)['access_token']
def send_message(client_id, client_secret, topic, msg): def send_message(module, client_id, client_secret, topic, msg):
""" """
send message to typetalk send message to typetalk
""" """
@ -72,9 +79,9 @@ def send_message(client_id, client_secret, topic, msg):
headers = { headers = {
'Authorization': 'Bearer %s' % access_token, 'Authorization': 'Bearer %s' % access_token,
} }
do_request(url, {'message': msg}, headers) do_request(module, url, {'message': msg}, headers)
return True, {'access_token': access_token} return True, {'access_token': access_token}
except urllib2.HTTPError, e: except ConnectionError, e:
return False, e return False, e
@ -98,7 +105,7 @@ def main():
topic = module.params["topic"] topic = module.params["topic"]
msg = module.params["msg"] msg = module.params["msg"]
res, error = send_message(client_id, client_secret, topic, msg) res, error = send_message(module, client_id, client_secret, topic, msg)
if not res: if not res:
module.fail_json(msg='fail to send message with response code %s' % error.code) module.fail_json(msg='fail to send message with response code %s' % error.code)
@ -107,4 +114,6 @@ def main():
# import module snippets # import module snippets
from ansible.module_utils.basic import * from ansible.module_utils.basic import *
main() from ansible.module_utils.urls import *
if __name__ == '__main__':
main()