ztimson/oacpl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Releases Activity

Added PDF file type restriction

Browse Source
This commit is contained in:
Zak Timson
2017-09-13 22:13:47 -04:00
parent 6a754b4eee
commit 863431ef41
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
case_law/models.py
View File

@@ -1,3 +1,4 @@
from django.core.exceptions import ValidationError
from django.db import models from django.db import models
@@ -15,9 +16,13 @@ class Case(models.Model):
('view_pdf', 'Can view PDF'), ('view_pdf', 'Can view PDF'),
) )
def validate_file_extension(value):
if not value.name.endswith('.pdf'):
raise ValidationError(u'File is not a PDF')
published = models.DateField() published = models.DateField()
headings = models.ManyToManyField(Heading) headings = models.ManyToManyField(Heading)
pdf = models.FileField(upload_to='secure') pdf = models.FileField('PDF', upload_to='secure', validators=[validate_file_extension])
synopsis = models.TextField() synopsis = models.TextField()
def __str__(self): def __str__(self):