Change all references from app.name to app instance
This commit is contained in:
@@ -1,4 +1,5 @@
|
||||
import logging
|
||||
import os
|
||||
|
||||
from django.utils.translation import ugettext_lazy as _
|
||||
from django.utils.translation import ugettext
|
||||
@@ -35,8 +36,12 @@ class Fixture(ElementDataBase):
|
||||
self.model_backup = model_backup
|
||||
self.content = content
|
||||
|
||||
@property
|
||||
def filename(self):
|
||||
return '%s_%s' % (self.__class__.name, self.model_backup.app_backup.app)
|
||||
|
||||
def save(self):
|
||||
return ContentFile(name='%s_%s' % (self.__class__.__name__, self.model_backup.app_backup.name), content=self.content)
|
||||
return ContentFile(name=self.filename, content=self.content)
|
||||
|
||||
#def load(self):
|
||||
|
||||
@@ -82,11 +87,12 @@ class ModelBackup(ElementBackupBase):
|
||||
|
||||
command = Command()
|
||||
if not self.model_list:
|
||||
result = [self.app_backup.name]
|
||||
result = [self.app_backup.app.name]
|
||||
else:
|
||||
result = [u'%s.%s' (self.app_backup.name, model) for model in self.model_list]
|
||||
result = [u'%s.%s' (self.app_backup.app.name, model) for model in self.model_list]
|
||||
|
||||
#TODO: a single Fixture or a list of Fixtures for each model?
|
||||
#Can't return multiple Fixture until a way to find all of an app's models is found
|
||||
return Fixture(
|
||||
model_backup=self,
|
||||
content=command.handle(u' '.join(result), format='json', indent=4, using=DEFAULT_DB_ALIAS, exclude=[], user_base_manager=False, use_natural_keys=False)
|
||||
@@ -133,16 +139,12 @@ class AppBackup(object):
|
||||
def get_all(cls):
|
||||
return cls._registry.values()
|
||||
|
||||
@classmethod
|
||||
def get_as_choices(cls):
|
||||
return [(key, key.label) for key, values in cls._registry.items()]
|
||||
|
||||
def __init__(self, name, label, backup_managers):
|
||||
self.label = label
|
||||
self.name = name
|
||||
def __init__(self, app, backup_managers):
|
||||
# app = App instance from app_registry app
|
||||
self.app = app
|
||||
self.backup_managers = [manager.link(self) for manager in backup_managers]
|
||||
self.state = self.__class__.STATE_IDLE
|
||||
self.__class__._registry[name] = self
|
||||
self.__class__._registry[app] = self
|
||||
|
||||
def info(self):
|
||||
results = []
|
||||
@@ -167,7 +169,7 @@ class AppBackup(object):
|
||||
self.state = self.__class__.STATE_IDLE
|
||||
|
||||
def __unicode__(self):
|
||||
return unicode(self.label)
|
||||
return unicode(self.app)
|
||||
|
||||
|
||||
#Storage
|
||||
@@ -249,8 +251,12 @@ class TestStorageModule(StorageModuleBase):
|
||||
|
||||
def backup(self, data, dry_run):
|
||||
print '***** received data'
|
||||
print data
|
||||
#print data.content
|
||||
#print 'name', data.filename
|
||||
print '***** saving to path: %s' % self.backup_path
|
||||
result = data.save()
|
||||
with open(os.path.join(self.backup_path, result.name), 'w') as descriptor:
|
||||
descriptor.write(result.read())
|
||||
|
||||
def restore(self):
|
||||
print 'restore from path: %s' % self.restore_path
|
||||
|
||||
@@ -37,10 +37,6 @@ class BackupJob(models.Model):
|
||||
storage_module_name = models.CharField(max_length=16, choices=StorageModuleBase.get_as_choices(), verbose_name=_(u'storage module'))
|
||||
storage_arguments_json = models.TextField(verbose_name=_(u'storage module arguments (in JSON)'), blank=True)
|
||||
|
||||
#@property
|
||||
#def apps(self):
|
||||
# return self.backupjobapp_set
|
||||
|
||||
def __unicode__(self):
|
||||
return self.name
|
||||
|
||||
@@ -54,7 +50,7 @@ class BackupJob(models.Model):
|
||||
storage_module = self.storage_module
|
||||
#TODO: loads
|
||||
for app in self.apps.all():
|
||||
app_backup = AppBackup.get(app.name)
|
||||
app_backup = AppBackup.get(app)
|
||||
app_backup.backup(storage_module(backup_path='/tmp', dry_run=dry_run), dry_run=dry_run)
|
||||
|
||||
def save(self, *args, **kwargs):
|
||||
@@ -70,9 +66,4 @@ class BackupJob(models.Model):
|
||||
verbose_name_plural = _(u'document checkouts')
|
||||
|
||||
|
||||
#class BackupJobApp(models.Model):
|
||||
# backup_job = models.ForeignKey(BackupJob)
|
||||
# app_backup = models.CharField(max_length=64, choices=AppBackup.get_as_choices())
|
||||
|
||||
|
||||
#class BackupJobLog
|
||||
|
||||
Reference in New Issue
Block a user