Page MenuHomec4science
Files F91879319

models.py
No OneTemporary
Actions

File Metadata

Created
Fri, Nov 15, 10:05

models.py
View Options

# -*- coding: utf-8 -*-
##
## This file is part of Invenio.
## Copyright (C) 2014 CERN.
##
## Invenio is free software; you can redistribute it and/or
## modify it under the terms of the GNU General Public License as
## published by the Free Software Foundation; either version 2 of the
## License, or (at your option) any later version.
##
## Invenio is distributed in the hope that it will be useful, but
## WITHOUT ANY WARRANTY; without even the implied warranty of
## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
## General Public License for more details.
##
## You should have received a copy of the GNU General Public License
## along with Invenio; if not, write to the Free Software Foundation, Inc.,
## 59 Temple Place, Suite 330, Boston, MA 02111-1307, USA.
from __future__ import absolute_import
import re
from flask import request, url_for, current_app
from . import signatures
from .registry import receivers_registry
#
# Errors
#
class WebhookError(Exception):
""" General WebHookError """
pass
class ReceiverDoesNotExists(WebhookError):
pass
class InvalidPayload(WebhookError):
pass
class InvalidSignature(WebhookError):
pass
#
# Models
#
class Receiver(object):
"""
Base class for all receivers. A receiver is responsible for receiving and
extracting a payload from a request, and passing it on to a method which
can handle the event notification.
"""
def __init__(self, fn, signature=''):
self._callable = fn
self.signature = signature
@classmethod
def get(cls, receiver_id):
try:
return receivers_registry[receiver_id]
except KeyError:
raise ReceiverDoesNotExists(receiver_id)
@classmethod
def all(cls):
return receivers_registry
@classmethod
def register(cls, receiver_id, receiver):
receivers_registry[receiver_id] = receiver
@classmethod
def unregister(cls, receiver_id):
del receivers_registry[receiver_id]
@classmethod
def get_hook_url(cls, receiver_id, access_token):
cls.get(receiver_id)
# Allow overwriting hook URL in debug mode.
if current_app.debug and \
current_app.config.get('WEBHOOKS_DEBUG_RECEIVER_URLS', None):
url_pattern = current_app.config[
'WEBHOOKS_DEBUG_RECEIVER_URLS'].get(receiver_id, None)
if url_pattern:
return url_pattern % dict(token=access_token)
return url_for(
'receivereventlistresource',
receiver_id='github',
access_token=access_token,
_external=True
)
#
# Instance methods (override if needed)
#
def consume_event(self, user_id):
"""
Consume a webhook event by calling the associated callable
"""
event = self._create_event(user_id)
self._callable(event)
def _create_event(self, user_id):
return Event(
user_id,
payload=self.extract_payload()
)
def check_signature(self):
if not self.signature:
return True
signature_value = request.headers.get(self.signature, None)
if signature_value:
validator = 'check_' + re.sub(r'[-]', '_', self.signature).lower()
check_signature = getattr(signatures, validator)
if check_signature(signature_value, request.data):
return True
return False
def extract_payload(self):
"""
Method to extract payload from request.
"""
if not self.check_signature():
raise InvalidSignature('Invalid Signature')
if request.content_type == 'application/json':
return request.get_json()
elif request.content_type == 'application/x-www-form-urlencoded':
return dict(request.form)
raise InvalidPayload(request.content_type)
class CeleryReceiver(Receiver):
"""
Receiver which will fire a celery task to handle payload instead of running
it synchronously during the request.
"""
def __init__(self, task_callable, signature='', **options):
super(CeleryReceiver, self).__init__(task_callable, signature)
self._task = task_callable
self._options = options
from celery import Task
assert isinstance(self._task, Task)
def consume_event(self, user_id):
event = self._create_event(user_id)
self._task.apply_async(args=[event.__getstate__()], **self._options)
class Event(object):
"""
Represents webhook event data which consists of a payload and a user id.
"""
def __init__(self, user_id=None, payload=None):
self.user_id = user_id
self.payload = payload
def __getstate__(self):
return dict(
user_id=self.user_id,
payload=self.payload,
)
def __setstate__(self, state):
self.user_id = state['user_id']
self.payload = state['payload']

Event Timeline

c4science · Help