diff options
Diffstat (limited to 'account/models.py')
| -rw-r--r-- | account/models.py | 50 |
1 files changed, 49 insertions, 1 deletions
diff --git a/account/models.py b/account/models.py index 257477d..cb806da 100644 --- a/account/models.py +++ b/account/models.py @@ -6,12 +6,14 @@ from django.db import models from django.contrib.auth.models import User +from django.db.models.fields.files import FieldFile from django.conf import settings from django.utils.translation import ugettext_lazy as _ +from django.db.models.signals import pre_delete from registration.signals import user_registered -from account.extra import ContentTypeRestrictedFileField +from account.extra import ContentTypeRestrictedFileField, OverwriteStorage ###### account models ###### @@ -68,6 +70,7 @@ class UserProfile(models.Model): # transcript: needed if undergraudate (junior and below) transcript = ContentTypeRestrictedFileField(upload_to=lambda instance, filename: u'account/{0}/{1}'.format(instance.user.username, filename), verbose_name=_("Transcript"), blank=True, null=True, + storage=OverwriteStorage(), content_types=settings.ALLOWED_CONTENT_TYPES, max_upload_size=settings.ALLOWED_MAX_UPLOAD_SIZE) # supplement: record additional information @@ -85,6 +88,18 @@ class UserProfile(models.Model): def __unicode__(self): return u'UserProfile for %s' % self.user.username + def save(self, *args, **kwargs): + """ + overwrite the original save method to delete old file + """ + old_obj = type(self).objects.get(pk=self.pk) + result = super(UserProfile, self).save(*args, **kwargs) + # if the path is the same, the file is overwritten already + if old_obj.transcript.path != self.transcript.path: + old_obj.transcript.delete(save=False) + # + return result + def is_transcript_required(self): """ if 'identify' is UG (undergraduate junior and below); then @@ -152,6 +167,7 @@ class UserFile(models.Model): description = models.TextField(_("Description"), blank=True) file = ContentTypeRestrictedFileField(upload_to=lambda instance, filename: u'account/{0}/{1}'.format(instance.user.username, filename), verbose_name=_("File"), + storage = OverwriteStorage(), content_types=settings.ALLOWED_CONTENT_TYPES, max_upload_size=settings.ALLOWED_MAX_UPLOAD_SIZE) created_at = models.DateTimeField(_("Created time"), @@ -166,6 +182,18 @@ class UserFile(models.Model): def __unicode__(self): return u'UserFile of %s: %s' % (self.user.username, self.title) + def save(self, *args, **kwargs): + """ + overwrite the original save method to delete old file + """ + old_obj = type(self).objects.get(pk=self.pk) + result = super(UserFile, self).save(*args, **kwargs) + # if the path is the same, the file is overwritten already + if old_obj.file.path != self.file.path: + old_obj.file.delete(save=False) + # + return result + ###### signal callback ###### def user_registered_callback(sender, user, request, **kwargs): @@ -185,3 +213,23 @@ def user_registered_callback(sender, user, request, **kwargs): user_registered.connect(user_registered_callback) +### delete files associated with model FileField +# Pre-delete signal function for deleting files a model +# https://djangosnippets.org/snippets/2820/ +def file_cleanup(sender, instance, *args, **kwargs): + """ + Deletes the file(s) associated with a model instance. The model + is not saved after deletion of the file(s) since this is meant + to be used with the pre_delete signal. + """ + for field_name, _ in instance.__dict__.iteritems(): + field = getattr(instance, field_name) + if (issubclass(field.__class__, FieldFile) and field.name): + # pass False so FileField does not save the model + field.delete(save=False) + +### connect to signal and sender +pre_delete.connect(file_cleanup, sender=UserProfile) +pre_delete.connect(file_cleanup, sender=UserFile) + + |