eventfilter.py

"""
Filters event based on schema and table
"""

from ConfigParser import NoOptionError, NoSectionError


class EventFilter(object):
    """
    Filters event based on schema and table
    """

    def __init__(self, config):
        self.config = config

    def filter(self, obj):
        """
        :param obj: object to filter, should contain values for schema and table
        :return: True if the object is on whitelist
        """
        if not hasattr(obj, 'schema'):
            return True

        try:
            self.config.get(obj.schema, obj.table)
            return True
        except (NoOptionError, NoSectionError):
            return False