"""

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')