fishyboteso/fishy/web/decorators.py

import logging
import traceback
from functools import wraps


def uses_session(f):
    @wraps(f)
    def wrapper(*args, **kwargs):
        from .web import get_session
        if get_session(args[0]) is None:
            logging.error("Couldn't create a session")
            return None
        else:
            return f(*args, **kwargs)

    return wrapper


def fallback(default):
    def inner(f):
        # noinspection PyBroadException
        @wraps(f)
        def wrapper(*args, **kwargs):
            try:
                return f(*args, **kwargs)
            except Exception:
                traceback.print_exc()
                return default
        return wrapper

    return inner