#!/usr/bin/env python3

"""Custom filters for use in playbooks."""


class FilterModule(object):
    """Custom Ansible filter plugins."""

    def filters(self):
        """Filters list."""
        return {"update_env": self.filter_update_env}

    def filter_update_env(self, config, new_env):
        """Update the config.env dictionary with new keys."""
        if "env" not in config:
            return config

        for key, val in new_env.items():
            config["env"][key] = val

        return config