summaryrefslogtreecommitdiff
path: root/tests/regressiontests/bug639/models.py
blob: 48fe5f888009f93db7c69fb06ed14b8ab8f3e27a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
import tempfile

from django.db import models
from django.core.files.storage import FileSystemStorage
from django.forms import ModelForm

temp_storage = FileSystemStorage(tempfile.gettempdir())

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