From 391da3fd27a48659578dc018254ae24de08d999a Mon Sep 17 00:00:00 2001 From: TheOrangeOne Date: Tue, 6 Sep 2016 14:05:03 +0100 Subject: [PATCH] add absolute url to file model --- project/files/models.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) 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)