diff --git a/project/files/models.py b/project/files/models.py index de3d3a7..c8e03d0 100644 --- a/project/files/models.py +++ b/project/files/models.py @@ -1,9 +1,9 @@ from django.db import models -import os import uuid from admin_resumable.fields import ModelAdminResumableFileField from shortuuidfield import ShortUUIDField from datetime import timedelta +from django.core.urlresolvers import reverse class SharedFile(models.Model): @@ -14,9 +14,15 @@ class SharedFile(models.Model): hotlink = models.BooleanField(default=False) published = models.BooleanField(default=True) + def __str__(self): + return self.get_original_filename() + def get_original_filename(self): return "_".join(self.file.name.split('_')[1:]) + def get_absolute_url(self): + return reverse('files:file', args=(self.short_id,)) + class FileToken(models.Model): valid_time = timedelta(minutes=5)