annotate downloads/management/commands/dlcatreport.py @ 973:6f55c086db1e
Guess file extension based on content-type.
When downloading a file, and no path is supplied to store it, guess the file
extension using mimetypes and the content-type header.
Also supply a unit test for the HotLinkImageForm.
| author |
Brian Neal <bgneal@gmail.com> |
| date |
Thu, 01 Oct 2015 19:44:45 -0500 |
| parents |
b59c154d0163 |
| children |
cb712b2c34af |
| rev |
line source |
|
bgneal@527
|
1 """
|
|
bgneal@527
|
2 dlcatreport - a management command to produce a HTML report of all the downloads
|
|
bgneal@527
|
3 in a given category.
|
|
bgneal@527
|
4
|
|
bgneal@527
|
5 """
|
|
bgneal@634
|
6 from optparse import make_option
|
|
bgneal@634
|
7
|
|
bgneal@527
|
8 from django.core.management.base import LabelCommand, CommandError
|
|
bgneal@527
|
9 from django.template.loader import render_to_string
|
|
bgneal@527
|
10
|
|
bgneal@527
|
11 from downloads.models import Category, Download
|
|
bgneal@527
|
12
|
|
bgneal@527
|
13
|
|
bgneal@527
|
14 class Command(LabelCommand):
|
|
bgneal@527
|
15 help = "Produce on standard output a report of all downloads in a category."
|
|
bgneal@527
|
16 args = "category-slug"
|
|
bgneal@527
|
17
|
|
bgneal@634
|
18 option_list = LabelCommand.option_list + (
|
|
bgneal@634
|
19 make_option('--titles-only',
|
|
bgneal@634
|
20 action='store_true',
|
|
bgneal@634
|
21 default=False,
|
|
bgneal@634
|
22 help='Output a text listing of titles only'),
|
|
bgneal@634
|
23 )
|
|
bgneal@634
|
24
|
|
bgneal@527
|
25 def handle_label(self, slug, **options):
|
|
bgneal@527
|
26 """
|
|
bgneal@527
|
27 Render a template using the downloads in a given category and send it to
|
|
bgneal@527
|
28 stdout.
|
|
bgneal@527
|
29
|
|
bgneal@527
|
30 """
|
|
bgneal@527
|
31 try:
|
|
bgneal@527
|
32 category = Category.objects.get(slug=slug)
|
|
bgneal@527
|
33 except Category.DoesNotExist:
|
|
bgneal@527
|
34 raise CommandError("category slug '%s' does not exist" % slug)
|
|
bgneal@527
|
35
|
|
bgneal@527
|
36 downloads = Download.public_objects.filter(category=category).order_by(
|
|
bgneal@527
|
37 'title').select_related()
|
|
bgneal@527
|
38
|
|
bgneal@634
|
39 if options.get('titles_only'):
|
|
bgneal@684
|
40 self.print_titles(downloads)
|
|
bgneal@634
|
41 return
|
|
bgneal@634
|
42
|
|
bgneal@527
|
43 report = render_to_string('downloads/commands/category_report.html', {
|
|
bgneal@527
|
44 'category': category,
|
|
bgneal@527
|
45 'downloads': downloads,
|
|
bgneal@527
|
46 })
|
|
bgneal@527
|
47
|
|
bgneal@874
|
48 self.stdout.write(report)
|
|
bgneal@527
|
49
|
|
bgneal@684
|
50 def print_titles(self, dls):
|
|
bgneal@684
|
51 """Print out the download titles"""
|
|
bgneal@684
|
52
|
|
bgneal@684
|
53 for dl in dls:
|
|
bgneal@874
|
54 self.stdout.write(dl.title)
|
|
bgneal@684
|
55
|
|
bgneal@684
|
56
|
