Add registry app updates
This commit is contained in:
Roberto Rosario
@@ -2,6 +2,8 @@ from __future__ import absolute_import
|
||||
|
||||
import datetime
|
||||
import logging
|
||||
import imp
|
||||
import sys
|
||||
|
||||
from django.db import models
|
||||
from django.db import DatabaseError, transaction
|
||||
@@ -9,22 +11,25 @@ from django.utils.translation import ugettext_lazy as _
|
||||
from django.utils.translation import ugettext
|
||||
from django.contrib.contenttypes.models import ContentType
|
||||
from django.contrib.contenttypes import generic
|
||||
from django.utils.importlib import import_module
|
||||
|
||||
from common.models import TranslatableLabelMixin, LiveObjectMixin
|
||||
from smart_settings import SettingsNamespace
|
||||
from project_setup.api import register_setup
|
||||
from project_tools.api import register_tool
|
||||
|
||||
from .classes import AppBackup, StorageModuleBase
|
||||
#from .classes import AppBackup, StorageModuleBase, Setting
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
class App(TranslatableLabelMixin, LiveObjectMixin, models.Model):
|
||||
translatables = ['label', 'description']
|
||||
translatables = ['label', 'description', 'icon']
|
||||
|
||||
class UnableToRegister(Exception):
|
||||
pass
|
||||
#class UnableToRegister(Exception):
|
||||
# pass
|
||||
|
||||
name = models.CharField(max_length=64, verbose_name=_(u'name'), unique=True)
|
||||
icon = models.CharField(max_length=64, verbose_name=_(u'icon'), blank=True)
|
||||
dependencies = models.ManyToManyField('self', verbose_name=_(u'dependencies'), symmetrical=False, blank=True, null=True)
|
||||
#version
|
||||
#top_urls
|
||||
@@ -32,30 +37,62 @@ class App(TranslatableLabelMixin, LiveObjectMixin, models.Model):
|
||||
|
||||
@classmethod
|
||||
@transaction.commit_on_success
|
||||
def register(cls, name, label, icon=None, description=None):
|
||||
def register(cls, app_name):
|
||||
try:
|
||||
app, created = App.objects.get_or_create(name=name)
|
||||
except DatabaseError:
|
||||
transaction.rollback()
|
||||
raise cls.UnableToRegister
|
||||
app_module = import_module(app_name)
|
||||
except ImportError:
|
||||
transaction.rollback
|
||||
else:
|
||||
app.label = label
|
||||
if icon:
|
||||
app.icon = icon
|
||||
if description:
|
||||
app.description = description
|
||||
app.dependencies.clear()
|
||||
app.save()
|
||||
return app
|
||||
|
||||
def set_dependencies(self, app_names):
|
||||
for app_name in app_names:
|
||||
app = App.objects.get(name=app_name)
|
||||
self.dependencies.add(app)
|
||||
|
||||
def set_backup(self, *args, **kwargs):
|
||||
return AppBackup(self, *args, **kwargs)
|
||||
|
||||
try:
|
||||
registration = import_module('%s.registry' % app_name)
|
||||
except ImportError:
|
||||
transaction.rollback
|
||||
else:
|
||||
disabled = getattr(registration, 'disabled', False)
|
||||
name = getattr(registration, 'name')
|
||||
label = getattr(registration, 'label')
|
||||
icon = getattr(registration, 'icon', None)
|
||||
description = getattr(registration, 'description', None)
|
||||
dependencies = getattr(registration, 'dependencies', [])
|
||||
settings = getattr(registration, 'settings', None)
|
||||
setup_links = getattr(registration, 'setup_links', [])
|
||||
tool_links = getattr(registration, 'tool_links', [])
|
||||
|
||||
if not disabled:
|
||||
try:
|
||||
app, created = App.objects.get_or_create(name=name)
|
||||
except DatabaseError:
|
||||
transaction.rollback()
|
||||
raise cls.UnableToRegister
|
||||
else:
|
||||
app.label = label
|
||||
if description:
|
||||
app.description = description
|
||||
app.dependencies.clear()
|
||||
app.save()
|
||||
app.icon = icon
|
||||
|
||||
for app_name in dependencies:
|
||||
dependency = App.objects.get(name=app_name)
|
||||
app.dependencies.add(dependency)
|
||||
|
||||
if settings:
|
||||
settings_module = imp.new_module('settings')
|
||||
setattr(app_module, 'settings', settings_module)
|
||||
sys.modules['%s.settings' % name] = settings_module
|
||||
settings_namespace = SettingsNamespace(name, label, '%s.settings' % name)
|
||||
for setting in settings:
|
||||
settings_namespace.add_setting(**setting)
|
||||
|
||||
for link in setup_links:
|
||||
register_setup(link)
|
||||
|
||||
for link in tool_links:
|
||||
register_tool(link)
|
||||
|
||||
#def set_backup(self, *args, **kwargs):
|
||||
# return AppBackup(self, *args, **kwargs)
|
||||
|
||||
def __unicode__(self):
|
||||
return unicode(self.label)
|
||||
|
||||
@@ -65,55 +102,3 @@ class App(TranslatableLabelMixin, LiveObjectMixin, models.Model):
|
||||
verbose_name_plural = _(u'apps')
|
||||
|
||||
|
||||
class BackupJob(models.Model):
|
||||
name = models.CharField(max_length=64, verbose_name=_(u'name'))
|
||||
enabled = models.BooleanField(default=True, verbose_name=_(u'enabled'))
|
||||
apps = models.ManyToManyField(App)
|
||||
begin_datetime = models.DateTimeField(verbose_name=_(u'begin date and time'), default=lambda: datetime.datetime.now())
|
||||
|
||||
# * repetition =
|
||||
# day - 1 days
|
||||
# weekly - days of week checkbox
|
||||
# month - day of month, day of week
|
||||
# * repetition option field
|
||||
# * ends
|
||||
# - never
|
||||
# - After # ocurrences
|
||||
# - On date
|
||||
# * end option field
|
||||
# * type
|
||||
# - Full
|
||||
# - Incremental
|
||||
storage_module_name = models.CharField(max_length=32, 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)
|
||||
|
||||
def __unicode__(self):
|
||||
return self.name
|
||||
|
||||
@property
|
||||
def storage_module(self):
|
||||
return StorageModuleBase.get(self.storage_module_name)
|
||||
|
||||
def backup(self, dry_run=False):
|
||||
logger.debug('starting: %s', self)
|
||||
logger.debug('dry_run: %s' % dry_run)
|
||||
storage_module = self.storage_module
|
||||
#TODO: loads
|
||||
for app in self.apps.all():
|
||||
app_backup = AppBackup.get(app)
|
||||
app_backup.backup(storage_module(backup_path='/tmp'), dry_run=dry_run)
|
||||
|
||||
def save(self, *args, **kwargs):
|
||||
#dump
|
||||
super(BackupJob, self).save(*args, **kwargs)
|
||||
|
||||
@models.permalink
|
||||
def get_absolute_url(self):
|
||||
return ('checkout_info', [self.document.pk])
|
||||
|
||||
class Meta:
|
||||
verbose_name = _(u'document checkout')
|
||||
verbose_name_plural = _(u'document checkouts')
|
||||
|
||||
|
||||
#class BackupJobLog
|
||||
|
||||
Reference in New Issue
Block a user