diff options
Diffstat (limited to 'account')
| -rw-r--r-- | account/extra.py | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/account/extra.py b/account/extra.py index 74c11e7..d0f74f2 100644 --- a/account/extra.py +++ b/account/extra.py @@ -12,9 +12,12 @@ from django.conf import settings from django.template.defaultfilters import filesizeformat from django.utils.translation import ugettext_lazy as _ +from south.modelsinspector import add_introspection_rules + import os +### custom fields ### class ContentTypeRestrictedFileField(models.FileField): """ Same as FileField, but you can specify: @@ -54,6 +57,18 @@ class ContentTypeRestrictedFileField(models.FileField): # return data +## add custom fields to south inspection +add_introspection_rules([ + ( + [ContentTypeRestrictedFileField], # class these apply to + [], # positional arguments + { # keyword argument + "content_types": ["content_types", {}], + "max_upload_size": ["max_upload_size", {}], + }, + ), +], ["^account\.extra\.ContentTypeRestrictedFileField"]) + ### OverwriteStorage ### class OverwriteStorage(FileSystemStorage): |