2011-11-09 26 views
2

我有一個使用Django-MPTT的模型,它也有一個通過template_name_field使用特定模板的字段。我希望能夠使用通用TemplateView,但傳入混合使用template_name_field如何在Django TemplateView中傳入mixin for template_name_field?

這裏的模型:

from django.db import models 
from mptt.models import MPTTModel 
from mptt.fields import TreeForeignKey 

class CategoryManager(models.Manager): 
    def get(self, **kwargs): 
     defaults = {} 
     defaults.update(kwargs) 
     if 'full_slug' in defaults: 
      if defaults['full_slug'] and defaults['full_slug'][-1] != "/": 
       defaults['full_slug'] += "/" 
     return super(CategoryManager, self).get(**defaults) 

class Category(MPTTModel): 
    title = models.CharField(max_length=255) 
    description = models.TextField(blank=True, help_text='A brief description of the category. No HTML allowed.') 
    slug = models.SlugField(help_text='Prepopulates from title field.') 
    full_slug = models.CharField(max_length=255, blank=True) 
    template_name = models.CharField(max_length=70, blank=True, help_text="Example: 'categories/category_parent.html'. If this isn't provided, the system will use 'categories/category_detail.html'.") 

    parent = TreeForeignKey('self', null=True, blank=True, related_name='children') 

    objects = CategoryManager() 

    class Meta: 
     verbose_name = 'category' 
     verbose_name_plural = 'categories' 

    def save(self, *args, **kwargs): 
     orig_full_slug = self.full_slug 
     if self.parent: 
      self.full_slug = "%s%s/" % (self.parent.full_slug, self.slug) 
     else: 
     self.full_slug = "%s/" % self.slug 
     obj = super(Category, self).save(*args, **kwargs) 
     if orig_full_slug != self.full_slug: 
      for child in self.get_children(): 
       child.save() 
     return obj 

    def __unicode__(self): 
     return "%s (%s)" % (self.title, self.full_slug) 

這裏的觀點:

from django.core.exceptions import ImproperlyConfigured 
from django.core.urlresolvers import reverse 
from django.views.generic import TemplateView, DetailView 
from django.views.generic.detail import SingleObjectTemplateResponseMixin 
from django.utils.translation import ugettext as _ 
from django.contrib.syndication.views import Feed 

from storefront.categories.models import Category 

class SimpleCategoryView(TemplateView): 

    def get_category(self): 
     return Category.objects.get(full_slug=self.kwargs['full_slug']) 

    def get_context_data(self, **kwargs): 
     context = super(SimpleCategoryView, self).get_context_data(**kwargs) 
     context["category"] = self.get_category() 
     return context 

下面是我的網址:

from django.conf.urls.defaults import patterns, include, url 
from storefront.categories.models import Category 
from storefront.categories.views import CategoryView 

urlpatterns = patterns('', 
    url(r'^(?P<full_slug>[-\w/]+)', SimpleCategoryView.as_view(template_name='categories/category_detail.html'), name='category_view'), 

),

我將如何去增加爲使用mixin g template_name_field以我的觀點?

回答

1

你不需要單獨的mixin。 TemplateView已經有TemplateResponseMixin,所以你可以簡單地在你的SimpleCategoryView子類中覆蓋get_template_names

class SimpleCategoryView(TemplateView): 
    ... 
    def get_template_names(self): 
     return [self.get_category().template_name] 
+0

工作,但如果'template_name'爲None,我該如何解決異常? –

+0

我想出了else:return ['categories/category_detail.html']' –

+0

FWIW,'get_template_names'返回一個模板列表。這意味着你可以提供回退。它會按順序嘗試每個項目,並按照與實際文件匹配的第一項進行。 –

0

DetailView應該有你需要的一切。您只需使用template_name_field屬性告訴DetailView.get_template_names()哪個字段可以包含模板名稱。下面是一個完整的視圖定義,它基於url中的slug獲取Category。請注意,Category將作爲objectcategory在模板中提供。

class SimpleCategoryView(DetailView): 
    model = Category 
    slug_url_kwarg = 'full_slug' 
    slug_field = 'full_slug' 
    template_name_field = 'template_name'