models.py
1.18 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
from django.db import models
from django.utils.translation import ugettext_lazy as _
import os
from django.core.exceptions import ValidationError
from django.core.urlresolvers import reverse_lazy
from topics.models import Resource
# Create your models here.
valid_formats = [
'application/pdf'
]
def validate_file_extension(value):
if hasattr(value.file, 'content_type'):
if not value.file.content_type in valid_formats:
raise ValidationError(_('File not supported, use PDF format instead.'))
class PDFFile(Resource):
file = models.FileField(_('File'), upload_to='files/', validators = [validate_file_extension])
class Meta:
verbose_name = "PDFFile"
verbose_name_plural = "PDFFiles"
@property
def filename(self):
return os.path.basename(self.file.name)
def __str__(self):
return self.name
def access_link(self):
return reverse_lazy('pdf_files:view', args = (), kwargs = {'slug': self.slug})
def update_link(self):
return 'pdf_files:update'
def delete_link(self):
return 'pdf_files:delete'
def delete_message(self):
return _('Are you sure you want delete the PDF File')