|
Pierre-Yves Chibon |
dd38bc |
#!/usr/bin/env python
|
|
Pierre-Yves Chibon |
dd38bc |
# -*- coding: utf-8 -*-
|
|
Pierre-Yves Chibon |
dd38bc |
|
|
Pierre-Yves Chibon |
dd38bc |
"""
|
|
Pierre-Yves Chibon |
dd38bc |
(c) 2016 - Copyright Red Hat Inc
|
|
Pierre-Yves Chibon |
dd38bc |
|
|
Pierre-Yves Chibon |
dd38bc |
Authors:
|
|
Pierre-Yves Chibon |
dd38bc |
Pierre-Yves Chibon <pingou@pingoured.fr></pingou@pingoured.fr>
|
|
Pierre-Yves Chibon |
dd38bc |
|
|
Pierre-Yves Chibon |
dd38bc |
|
|
Pierre-Yves Chibon |
dd38bc |
This server listens to message sent via redis and send the corresponding
|
|
Pierre-Yves Chibon |
dd38bc |
web-hook request.
|
|
Pierre-Yves Chibon |
dd38bc |
|
|
Pierre-Yves Chibon |
dd38bc |
Using this mechanism, we no longer block the main application if the
|
|
Pierre-Yves Chibon |
dd38bc |
receiving end is offline or so.
|
|
Pierre-Yves Chibon |
dd38bc |
|
|
Pierre-Yves Chibon |
dd38bc |
"""
|
|
Pierre-Yves Chibon |
dd38bc |
|
|
Pierre-Yves Chibon |
dd38bc |
import json
|
|
Pierre-Yves Chibon |
dd38bc |
import logging
|
|
Pierre-Yves Chibon |
dd38bc |
import os
|
|
Pierre-Yves Chibon |
dd38bc |
import requests
|
|
Pierre-Yves Chibon |
dd38bc |
|
|
Pierre-Yves Chibon |
dd38bc |
import trollius
|
|
Pierre-Yves Chibon |
dd38bc |
import trollius_redis
|
|
Pierre-Yves Chibon |
dd38bc |
|
|
Pierre-Yves Chibon |
dd38bc |
|
|
Pierre-Yves Chibon |
d31d7e |
LOG = logging.getLogger(__name__)
|
|
Pierre-Yves Chibon |
dd38bc |
|
|
Pierre-Yves Chibon |
dd38bc |
if 'PAGURE_CONFIG' not in os.environ \
|
|
Pierre-Yves Chibon |
dd38bc |
and os.path.exists('/etc/pagure/pagure.cfg'):
|
|
Pierre-Yves Chibon |
dd38bc |
print 'Using configuration file `/etc/pagure/pagure.cfg`'
|
|
Pierre-Yves Chibon |
dd38bc |
os.environ['PAGURE_CONFIG'] = '/etc/pagure/pagure.cfg'
|
|
Pierre-Yves Chibon |
dd38bc |
|
|
Pierre-Yves Chibon |
dd38bc |
|
|
Pierre-Yves Chibon |
dd38bc |
import pagure
|
|
Pierre-Yves Chibon |
dd38bc |
import pagure.lib
|
|
Pierre-Yves Chibon |
dd38bc |
|
|
Pierre-Yves Chibon |
dd38bc |
|
|
Pierre-Yves Chibon |
dd38bc |
@trollius.coroutine
|
|
Pierre-Yves Chibon |
dd38bc |
def handle_messages():
|
|
Pierre-Yves Chibon |
d31d7e |
''' Handles connecting to redis and acting upon messages received.
|
|
Pierre-Yves Chibon |
d31d7e |
In this case, it means triggering a build on jenkins based on the
|
|
Pierre-Yves Chibon |
d31d7e |
information provided.
|
|
Pierre-Yves Chibon |
d31d7e |
'''
|
|
Pierre-Yves Chibon |
d31d7e |
|
|
Pierre-Yves Chibon |
4290d0 |
host = pagure.APP.config.get('REDIS_HOST', '0.0.0.0')
|
|
Pierre-Yves Chibon |
4290d0 |
port = pagure.APP.config.get('REDIS_PORT', 6379)
|
|
Pierre-Yves Chibon |
d31d7e |
dbname = pagure.APP.config.get('REDIS_DB', 0)
|
|
Pierre-Yves Chibon |
dd38bc |
connection = yield trollius.From(trollius_redis.Connection.create(
|
|
Pierre-Yves Chibon |
d31d7e |
host=host, port=port, db=dbname))
|
|
Pierre-Yves Chibon |
dd38bc |
|
|
Pierre-Yves Chibon |
dd38bc |
# Create subscriber.
|
|
Pierre-Yves Chibon |
dd38bc |
subscriber = yield trollius.From(connection.start_subscribe())
|
|
Pierre-Yves Chibon |
dd38bc |
|
|
Pierre-Yves Chibon |
dd38bc |
# Subscribe to channel.
|
|
Pierre-Yves Chibon |
dd38bc |
yield trollius.From(subscriber.subscribe(['pagure.ci']))
|
|
Pierre-Yves Chibon |
dd38bc |
|
|
Pierre-Yves Chibon |
dd38bc |
# Inside a while loop, wait for incoming events.
|
|
Pierre-Yves Chibon |
dd38bc |
while True:
|
|
Pierre-Yves Chibon |
dd38bc |
reply = yield trollius.From(subscriber.next_published())
|
|
Pierre-Yves Chibon |
d31d7e |
LOG.info(
|
|
Pierre-Yves Chibon |
dd38bc |
'Received: %s on channel: %s',
|
|
Pierre-Yves Chibon |
dd38bc |
repr(reply.value), reply.channel)
|
|
Pierre-Yves Chibon |
dd38bc |
data = json.loads(reply.value)
|
|
Pierre-Yves Chibon |
dd38bc |
|
|
Pierre-Yves Chibon |
dd38bc |
pr_id = data['pr']['id']
|
|
Pierre-Yves Chibon |
dd38bc |
branch = data['pr']['branch_from']
|
|
Pierre-Yves Chibon |
2ae1ff |
projectname = data['pr']['project']['name']
|
|
Pierre-Yves Chibon |
05271f |
namespace = data['pr']['project'].get('namespace')
|
|
Pierre-Yves Chibon |
dd38bc |
|
|
Pierre-Yves Chibon |
dd38bc |
username = None
|
|
Pierre-Yves Chibon |
dd38bc |
if data['pr'].get('parent'):
|
|
Pierre-Yves Chibon |
feb4b6 |
username = data['pr']['project']['user']['user']
|
|
Pierre-Yves Chibon |
dd38bc |
|
|
Pierre-Yves Chibon |
dd38bc |
project = pagure.lib.get_project(
|
|
Pierre-Yves Chibon |
05271f |
session=pagure.SESSION,
|
|
Pierre-Yves Chibon |
05271f |
name=projectname,
|
|
Pierre-Yves Chibon |
05271f |
user=username,
|
|
Pierre-Yves Chibon |
05271f |
namespace=namespace)
|
|
Pierre-Yves Chibon |
dd38bc |
|
|
Pierre-Yves Chibon |
dd38bc |
if not project:
|
|
Pierre-Yves Chibon |
d31d7e |
LOG.warning(
|
|
Pierre-Yves Chibon |
d31d7e |
'No project could be found from the message %s', data)
|
|
Pierre-Yves Chibon |
dd38bc |
continue
|
|
Pierre-Yves Chibon |
dd38bc |
|
|
Pierre-Yves Chibon |
d31d7e |
LOG.info(
|
|
Pierre-Yves Chibon |
bf6903 |
"Trigger on %s PR #%s from %s: %s",
|
|
Pierre-Yves Chibon |
d31d7e |
project.fullname, pr_id, project.fullname, branch)
|
|
Pierre-Yves Chibon |
dd38bc |
|
|
Pierre-Yves Chibon |
9ef674 |
url = project.ci_hook.ci_url.rstrip('/')
|
|
Pierre-Yves Chibon |
dd38bc |
|
|
Pierre-Yves Chibon |
dd38bc |
if data['ci_type'] == 'jenkins':
|
|
Pierre-Yves Chibon |
9ef674 |
url = url + '/buildWithParameters'
|
|
Pierre-Yves Chibon |
3e1101 |
repo = '%s/%s' % (
|
|
Pierre-Yves Chibon |
3e1101 |
pagure.APP.config['GIT_URL_GIT'].rstrip('/'), project.path)
|
|
Pierre-Yves Chibon |
3e1101 |
LOG.info(
|
|
Pierre-Yves Chibon |
3e1101 |
'Triggering the build at: %s, for repo: %s', url, repo)
|
|
Pierre-Yves Chibon |
bb02dc |
requests.post(
|
|
Pierre-Yves Chibon |
dd38bc |
url,
|
|
Pierre-Yves Chibon |
dd38bc |
data={
|
|
Pierre-Yves Chibon |
dcbd99 |
'token': project.ci_hook.pagure_ci_token,
|
|
Pierre-Yves Chibon |
dd38bc |
'cause': pr_id,
|
|
Pierre-Yves Chibon |
3e1101 |
'REPO': repo,
|
|
Pierre-Yves Chibon |
dd38bc |
'BRANCH': branch
|
|
Pierre-Yves Chibon |
dd38bc |
}
|
|
Pierre-Yves Chibon |
dd38bc |
)
|
|
Pierre-Yves Chibon |
dd38bc |
else:
|
|
Pierre-Yves Chibon |
d31d7e |
LOG.warning('Un-supported CI type')
|
|
Pierre-Yves Chibon |
dd38bc |
|
|
Pierre-Yves Chibon |
d31d7e |
LOG.info('Ready for another')
|
|
Pierre-Yves Chibon |
bb02dc |
|
|
Pierre-Yves Chibon |
dd38bc |
|
|
Pierre-Yves Chibon |
dd38bc |
def main():
|
|
Pierre-Yves Chibon |
d31d7e |
''' Start the main async loop. '''
|
|
Pierre-Yves Chibon |
d31d7e |
|
|
Pierre-Yves Chibon |
dd38bc |
try:
|
|
Pierre-Yves Chibon |
dd38bc |
loop = trollius.get_event_loop()
|
|
Pierre-Yves Chibon |
dd38bc |
tasks = [
|
|
Pierre-Yves Chibon |
dd38bc |
trollius.async(handle_messages()),
|
|
Pierre-Yves Chibon |
dd38bc |
]
|
|
Pierre-Yves Chibon |
dd38bc |
loop.run_until_complete(trollius.wait(tasks))
|
|
Pierre-Yves Chibon |
dd38bc |
loop.run_forever()
|
|
Pierre-Yves Chibon |
dd38bc |
except KeyboardInterrupt:
|
|
Pierre-Yves Chibon |
dd38bc |
pass
|
|
Pierre-Yves Chibon |
dd38bc |
except trollius.ConnectionResetError:
|
|
Pierre-Yves Chibon |
dd38bc |
pass
|
|
Pierre-Yves Chibon |
dd38bc |
|
|
Pierre-Yves Chibon |
d31d7e |
LOG.info("End Connection")
|
|
Pierre-Yves Chibon |
dd38bc |
loop.close()
|
|
Pierre-Yves Chibon |
d31d7e |
LOG.info("End")
|
|
Pierre-Yves Chibon |
dd38bc |
|
|
Pierre-Yves Chibon |
dd38bc |
|
|
Pierre-Yves Chibon |
dd38bc |
if __name__ == '__main__':
|
|
Pierre-Yves Chibon |
dd38bc |
formatter = logging.Formatter(
|
|
Pierre-Yves Chibon |
dd38bc |
"%(asctime)s %(levelname)s [%(module)s:%(lineno)d] %(message)s")
|
|
Pierre-Yves Chibon |
dd38bc |
|
|
Pierre-Yves Chibon |
dd38bc |
# setup console logging
|
|
Pierre-Yves Chibon |
d31d7e |
LOG.setLevel(logging.DEBUG)
|
|
Pierre-Yves Chibon |
d31d7e |
shellhandler = logging.StreamHandler()
|
|
Pierre-Yves Chibon |
d31d7e |
shellhandler.setLevel(logging.DEBUG)
|
|
Pierre-Yves Chibon |
dd38bc |
|
|
Pierre-Yves Chibon |
dd38bc |
aslog = logging.getLogger("asyncio")
|
|
Pierre-Yves Chibon |
dd38bc |
aslog.setLevel(logging.DEBUG)
|
|
Pierre-Yves Chibon |
dd38bc |
|
|
Pierre-Yves Chibon |
d31d7e |
shellhandler.setFormatter(formatter)
|
|
Pierre-Yves Chibon |
d31d7e |
LOG.addHandler(shellhandler)
|
|
Pierre-Yves Chibon |
dd38bc |
main()
|