Add code to check if there is existing data before executing a bootstrap setup

This commit is contained in:
Roberto Rosario
2012-09-27 01:45:52 -04:00
parent b42a43bafd
commit 3b895d7560
4 changed files with 26 additions and 0 deletions

View File

@@ -10,6 +10,7 @@ from django.core import management
from .literals import (FIXTURE_TYPES_CHOICES, FIXTURE_FILE_TYPE,
FIXTURE_TYPE_PK_NULLIFIER, COMMAND_LOADDATA)
from .managers import BootstrapSetupManager
from .classes import BootstrapModel
class BootstrapSetup(models.Model):
@@ -30,6 +31,7 @@ class BootstrapSetup(models.Model):
return FIXTURE_FILE_TYPE[self.type]
def execute(self):
BootstrapModel.check_for_data()
handle, filepath = tempfile.mkstemp()
# Just need the filepath, close the file description
os.close(handle)