Dynamic Message Contexts

client.py

from universal.helpers import message_context
from myjobs.models import User
from seo.models import Company

u = User.objects.get(email="[email protected]")
company = Company.objects.get(name="Test Company")

invitation = u.create_invitation(company, "[email protected]")

# for a default invitation email
invitation.send()

# for a custom invitation email
invitation.send("This is a test")

# for a saved search invitation email
saved_search = SavedSearch.objects.first()
invitation.send(saved_search)

# for a role invitation
role = Role.objects.filter(company=company, name="Admin")
invitation.send(role)

dunder.py

# in universal/helpers.py
def message_context(obj):
    if isinstance(obj, basestring):
        return {"message": obj}
    elif obj is None:
        return {"message": ""}
    
    if hasattr(obj, "__message_context__"):
        return obj.__message_context__()
    else:
        raise TypeError("object of type '%s' has no message_context()" % type(obj))
    
# in mysearches/models.py
class SavedSearch(Model):
    def __message_context___(self):
        return {"message": "to receive saved searches", "saved_search": self}
        
# in myjobs/models.py
class Role(Model):
    def __message_context__(self):
        return {"message": "as a(n) %s for %s" % (self.name, self.company), "role": self}

generics.py

from pkgutil import simplegeneric
from mysearches.models import SavedSearch

# in universal/helpers.py
@simplegeneric
def message_context(obj):
    raise TypeError("object of type '%s' has no message_context()" % type(obj)
 
message_context.register(None)(lambda obj: {"message": ""}
message_context.register(str)(lambda obj: {"message": obj}

# in mysearches/models.py
message_context.register(SavedSearch)(lambda obj: {"message": "to receive saved searches", "saved_search": obj}

# in myjobs/models.py
message_context.register(Role)(lambda obj: {"message": "as a(n) %s for %s" % (obj.self, obj.company), "role": obj})