st2api.controllers.v1.HooksHolder.remove_hook() - Code Metrics - Inspection of "update CHANGELOG" - Plexxi/st2 - Measure and Improve Code Quality continuously with Scrutinizer

Passed

Push — kale/submit-debug-info ( c0cb8c...f2d693 )

unknown

created 2016-02-24 22:51 UTC

st2api.controllers.v1.HooksHolder.remove_hook() B

↳ Parent: st2api.controllers.v1.HooksHolder

Complexity

Conditions

Size

Total Lines

Duplication

Lines	0
Ratio	0 %

Metric	Value
dl	0
loc	14
rs	8
cc	6

# Licensed to the StackStorm, Inc ('StackStorm') under one or more
# contributor license agreements.  See the NOTICE file distributed with
# this work for additional information regarding copyright ownership.
# The ASF licenses this file to You under the Apache License, Version 2.0
# (the "License"); you may not use this file except in compliance with
# the License.  You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

try:
    import simplejson as json
except ImportError:
    import json

import six
import pecan
import uuid
from pecan import abort
from pecan.rest import RestController
from six.moves.urllib import parse as urlparse
urljoin = urlparse.urljoin

from st2common import log as logging
from st2common.constants.triggers import WEBHOOK_TRIGGER_TYPES
from st2common.models.api.base import jsexpose
from st2common.models.api.trace import TraceContext
import st2common.services.triggers as trigger_service
from st2common.services.triggerwatcher import TriggerWatcher
from st2common.transport.reactor import TriggerDispatcher
from st2common.util.http import parse_content_type_header
from st2common.rbac.types import PermissionType
from st2common.rbac.decorators import request_user_has_webhook_permission

http_client = six.moves.http_client

LOG = logging.getLogger(__name__)

TRACE_TAG_HEADER = 'St2-Trace-Tag'


class HooksHolder(object):
    """
    Maintains a hook to Trigger mapping.
    """
    def __init__(self):
        self._triggers_by_hook = {}

    def __contains__(self, key):
        return key in self._triggers_by_hook

    def add_hook(self, hook, trigger):
        if hook not in self._triggers_by_hook:
            self._triggers_by_hook[hook] = []
        self._triggers_by_hook[hook].append(trigger)

    def remove_hook(self, hook, trigger):
        if hook not in self._triggers_by_hook:
            return False
        remove_index = -1
        for idx, item in enumerate(self._triggers_by_hook[hook]):
            if item['id'] == trigger['id']:
                remove_index = idx
                break
        if remove_index < 0:
            return False
        self._triggers_by_hook[hook].pop(remove_index)
        if not self._triggers_by_hook[hook]:
            del self._triggers_by_hook[hook]
        return True

    def get_triggers_for_hook(self, hook):
        return self._triggers_by_hook.get(hook, [])

    def get_all(self):
        triggers = []
        for values in six.itervalues(self._triggers_by_hook):
            triggers.extend(values)
        return triggers


