From 4a1acbbaac707803e3b624a6b309d941e23a7da2 Mon Sep 17 00:00:00 2001 From: Roberto Rosario Date: Sat, 16 Jun 2012 01:19:22 -0400 Subject: [PATCH] Preserve filename existing extension if new filename doesn't has one Fixes issue #24 --- apps/documents/models.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/apps/documents/models.py b/apps/documents/models.py index 8185bb8297..d941c86c7d 100644 --- a/apps/documents/models.py +++ b/apps/documents/models.py @@ -524,8 +524,14 @@ class DocumentVersion(models.Model): return None def rename(self, new_name): + new_filename, new_extension = os.path.splitext(new_name) name, extension = os.path.splitext(self.filename) - self.filename = u''.join([new_name, extension]) + + # Preserve existing extension if new name doesn't has one + if new_extension: + extension = new_extension + + self.filename = u''.join([new_filename, extension]) self.save()