def domain_relative_to(domain, zone):
    if domain == '@':
        return zone
    if domain[-1] != '.':
        return f"{domain}.{zone}"
    return domain

class FilterModule(object):
    def filters(self):
        return {'domain_relative_to': domain_relative_to}