Add support for exporting bootstrap setups to file

This commit is contained in:
Roberto Rosario
2012-10-14 05:10:55 -04:00
parent 4b143a9455
commit 37372ba554
7 changed files with 40 additions and 5 deletions

View File

@@ -13,8 +13,10 @@ except ImportError:
from django.db import models
from django.utils.translation import ugettext_lazy as _
from django.core import management
from django.core.files.uploadedfile import SimpleUploadedFile
from .literals import (FIXTURE_TYPES_CHOICES, FIXTURE_FILE_TYPE, COMMAND_LOADDATA)
from .literals import (FIXTURE_TYPES_CHOICES, FIXTURE_FILE_TYPE, COMMAND_LOADDATA,
BOOTSTRAP_EXTENSION)
from .managers import BootstrapSetupManager
from .classes import BootstrapModel, FixtureMetadata
@@ -84,6 +86,12 @@ class BootstrapSetup(models.Model):
"""
return FixtureMetadata.generate_all(self)
def get_filename(self):
return os.extsep.join([self.name, BOOTSTRAP_EXTENSION])
def as_file(self):
return SimpleUploadedFile(name=self.get_filename(), content=self.fixture)
def save(self, *args, **kwargs):
self.fixture = '%s\n\n%s' % (
self.get_metadata_string(),