Preserve filename existing extension if new filename doesn't has one

Fixes issue #24
This commit is contained in:
Roberto Rosario
2012-06-16 01:19:22 -04:00
parent 4322ac0a02
commit 4a1acbbaac

View File

@@ -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()