NewsBlur-viq/apps/categories/views.py

from apps.categories.models import MCategory
from apps.reader.models import UserSubscriptionFolders
from utils import json_functions as json
from utils.user_functions import ajax_login_required


@json.json_view
def all_categories(request):
    categories = MCategory.serialize()

    return categories


@ajax_login_required
@json.json_view
def subscribe(request):
    user = request.user
    categories = MCategory.serialize()
    category_titles = [c["title"] for c in categories["categories"]]
    subscribe_category_titles = request.POST.getlist("category") or request.POST.getlist("category[]")

    invalid_category_title = False
    for category_title in subscribe_category_titles:
        if category_title not in category_titles:
            invalid_category_title = True

    if not subscribe_category_titles or invalid_category_title:
        message = "Choose one or more of these categories: %s" % ", ".join(category_titles)
        return dict(code=-1, message=message)

    for category_title in subscribe_category_titles:
        MCategory.subscribe(user.pk, category_title)

    usf = UserSubscriptionFolders.objects.get(user=user.pk)

    return dict(
        code=1,
        message="Subscribed to %s %s"
        % (
            len(subscribe_category_titles),
            "category" if len(subscribe_category_titles) == 1 else "categories",
        ),
        folders=json.decode(usf.folders),
    )