diff options
Diffstat (limited to 'parts/django/tests/regressiontests/model_fields/models.py')
-rw-r--r-- | parts/django/tests/regressiontests/model_fields/models.py | 154 |
1 files changed, 0 insertions, 154 deletions
diff --git a/parts/django/tests/regressiontests/model_fields/models.py b/parts/django/tests/regressiontests/model_fields/models.py deleted file mode 100644 index 45cd223..0000000 --- a/parts/django/tests/regressiontests/model_fields/models.py +++ /dev/null @@ -1,154 +0,0 @@ -import os -import tempfile - -# Try to import PIL in either of the two ways it can end up installed. -# Checking for the existence of Image is enough for CPython, but for PyPy, -# you need to check for the underlying modules. - -try: - from PIL import Image, _imaging -except ImportError: - try: - import Image, _imaging - except ImportError: - Image = None - -from django.core.files.storage import FileSystemStorage -from django.db import models -from django.db.models.fields.files import ImageFieldFile, ImageField - - -class Foo(models.Model): - a = models.CharField(max_length=10) - d = models.DecimalField(max_digits=5, decimal_places=3) - -def get_foo(): - return Foo.objects.get(id=1) - -class Bar(models.Model): - b = models.CharField(max_length=10) - a = models.ForeignKey(Foo, default=get_foo) - -class Whiz(models.Model): - CHOICES = ( - ('Group 1', ( - (1,'First'), - (2,'Second'), - ) - ), - ('Group 2', ( - (3,'Third'), - (4,'Fourth'), - ) - ), - (0,'Other'), - ) - c = models.IntegerField(choices=CHOICES, null=True) - -class BigD(models.Model): - d = models.DecimalField(max_digits=38, decimal_places=30) - -class BigS(models.Model): - s = models.SlugField(max_length=255) - -class BigInt(models.Model): - value = models.BigIntegerField() - null_value = models.BigIntegerField(null = True, blank = True) - -class Post(models.Model): - title = models.CharField(max_length=100) - body = models.TextField() - -class NullBooleanModel(models.Model): - nbfield = models.NullBooleanField() - -class BooleanModel(models.Model): - bfield = models.BooleanField() - string = models.CharField(max_length=10, default='abc') - -############################################################################### -# ImageField - -# If PIL available, do these tests. -if Image: - class TestImageFieldFile(ImageFieldFile): - """ - Custom Field File class that records whether or not the underlying file - was opened. - """ - def __init__(self, *args, **kwargs): - self.was_opened = False - super(TestImageFieldFile, self).__init__(*args,**kwargs) - def open(self): - self.was_opened = True - super(TestImageFieldFile, self).open() - - class TestImageField(ImageField): - attr_class = TestImageFieldFile - - # Set up a temp directory for file storage. - temp_storage_dir = tempfile.mkdtemp() - temp_storage = FileSystemStorage(temp_storage_dir) - temp_upload_to_dir = os.path.join(temp_storage.location, 'tests') - - class Person(models.Model): - """ - Model that defines an ImageField with no dimension fields. - """ - name = models.CharField(max_length=50) - mugshot = TestImageField(storage=temp_storage, upload_to='tests') - - class PersonWithHeight(models.Model): - """ - Model that defines an ImageField with only one dimension field. - """ - name = models.CharField(max_length=50) - mugshot = TestImageField(storage=temp_storage, upload_to='tests', - height_field='mugshot_height') - mugshot_height = models.PositiveSmallIntegerField() - - class PersonWithHeightAndWidth(models.Model): - """ - Model that defines height and width fields after the ImageField. - """ - name = models.CharField(max_length=50) - mugshot = TestImageField(storage=temp_storage, upload_to='tests', - height_field='mugshot_height', - width_field='mugshot_width') - mugshot_height = models.PositiveSmallIntegerField() - mugshot_width = models.PositiveSmallIntegerField() - - class PersonDimensionsFirst(models.Model): - """ - Model that defines height and width fields before the ImageField. - """ - name = models.CharField(max_length=50) - mugshot_height = models.PositiveSmallIntegerField() - mugshot_width = models.PositiveSmallIntegerField() - mugshot = TestImageField(storage=temp_storage, upload_to='tests', - height_field='mugshot_height', - width_field='mugshot_width') - - class PersonTwoImages(models.Model): - """ - Model that: - * Defines two ImageFields - * Defines the height/width fields before the ImageFields - * Has a nullalble ImageField - """ - name = models.CharField(max_length=50) - mugshot_height = models.PositiveSmallIntegerField() - mugshot_width = models.PositiveSmallIntegerField() - mugshot = TestImageField(storage=temp_storage, upload_to='tests', - height_field='mugshot_height', - width_field='mugshot_width') - headshot_height = models.PositiveSmallIntegerField( - blank=True, null=True) - headshot_width = models.PositiveSmallIntegerField( - blank=True, null=True) - headshot = TestImageField(blank=True, null=True, - storage=temp_storage, upload_to='tests', - height_field='headshot_height', - width_field='headshot_width') - -############################################################################### |