25 lines
691 B
Python
25 lines
691 B
Python
|
from .product import ProductController
|
||
|
|
|
||
|
|
from registrasion import models as rego
|
||
|
|
|
||
|
|
class AllProducts(object):
|
||
|
|
pass
|
||
|
|
|
||
|
|
class CategoryController(object):
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def available_categories(cls, user, products=AllProducts):
|
||
|
|
''' Returns the categories available to the user. Specify `products` if
|
||
|
|
you want to restrict to just the categories that hold the specified
|
||
|
|
products, otherwise it'll do all. '''
|
||
|
|
|
||
|
|
if products is AllProducts:
|
||
|
|
products = rego.Product.objects.all()
|
||
|
|
|
||
|
|
available = ProductController.available_products(
|
||
|
|
user,
|
||
|
|
products=products,
|
||
|
|
)
|
||
|
|
|
||
|
|
return set(i.category for i in available)
|