diff downloads/management/commands/dlcatreport.py @ 581:ee87ea74d46b

For Django 1.4, rearranged project structure for new manage.py.
author Brian Neal <bgneal@gmail.com>
date Sat, 05 May 2012 17:10:48 -0500
parents gpp/downloads/management/commands/dlcatreport.py@645025056dfa
children d6489e6a40f6
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/downloads/management/commands/dlcatreport.py	Sat May 05 17:10:48 2012 -0500
@@ -0,0 +1,38 @@
+"""
+dlcatreport - a management command to produce a HTML report of all the downloads
+in a given category.
+
+"""
+from django.core.management.base import LabelCommand, CommandError
+from django.template.loader import render_to_string
+
+from downloads.models import Category, Download
+
+
+class Command(LabelCommand):
+    help = "Produce on standard output a report of all downloads in a category."
+    args = "category-slug"
+
+    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()
+
+        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')
+