class WebhooksController(RestController):
    def __init__(self, *args, **kwargs):
        super(WebhooksController, self).__init__(*args, **kwargs)
        self._hooks = HooksHolder()
        self._base_url = '/webhooks/'
        self._trigger_types = WEBHOOK_TRIGGER_TYPES.keys()

        self._trigger_dispatcher = TriggerDispatcher(LOG)
        queue_suffix = self.__class__.__name__
        self._trigger_watcher = TriggerWatcher(create_handler=self._handle_create_trigger,
                                               update_handler=self._handle_update_trigger,
                                               delete_handler=self._handle_delete_trigger,
                                               trigger_types=self._trigger_types,
                                               queue_suffix=queue_suffix,
                                               exclusive=True)
        self._trigger_watcher.start()
        self._register_webhook_trigger_types()

    @jsexpose()
    def get_all(self):
        # Return only the hooks known by this controller.
        return self._hooks.get_all()

    @jsexpose()
    def get_one(self, name):
        triggers = self._hooks.get_triggers_for_hook(name)

        if not triggers:
            abort(http_client.NOT_FOUND)
            return

        # For demonstration purpose return 1st
        return triggers[0]

    @request_user_has_webhook_permission(permission_type=PermissionType.WEBHOOK_SEND)
    @jsexpose(arg_types=[str], status_code=http_client.ACCEPTED)
    def post(self, *args, **kwargs):
        hook = '/'.join(args)  # TODO: There must be a better way to do this.

        # Note: For backward compatibility reasons we default to application/json if content
        # type is not explicitly provided
        content_type = pecan.request.headers.get('Content-Type', 'application/json')
        content_type = parse_content_type_header(content_type=content_type)[0]
        body = pecan.request.body

        try:
            body = self._parse_request_body(content_type=content_type, body=body)
        except Exception as e:
            self._log_request('Failed to parse request body: %s.' % (str(e)), pecan.request)
            msg = 'Failed to parse request body "%s": %s' % (body, str(e))
            return pecan.abort(http_client.BAD_REQUEST, msg)

        headers = self._get_headers_as_dict(pecan.request.headers)
        # If webhook contains a trace-tag use that else create create a unique trace-tag.
        trace_context = self._create_trace_context(trace_tag=headers.pop(TRACE_TAG_HEADER, None),
                                                   hook=hook)

        if hook == 'st2' or hook == 'st2/':
            return self._handle_st2_webhook(body, trace_context=trace_context)

        if not self._is_valid_hook(hook):
            self._log_request('Invalid hook.', pecan.request)
            msg = 'Webhook %s not registered with st2' % hook
            return pecan.abort(http_client.NOT_FOUND, msg)

        triggers = self._hooks.get_triggers_for_hook(hook)
        payload = {}

        payload['headers'] = headers
        payload['body'] = body
        # Dispatch trigger instance for each of the trigger found
        for trigger in triggers:
            self._trigger_dispatcher.dispatch(trigger, payload=payload,
                trace_context=trace_context)

        return body

    def _parse_request_body(self, content_type, body):
        if content_type == 'application/json':
            self._log_request('Parsing request body as JSON', request=pecan.request)
            body = json.loads(body)
        elif content_type in ['application/x-www-form-urlencoded', 'multipart/form-data']:
            self._log_request('Parsing request body as form encoded data', request=pecan.request)
            body = urlparse.parse_qs(body)
        else:
            raise ValueError('Unsupported Content-Type: "%s"' % (content_type))

        return body

    def _handle_st2_webhook(self, body, trace_context):
        trigger = body.get('trigger', None)
        payload = body.get('payload', None)
        if not trigger:
            msg = 'Trigger not specified.'
            return pecan.abort(http_client.BAD_REQUEST, msg)
        self._trigger_dispatcher.dispatch(trigger, payload=payload, trace_context=trace_context)

        return body

    def _is_valid_hook(self, hook):
        # TODO: Validate hook payload with payload_schema.
        return hook in self._hooks

    def _register_webhook_trigger_types(self):
        for trigger_type in WEBHOOK_TRIGGER_TYPES.values():
            trigger_service.create_trigger_type_db(trigger_type)

    def _create_trace_context(self, trace_tag, hook):
        # if no trace_tag then create a unique one
        if not trace_tag:
            trace_tag = 'webhook-%s-%s' % (hook, uuid.uuid4().hex)
        return TraceContext(trace_tag=trace_tag)

    def add_trigger(self, trigger):
        # Note: Permission checking for creating and deleting a webhook is done during rule
        # creation
        url = self._get_normalized_url(trigger)
        LOG.info('Listening to endpoint: %s', urljoin(self._base_url, url))
        self._hooks.add_hook(url, trigger)

    def update_trigger(self, trigger):
        pass

    def remove_trigger(self, trigger):
        # Note: Permission checking for creating and deleting a webhook is done during rule
        # creation
        url = self._get_normalized_url(trigger)

        removed = self._hooks.remove_hook(url, trigger)
        if removed:
            LOG.info('Stop listening to endpoint: %s', urljoin(self._base_url, url))

    def _get_normalized_url(self, trigger):
        """
        remove the trailing and leading / so that the hook url and those coming
        from trigger parameters end up being the same.
        """
        return trigger['parameters']['url'].strip('/')

    def _get_headers_as_dict(self, headers):
        headers_dict = {}
        for key, value in headers.items():
            headers_dict[key] = value
        return headers_dict

    def _log_request(self, msg, request, log_method=LOG.debug):
        headers = self._get_headers_as_dict(request.headers)
        body = str(request.body)
        log_method('%s\n\trequest.header: %s.\n\trequest.body: %s.', msg, headers, body)

    ##############################################
    # Event handler methods for the trigger events
    ##############################################

    def _handle_create_trigger(self, trigger):
        LOG.debug('Calling "add_trigger" method (trigger.type=%s)' % (trigger.type))
        trigger = self._sanitize_trigger(trigger=trigger)
        self.add_trigger(trigger=trigger)

    def _handle_update_trigger(self, trigger):
        LOG.debug('Calling "update_trigger" method (trigger.type=%s)' % (trigger.type))
        trigger = self._sanitize_trigger(trigger=trigger)
        self.update_trigger(trigger=trigger)

    def _handle_delete_trigger(self, trigger):
        LOG.debug('Calling "remove_trigger" method (trigger.type=%s)' % (trigger.type))
        trigger = self._sanitize_trigger(trigger=trigger)
        self.remove_trigger(trigger=trigger)

    def _sanitize_trigger(self, trigger):
        sanitized = trigger._data
