Prepend a magic number to bootstrap setup to identify them as valid bootstrap setups

This commit is contained in:
Roberto Rosario
2012-10-16 02:44:01 -04:00
parent b076c73d2b
commit 1de90cfaa9
6 changed files with 30 additions and 6 deletions

View File

@@ -8,8 +8,7 @@ from django.db import models
from django.core import serializers
from .classes import BootstrapModel, FixtureMetadata
from .literals import (FIXTURE_TYPE_FIXTURE_PROCESS, FIXTURE_TYPE_EMPTY_FIXTURE,
DEFAULT_REPOSITORY)
from .literals import (FIXTURE_TYPE_FIXTURE_PROCESS, FIXTURE_TYPE_EMPTY_FIXTURE)
logger = logging.getLogger(__name__)
@@ -30,6 +29,7 @@ class BootstrapSetupManager(models.Manager):
return FIXTURE_TYPE_FIXTURE_PROCESS[serialization_format]('\n'.join(result))
def import_setup(self, file_data):
BootstrapModel.check_magic_number(file_data)
metadata = FixtureMetadata.read_all(file_data)
instance = self.model(fixture=file_data, **metadata)
instance.save(update_metadata=False)