view downloads/management/commands/dlcatreport.py @ 656:79e785f0bdad

For issue #38, change IRC bot to use Redis instead of MySQL. Also deleting the bot from this repo as it now has its own repo.
author Brian Neal <bgneal@gmail.com>
date Sat, 11 May 2013 15:22:45 -0500 (2013-05-11)
parents d6489e6a40f6
children 161b56849114
line wrap: on
line source
"""
dlcatreport - a management command to produce a HTML report of all the downloads
in a given category.

"""
from optparse import make_option

from django.core.management.base import LabelCommand, CommandError
from django.template.loader import render_to_string

from downloads.models import Category, Download


def print_titles(dls):
    """Print out the download titles"""

    for dl in dls:
        print dl.title.encode('utf-8')


class Command(LabelCommand):
    help = "Produce on standard output a report of all downloads in a category."
    args = "category-slug"

    option_list = LabelCommand.option_list + (
            make_option('--titles-only',
                action='store_true',
                default=False,
                help='Output a text listing of titles only'),
            )

    def handle_label(self, slug, **options):
        """
        Render a template using the downloads in a given category and send it to
        stdout.

        """
        try:
            category = Category.objects.get(slug=slug)
        except Category.DoesNotExist:
            raise CommandError("category slug '%s' does not exist" % slug)

        downloads = Download.public_objects.filter(category=category).order_by(
                'title').select_related()

        if options.get('titles_only'):
            print_titles(downloads)
            return

        report = render_to_string('downloads/commands/category_report.html', {
            'category': category,
            'downloads': downloads,
            })

        # encode it ourselves since it can fail if you try to redirect output to
        # a file and any of the content is not ASCII...
        print report.encode('utf-8')