class MyParent:
    def __init__(self):
        self._x = 1;
        self.y = 2;

class MyChild(MyParent):
    def some_method(self):
        return self._x    # Ok, since accessed from a child class

class AnotherClass:
    def some_method(self, instance_of_my_child):
        return instance_of_my_child._x   # Would be flagged as AnotherClass is not
                                         # a child class of MyParent
        if 'id' in sanitized:
            # Friendly objectid rather than the MongoEngine representation.
            sanitized['id'] = str(sanitized['id'])
        return sanitized


1			# Licensed to the StackStorm, Inc ('StackStorm') under one or more
2			# contributor license agreements. See the NOTICE file distributed with
3			# this work for additional information regarding copyright ownership.
4			# The ASF licenses this file to You under the Apache License, Version 2.0
5			# (the "License"); you may not use this file except in compliance with
6			# the License. You may obtain a copy of the License at
7			#
8			# http://www.apache.org/licenses/LICENSE-2.0
9			#
10			# Unless required by applicable law or agreed to in writing, software
11			# distributed under the License is distributed on an "AS IS" BASIS,
12			# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13			# See the License for the specific language governing permissions and
14			# limitations under the License.
15
16			try:
17			import simplejson as json
18			except ImportError:
19			import json
20
21			import six
22			import pecan
23			import uuid
24			from pecan import abort
25			from pecan.rest import RestController
26			from six.moves.urllib import parse as urlparse
27			urljoin = urlparse.urljoin
28
29			from st2common import log as logging
30			from st2common.constants.triggers import WEBHOOK_TRIGGER_TYPES
31			from st2common.models.api.base import jsexpose
32			from st2common.models.api.trace import TraceContext
33			import st2common.services.triggers as trigger_service
34			from st2common.services.triggerwatcher import TriggerWatcher
35			from st2common.transport.reactor import TriggerDispatcher
36			from st2common.util.http import parse_content_type_header
37			from st2common.rbac.types import PermissionType
38			from st2common.rbac.decorators import request_user_has_webhook_permission
39
40			http_client = six.moves.http_client
41
42			LOG = logging.getLogger(__name__)
43
44			TRACE_TAG_HEADER = 'St2-Trace-Tag'
45
46
47			class HooksHolder(object):
48			"""
49			Maintains a hook to Trigger mapping.
50			"""
51			def __init__(self):
52			self._triggers_by_hook = {}
53
54			def __contains__(self, key):
55			return key in self._triggers_by_hook
56
57			def add_hook(self, hook, trigger):
58			if hook not in self._triggers_by_hook:
59			self._triggers_by_hook[hook] = []
60			self._triggers_by_hook[hook].append(trigger)
61
62			def remove_hook(self, hook, trigger):
63			if hook not in self._triggers_by_hook:
64			return False
65			remove_index = -1
66			for idx, item in enumerate(self._triggers_by_hook[hook]):
67			if item['id'] == trigger['id']:
68			remove_index = idx
69			break
70			if remove_index < 0:
71			return False
72			self._triggers_by_hook[hook].pop(remove_index)
73			if not self._triggers_by_hook[hook]:
74			del self._triggers_by_hook[hook]
75			return True
76
77			def get_triggers_for_hook(self, hook):
78			return self._triggers_by_hook.get(hook, [])
79
80			def get_all(self):
81			triggers = []
82			for values in six.itervalues(self._triggers_by_hook):
83			triggers.extend(values)
84			return triggers
85
86
87			class WebhooksController(RestController):
88			def __init__(self, args, *kwargs):
89			super(WebhooksController, self).__init__(args, *kwargs)
90			self._hooks = HooksHolder()
91			self._base_url = '/webhooks/'
92			self._trigger_types = WEBHOOK_TRIGGER_TYPES.keys()
93
94			self._trigger_dispatcher = TriggerDispatcher(LOG)
95			queue_suffix = self.__class__.__name__
96			self._trigger_watcher = TriggerWatcher(create_handler=self._handle_create_trigger,
97			update_handler=self._handle_update_trigger,
98			delete_handler=self._handle_delete_trigger,
99			trigger_types=self._trigger_types,
100			queue_suffix=queue_suffix,
101			exclusive=True)
102			self._trigger_watcher.start()
103			self._register_webhook_trigger_types()
104
105			@jsexpose()
106			def get_all(self):
107			# Return only the hooks known by this controller.
108			return self._hooks.get_all()
109
110			@jsexpose()
111			def get_one(self, name):
112			triggers = self._hooks.get_triggers_for_hook(name)
113
114			if not triggers:
115			abort(http_client.NOT_FOUND)
116			return
117
118			# For demonstration purpose return 1st
119			return triggers[0]
120
121			@request_user_has_webhook_permission(permission_type=PermissionType.WEBHOOK_SEND)
122			@jsexpose(arg_types=[str], status_code=http_client.ACCEPTED)
123			def post(self, args, *kwargs):
124			hook = '/'.join(args) # TODO: There must be a better way to do this.
125
126			# Note: For backward compatibility reasons we default to application/json if content
127			# type is not explicitly provided
128			content_type = pecan.request.headers.get('Content-Type', 'application/json')
129			content_type = parse_content_type_header(content_type=content_type)[0]
130			body = pecan.request.body
131
132			try:
133			body = self._parse_request_body(content_type=content_type, body=body)
134			except Exception as e:
135			self._log_request('Failed to parse request body: %s.' % (str(e)), pecan.request)
136			msg = 'Failed to parse request body "%s": %s' % (body, str(e))
137			return pecan.abort(http_client.BAD_REQUEST, msg)
138
139			headers = self._get_headers_as_dict(pecan.request.headers)
140			# If webhook contains a trace-tag use that else create create a unique trace-tag.
141			trace_context = self._create_trace_context(trace_tag=headers.pop(TRACE_TAG_HEADER, None),
142			hook=hook)
143
144			if hook == 'st2' or hook == 'st2/':
145			return self._handle_st2_webhook(body, trace_context=trace_context)
146
147			if not self._is_valid_hook(hook):
148			self._log_request('Invalid hook.', pecan.request)
149			msg = 'Webhook %s not registered with st2' % hook
150			return pecan.abort(http_client.NOT_FOUND, msg)
151
152			triggers = self._hooks.get_triggers_for_hook(hook)
153			payload = {}
154
155			payload['headers'] = headers
156			payload['body'] = body
157			# Dispatch trigger instance for each of the trigger found
158			for trigger in triggers:
159			self._trigger_dispatcher.dispatch(trigger, payload=payload,
160			trace_context=trace_context)
161
162			return body
163
164			def _parse_request_body(self, content_type, body):
165			if content_type == 'application/json':
166			self._log_request('Parsing request body as JSON', request=pecan.request)
167			body = json.loads(body)
168			elif content_type in ['application/x-www-form-urlencoded', 'multipart/form-data']:
169			self._log_request('Parsing request body as form encoded data', request=pecan.request)
170			body = urlparse.parse_qs(body)
171			else:
172			raise ValueError('Unsupported Content-Type: "%s"' % (content_type))
173
174			return body
175
176			def _handle_st2_webhook(self, body, trace_context):
177			trigger = body.get('trigger', None)
178			payload = body.get('payload', None)
179			if not trigger:
180			msg = 'Trigger not specified.'
181			return pecan.abort(http_client.BAD_REQUEST, msg)
182			self._trigger_dispatcher.dispatch(trigger, payload=payload, trace_context=trace_context)
183
184			return body
185
186			def _is_valid_hook(self, hook):
187			# TODO: Validate hook payload with payload_schema.
188			return hook in self._hooks
189
190			def _register_webhook_trigger_types(self):
191			for trigger_type in WEBHOOK_TRIGGER_TYPES.values():
192			trigger_service.create_trigger_type_db(trigger_type)
193
194			def _create_trace_context(self, trace_tag, hook):
195			# if no trace_tag then create a unique one
196			if not trace_tag:
197			trace_tag = 'webhook-%s-%s' % (hook, uuid.uuid4().hex)
198			return TraceContext(trace_tag=trace_tag)
199
200			def add_trigger(self, trigger):
201			# Note: Permission checking for creating and deleting a webhook is done during rule
202			# creation
203			url = self._get_normalized_url(trigger)
204			LOG.info('Listening to endpoint: %s', urljoin(self._base_url, url))
205			self._hooks.add_hook(url, trigger)
206
207			def update_trigger(self, trigger):
208			pass
209
210			def remove_trigger(self, trigger):
211			# Note: Permission checking for creating and deleting a webhook is done during rule
212			# creation
213			url = self._get_normalized_url(trigger)
214
215			removed = self._hooks.remove_hook(url, trigger)
216			if removed:
217			LOG.info('Stop listening to endpoint: %s', urljoin(self._base_url, url))
218
219			def _get_normalized_url(self, trigger):
220			"""
221			remove the trailing and leading / so that the hook url and those coming
222			from trigger parameters end up being the same.
223			"""
224			return trigger['parameters']['url'].strip('/')
225
226			def _get_headers_as_dict(self, headers):
227			headers_dict = {}
228			for key, value in headers.items():
229			headers_dict[key] = value
230			return headers_dict
231
232			def _log_request(self, msg, request, log_method=LOG.debug):
233			headers = self._get_headers_as_dict(request.headers)
234			body = str(request.body)
235			log_method('%s\n\trequest.header: %s.\n\trequest.body: %s.', msg, headers, body)
236
237			##############################################
238			# Event handler methods for the trigger events
239			##############################################
240
241			def _handle_create_trigger(self, trigger):
242			LOG.debug('Calling "add_trigger" method (trigger.type=%s)' % (trigger.type))
243			trigger = self._sanitize_trigger(trigger=trigger)
244			self.add_trigger(trigger=trigger)
245
246			def _handle_update_trigger(self, trigger):
247			LOG.debug('Calling "update_trigger" method (trigger.type=%s)' % (trigger.type))
248			trigger = self._sanitize_trigger(trigger=trigger)
249			self.update_trigger(trigger=trigger)
250
251			def _handle_delete_trigger(self, trigger):
252			LOG.debug('Calling "remove_trigger" method (trigger.type=%s)' % (trigger.type))
253			trigger = self._sanitize_trigger(trigger=trigger)
254			self.remove_trigger(trigger=trigger)
255
256			def _sanitize_trigger(self, trigger):
257			sanitized = trigger._data
			0 ignored issues – show Coding Style Best Practice introduced 2016-01-14 18:36 UTC by Report Bug Copy Issue Report It seems like `_data` was declared protected and should not be accessed from this context. Prefixing a member variable `_` is usually regarded as the equivalent of declaring it with protected visibility that exists in other languages. Consequentially, such a member should only be accessed from the same class or a child class: class MyParent: def __init__(self): self._x = 1; self.y = 2; class MyChild(MyParent): def some_method(self): return self._x # Ok, since accessed from a child class class AnotherClass: def some_method(self, instance_of_my_child): return instance_of_my_child._x # Would be flagged as AnotherClass is not # a child class of MyParent Loading history...
258			if 'id' in sanitized:
259			# Friendly objectid rather than the MongoEngine representation.
260			sanitized['id'] = str(sanitized['id'])
261			return sanitized
262

Plexxi / st2

GitHub Access Token became invalid

Push — kale/submit-debug-info ( c0cb8c...f2d693 )

st2api.controllers.v1.HooksHolder.remove_hook() B

Complexity

Size

Duplication

Duplication Side-by-Side

Filter issues like