Mercurial > public > sg101
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')