|
gremmie@1
|
1 """
|
|
gremmie@1
|
2 Models for the downloads application.
|
|
gremmie@1
|
3 """
|
|
gremmie@1
|
4 import os
|
|
gremmie@1
|
5
|
|
gremmie@1
|
6 import datetime
|
|
gremmie@1
|
7 from django.db import models
|
|
gremmie@1
|
8 from django.contrib.auth.models import User
|
|
gremmie@1
|
9 from django.template.defaultfilters import filesizeformat
|
|
gremmie@1
|
10
|
|
bgneal@128
|
11 from core.markup import site_markup
|
|
bgneal@124
|
12
|
|
gremmie@1
|
13
|
|
gremmie@1
|
14 class Category(models.Model):
|
|
gremmie@1
|
15 """Downloads belong to categories."""
|
|
gremmie@1
|
16 title = models.CharField(max_length=64)
|
|
gremmie@1
|
17 description = models.TextField(blank=True)
|
|
bgneal@192
|
18 count = models.IntegerField(default=0, blank=True)
|
|
gremmie@1
|
19
|
|
gremmie@1
|
20 class Meta:
|
|
gremmie@1
|
21 verbose_name_plural = 'Categories'
|
|
gremmie@1
|
22 ordering = ('title', )
|
|
gremmie@1
|
23
|
|
gremmie@1
|
24 def __unicode__(self):
|
|
gremmie@1
|
25 return self.title
|
|
gremmie@1
|
26
|
|
gremmie@1
|
27
|
|
gremmie@1
|
28 def download_path(instance, filename):
|
|
gremmie@1
|
29 """
|
|
gremmie@1
|
30 Creates a path for a download. Uses the current date to avoid filename
|
|
gremmie@1
|
31 clashes. Uses the current microsecond also to make the directory name
|
|
gremmie@1
|
32 harder to guess.
|
|
gremmie@1
|
33 """
|
|
gremmie@1
|
34 now = datetime.datetime.now()
|
|
gremmie@1
|
35 parts = ['downloads']
|
|
gremmie@1
|
36 parts.extend([str(p) for p in (now.year, now.month, now.day, now.microsecond)])
|
|
gremmie@1
|
37 parts.append(filename)
|
|
gremmie@1
|
38 return os.path.join(*parts)
|
|
gremmie@1
|
39
|
|
gremmie@1
|
40
|
|
gremmie@1
|
41 class PublicDownloadManager(models.Manager):
|
|
gremmie@1
|
42 """The manager for all public downloads."""
|
|
gremmie@1
|
43 def get_query_set(self):
|
|
bgneal@190
|
44 return super(PublicDownloadManager, self).get_query_set().filter(
|
|
bgneal@190
|
45 is_public=True).select_related()
|
|
gremmie@1
|
46
|
|
gremmie@1
|
47
|
|
gremmie@1
|
48 class Download(models.Model):
|
|
gremmie@1
|
49 """Model to represent a download."""
|
|
gremmie@1
|
50 title = models.CharField(max_length=128)
|
|
gremmie@1
|
51 category = models.ForeignKey(Category)
|
|
gremmie@1
|
52 description = models.TextField()
|
|
gremmie@1
|
53 html = models.TextField(blank=True)
|
|
gremmie@1
|
54 file = models.FileField(upload_to=download_path)
|
|
gremmie@1
|
55 user = models.ForeignKey(User)
|
|
gremmie@1
|
56 date_added = models.DateTimeField(auto_now_add=True)
|
|
gremmie@1
|
57 ip_address = models.IPAddressField('IP Address')
|
|
gremmie@1
|
58 hits = models.IntegerField(default=0)
|
|
gremmie@1
|
59 average_score = models.FloatField(default=0.0)
|
|
gremmie@1
|
60 total_votes = models.IntegerField(default=0)
|
|
gremmie@1
|
61 is_public = models.BooleanField(default=False, db_index=True)
|
|
gremmie@1
|
62
|
|
gremmie@1
|
63 # Managers:
|
|
gremmie@1
|
64 objects = models.Manager()
|
|
gremmie@1
|
65 public_objects = PublicDownloadManager()
|
|
gremmie@1
|
66
|
|
gremmie@1
|
67 def __unicode__(self):
|
|
gremmie@1
|
68 return self.title
|
|
gremmie@1
|
69
|
|
bgneal@14
|
70 @models.permalink
|
|
bgneal@14
|
71 def get_absolute_url(self):
|
|
bgneal@23
|
72 return ('downloads-details', [str(self.id)])
|
|
bgneal@14
|
73
|
|
bgneal@182
|
74 def save(self, *args, **kwargs):
|
|
bgneal@128
|
75 self.html = site_markup(self.description)
|
|
bgneal@182
|
76 super(Download, self).save(*args, **kwargs)
|
|
gremmie@1
|
77
|
|
gremmie@1
|
78 def vote(self, vote_value):
|
|
gremmie@1
|
79 """receives a vote_value and updates internal score accordingly"""
|
|
gremmie@1
|
80 total_score = self.average_score * self.total_votes
|
|
gremmie@1
|
81 total_score += vote_value
|
|
gremmie@1
|
82 self.total_votes += 1
|
|
gremmie@1
|
83 self.average_score = total_score / self.total_votes
|
|
gremmie@1
|
84 return self.average_score
|
|
gremmie@1
|
85
|
|
gremmie@1
|
86 def size(self):
|
|
gremmie@1
|
87 return filesizeformat(self.file.size)
|
|
gremmie@1
|
88
|
|
gremmie@1
|
89
|
|
gremmie@1
|
90 class AllowedExtensionManager(models.Manager):
|
|
gremmie@1
|
91 def get_extension_list(self):
|
|
gremmie@1
|
92 return self.values_list('extension', flat=True)
|
|
gremmie@1
|
93
|
|
gremmie@1
|
94
|
|
gremmie@1
|
95 class AllowedExtension(models.Model):
|
|
gremmie@1
|
96 """Model to represent the list of allowed file extensions."""
|
|
gremmie@1
|
97 extension = models.CharField(max_length=8)
|
|
gremmie@1
|
98
|
|
gremmie@1
|
99 objects = AllowedExtensionManager()
|
|
gremmie@1
|
100
|
|
gremmie@1
|
101 def __unicode__(self):
|
|
gremmie@1
|
102 return self.extension
|
|
gremmie@1
|
103
|
|
gremmie@1
|
104 class Meta:
|
|
gremmie@1
|
105 ordering = ('extension', )
|
|
gremmie@1
|
106
|
|
gremmie@1
|
107
|
|
gremmie@1
|
108 class VoteRecord(models.Model):
|
|
gremmie@1
|
109 """Model to record the date that a user voted on a download."""
|
|
gremmie@1
|
110 download = models.ForeignKey(Download)
|
|
gremmie@1
|
111 user = models.ForeignKey(User)
|
|
gremmie@1
|
112 vote_date = models.DateTimeField(auto_now_add=True)
|
|
gremmie@1
|
113
|
|
gremmie@1
|
114 def __unicode__(self):
|
|
bgneal@192
|
115 return u"%s voted on '%s' on %s" % (
|
|
gremmie@1
|
116 self.user.username,
|
|
gremmie@1
|
117 self.download.title,
|
|
gremmie@1
|
118 self.vote_date.strftime('%b %d, %Y %H:%M:%S'))
|
|
gremmie@1
|
119
|
|
gremmie@1
|
120 class Meta:
|
|
gremmie@1
|
121 ordering = ('-vote_date', )
|
