listenermethods.py

Go to the documentation of this file.

#  Copyright 2008-2015 Nokia Networks
#  Copyright 2016-     Robot Framework Foundation
#
#  Licensed 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.
 
from robot.errors import TimeoutError
from robot.utils import get_error_details
 
from .listenerarguments import ListenerArguments
from .logger import LOGGER
 
 
class ListenerMethods:
 
    def __init__(self, method_name, listeners):
        self._methods_methods = []
        self._method_name_method_name = method_name
        if listeners:
            self._register_methods_register_methods(method_name, listeners)
 
    def _register_methods(self, method_name, listeners):
        for listener in listeners:
            method = getattr(listener, method_name)
            if method:
                self._methods_methods.append(ListenerMethod(method, listener))
 
    def __call__(self, *args):
        if self._methods_methods:
            args = ListenerArguments.by_method_name(self._method_name_method_name, args)
            for method in self._methods_methods:
                method(args.get_arguments(method.version))
 
    def __bool__(self):
        return bool(self._methods_methods)
 
 
class LibraryListenerMethods:
 
    def __init__(self, method_name):
        self._method_stack_method_stack = []
        self._method_name_method_name = method_name
 
    def new_suite_scope(self):
        self._method_stack_method_stack.append([])
 
    def discard_suite_scope(self):
        self._method_stack_method_stack.pop()
 
    def register(self, listeners, library):
        methods = self._method_stack_method_stack[-1]
        for listener in listeners:
            method = getattr(listener, self._method_name_method_name)
            if method:
                info = ListenerMethod(method, listener, library)
                methods.append(info)
 
    def unregister(self, library):
        methods = [m for m in self._method_stack_method_stack[-1] if m.library is not library]
        self._method_stack_method_stack[-1] = methods
 
    def __call__(self, *args, **conf):
        methods = self._get_methods_get_methods(**conf)
        if methods:
            args = ListenerArguments.by_method_name(self._method_name_method_name, args)
            for method in methods:
                method(args.get_arguments(method.version))
 
    def _get_methods(self, library=None):
        if not (self._method_stack_method_stack and self._method_stack_method_stack[-1]):
            return []
        methods = self._method_stack_method_stack[-1]
        if library:
            return [m for m in methods if m.library is library]
        return methods
 
 
class ListenerMethod:
    # Flag to avoid recursive listener calls.
    called = False
 
    def __init__(self, method, listener, library=None):
        self.methodmethod = method
        self.listener_namelistener_name = listener.name
        self.versionversion = listener.version
        self.librarylibrary = library
 
    def __call__(self, args):
        if self.calledcalled:
            return
        try:
            ListenerMethod.called = True
            self.methodmethod(*args)
        except TimeoutError:
            # Propagate possible timeouts:
            # https://github.com/robotframework/robotframework/issues/2763
            raise
        except:
            message, details = get_error_details()
            LOGGER.error("Calling method '%s' of listener '%s' failed: %s"
                         % (self.methodmethod.__name__, self.listener_namelistener_name, message))
            LOGGER.info("Details:\n%s" % details)
        finally:
            ListenerMethod.called = False