Django: проверьте шаблон в шаблоне?

Question

Я использую родовые типы в моей Profile модели:

user_type = models.ForeignKey(ContentType) 
object_id = models.PositiveIntegerField() 
details = generic.GenericForeignKey('user_type', 'object_id') 

Но теперь я хочу, чтобы проверить, является ли пользователь определенного типа внутри моего шаблона. Я могу получить тип пользователя с {{ user.get_profile.user_type }}, но что тогда? Или как бы добавить метод к модели, например is_type_xxx, чтобы я мог использовать его в шаблоне?

Answer 1

Возможно, я не совсем понимаю вопрос, но, похоже, вы можете просто определить функцию в своей модели, которая возвращает true, если тип является тем, что вам нужно, тогда вы можете вызвать эту функцию из шаблона так же, как вы если вы обращались к переменной.

В модели ...

user_type = models.ForeignKey(ContentType) 
object_id = models.PositiveIntegerField() 
details = generic.GenericForeignKey('user_type', 'object_id') 

def IsTypeX(): 
    return user_type == x 

В шаблоне ...

{% if user.get_profile.IsTypeX %} 
{% endif %} 

Answer 2

user_type - ForeignKey - ContentType model, поэтому относитесь к нему так, как если бы вы имели какое-либо отношение.

Answer 3

Хотя Игнасио в основном правильный, я нахожу, что может быть действительно тенденция получать много непреднамеренных удалений БД, если вы не будете осторожны. Поскольку число ContentTypes имеет тенденцию быть маленьким и относительно неизменным, я кэширую идентификатор пар имя/идентификатор, чтобы избежать его. Вы можете использовать сигнал, чтобы обновить свой dict при невозможности создания нового ContentType.

После этого вы можете автоматически создать необходимые функции is_type_xxx()/методы. Это немного klunky, но код не очень сложный.