from shop.views import ShopListView, ShopDetailView
from shop_richcatalog.models import Catalog
from shop.models import Product


class CatalogListView(ShopListView):
    '''
    TODO.
    '''

    model = Catalog
    #generic_template = "shop_richcatalog/catalog_list.html"

class CatalogDetailView(ShopDetailView):
    '''
    TODO.
    '''

    model = Catalog
    #generic_template = "shop_richcatalog/catalog_detail.html"

    def get_context_data(self, **kwargs):
        '''
        TODO.
        '''

        # get context data from superclass
        ctx = super(CatalogDetailView, self).get_context_data(**kwargs)

        # update the context with active products in this catalog
        product_list = self.object.products.filter(active=True)
        if product_list:
            ctx.update({"product_list": product_list})

        # return the context
        return ctx

from shop.views import ShopListView, ShopDetailView
from shop_richcatalog.models import Catalog
from shop.models import Product


class CatalogListView(ShopListView):
    '''
    Display all catalogs in a tree.
    '''

    model = Catalog


class CatalogDetailView(ShopDetailView):
    '''
    Display detailed catalog information.
    '''

    model = Catalog

    def get_context_data(self, **kwargs):
        '''
        Get catalog context data.
        '''

        # get context data from superclass
        ctx = super(CatalogDetailView, self).get_context_data(**kwargs)

        # update the context with active products in this catalog
        product_list = self.object.products.filter(active=True)
        if product_list:
            ctx.update({"product_list": product_list})

        # return the context
        return ctx

Document the view classes and make fix their spacing for pep8.
Document the view classes and make fix their spacing for pep8.

  from shop.views import ShopListView, ShopDetailView
  from shop_richcatalog.models import Catalog
  from shop.models import Product
  
  
  class CatalogListView(ShopListView):
      '''
-     TODO.
+     Display all catalogs in a tree.
      '''
  
      model = Catalog
-     #generic_template = "shop_richcatalog/catalog_list.html"
+ 
  
  class CatalogDetailView(ShopDetailView):
      '''
-     TODO.
+     Display detailed catalog information.
      '''
  
      model = Catalog
-     #generic_template = "shop_richcatalog/catalog_detail.html"
  
      def get_context_data(self, **kwargs):
          '''
-         TODO.
+         Get catalog context data.
          '''
  
          # get context data from superclass
          ctx = super(CatalogDetailView, self).get_context_data(**kwargs)
  
          # update the context with active products in this catalog
          product_list = self.object.products.filter(active=True)
          if product_list:
              ctx.update({"product_list": product_list})
  
          # return the context
          return ctx
  
Document the view classes and make fix their spacing for pep8.
## Code Before:
from shop.views import ShopListView, ShopDetailView
from shop_richcatalog.models import Catalog
from shop.models import Product


class CatalogListView(ShopListView):
    '''
    TODO.
    '''

    model = Catalog
    #generic_template = "shop_richcatalog/catalog_list.html"

class CatalogDetailView(ShopDetailView):
    '''
    TODO.
    '''

    model = Catalog
    #generic_template = "shop_richcatalog/catalog_detail.html"

    def get_context_data(self, **kwargs):
        '''
        TODO.
        '''

        # get context data from superclass
        ctx = super(CatalogDetailView, self).get_context_data(**kwargs)

        # update the context with active products in this catalog
        product_list = self.object.products.filter(active=True)
        if product_list:
            ctx.update({"product_list": product_list})

        # return the context
        return ctx

## Instruction:
Document the view classes and make fix their spacing for pep8.
## Code After:
from shop.views import ShopListView, ShopDetailView
from shop_richcatalog.models import Catalog
from shop.models import Product


class CatalogListView(ShopListView):
    '''
    Display all catalogs in a tree.
    '''

    model = Catalog


class CatalogDetailView(ShopDetailView):
    '''
    Display detailed catalog information.
    '''

    model = Catalog

    def get_context_data(self, **kwargs):
        '''
        Get catalog context data.
        '''

        # get context data from superclass
        ctx = super(CatalogDetailView, self).get_context_data(**kwargs)

        # update the context with active products in this catalog
        product_list = self.object.products.filter(active=True)
        if product_list:
            ctx.update({"product_list": product_list})

        # return the context
        return ctx
