/tests/regressiontests/bug639/models.py
Python | 26 lines | 18 code | 6 blank | 2 comment | 0 complexity | b04b81e5f2b4ec025f29e73bc2b1343c MD5 | raw file
Possible License(s): BSD-3-Clause
- import tempfile
- from django.db import models
- from django.core.files.storage import FileSystemStorage
- from django.forms import ModelForm
- temp_storage_dir = tempfile.mkdtemp()
- temp_storage = FileSystemStorage(temp_storage_dir)
- class Photo(models.Model):
- title = models.CharField(max_length=30)
- image = models.FileField(storage=temp_storage, upload_to='tests')
- # Support code for the tests; this keeps track of how many times save()
- # gets called on each instance.
- def __init__(self, *args, **kwargs):
- super(Photo, self).__init__(*args, **kwargs)
- self._savecount = 0
- def save(self, force_insert=False, force_update=False):
- super(Photo, self).save(force_insert, force_update)
- self._savecount += 1
- class PhotoForm(ModelForm):
- class Meta:
- model = Photo