/tests/modeltests/files/models.py
https://code.google.com/p/mango-py/ · Python · 34 lines · 17 code · 8 blank · 9 comment · 0 complexity · 1f115d71467bf367b818e1c66a7d9656 MD5 · raw file
- """
- 42. Storing files according to a custom storage system
- ``FileField`` and its variations can take a ``storage`` argument to specify how
- and where files should be stored.
- """
- import random
- import tempfile
- from django.db import models
- from django.core.files.base import ContentFile
- from django.core.files.storage import FileSystemStorage
- temp_storage_location = tempfile.mkdtemp()
- temp_storage = FileSystemStorage(location=temp_storage_location)
- # Write out a file to be used as default content
- temp_storage.save('tests/default.txt', ContentFile('default content'))
- class Storage(models.Model):
- def custom_upload_to(self, filename):
- return 'foo'
- def random_upload_to(self, filename):
- # This returns a different result each time,
- # to make sure it only gets called once.
- return '%s/%s' % (random.randint(100, 999), filename)
- normal = models.FileField(storage=temp_storage, upload_to='tests')
- custom = models.FileField(storage=temp_storage, upload_to=custom_upload_to)
- random = models.FileField(storage=temp_storage, upload_to=random_upload_to)
- default = models.FileField(storage=temp_storage, upload_to='tests', default='tests/default.txt')