Improvements on resource monitoring

TODO.md

@@ -48,5 +48,4 @@ Remember that, as always with QuerySets, any subsequent chained methods which im
 * passlib; nano /usr/local/lib/python2.7/dist-packages/passlib/ext/django/utils.py SortedDict -> collections.OrderedDict
 * pip install pyinotify
 
-
+* create custom field that returns backend python objects
-* Backend.operations dynamically generated based on defined methods

orchestra/apps/accounts/admin.py

@@ -71,6 +71,8 @@ class AccountAdmin(ExtendedModelAdmin):
     
     def save_model(self, request, obj, form, change):
         """ Save user and account, they are interdependent """
+        if change:
+            return super(AccountAdmin, self).save_model(request, obj, form, change)
         obj.user.save()
         obj.user_id = obj.user.pk
         obj.save()

orchestra/apps/domains/api.py

@@ -22,7 +22,9 @@ class DomainViewSet(AccountApiMixin, viewsets.ModelViewSet):
     @link()
     def view_zone(self, request, pk=None):
         domain = self.get_object()
-        return Response({'zone': domain.render_zone()})
+        return Response({
+            'zone': domain.render_zone()
+        })
     
     def metadata(self, request):
         ret = super(DomainViewSet, self).metadata(request)

orchestra/apps/orchestration/backends.py

@@ -69,7 +69,7 @@ class ServiceBackend(object):
     
     @classmethod
     def get_backend(cls, name):
-        for backend in ServiceMonitor.get_backends():
+        for backend in ServiceBackend.get_backends():
             if backend.get_name() == name:
                 return backend
         raise KeyError('This backend is not registered')

orchestra/apps/orchestration/helpers.py

@@ -38,7 +38,7 @@ def message_user(request, logs):
     errors = total-successes
     if errors:
         msg = 'backends have' if errors > 1 else 'backend has'
-        msg = _("%d out of %d {0} fail to executed".format(msg))
+        msg = _("%d out of %d {0} fail to execute".format(msg))
         messages.warning(request, msg % (errors, total))
     else:
         msg = 'backends have' if successes > 1 else 'backend has'

orchestra/apps/orchestration/methods.py

@@ -40,12 +40,12 @@ def BashSSH(backend, log, server, cmds):
         channel = transport.open_session()
         
         sftp = paramiko.SFTPClient.from_transport(transport)
-        sftp.put(path, path)
+        sftp.put(path, "%s.remote" % path)
         sftp.close()
         os.remove(path)
         
         context = {
-            'path': path,
+            'path': "%s.remote" % path,
             'digest': digest
         }
         cmd = (

orchestra/apps/orchestration/models.py

@@ -4,6 +4,7 @@ from django.core.exceptions import ValidationError
 from django.db import models
 from django.utils.translation import ugettext_lazy as _
 
+from orchestra.models.fields import NullableCharField
 from orchestra.utils.apps import autodiscover
 from orchestra.utils.functional import cached
 
@@ -14,9 +15,8 @@ from .backends import ServiceBackend
 class Server(models.Model):
     """ Machine runing daemons (services) """
     name = models.CharField(_("name"), max_length=256, unique=True)
-    # TODO unique address with blank=True (nullablecharfield)
+    address = NullableCharField(_("address"), max_length=256, blank=True,
-    address = models.CharField(_("address"), max_length=256, blank=True,
+            null=True, unique=True, help_text=_("IP address or domain name"))
-            help_text=_("IP address or domain name"))
     description = models.TextField(_("description"), blank=True)
     os = models.CharField(_("operative system"), max_length=32,
             choices=settings.ORCHESTRATION_OS_CHOICES,
@@ -82,8 +82,7 @@ class BackendOperation(models.Model):
     MONITOR = 'monitor'
     
     log = models.ForeignKey('orchestration.BackendLog', related_name='operations')
-    # TODO backend and backend_class() (like content_type)
+    backend = models.CharField(_("backend"), max_length=256)
-    backend_class = models.CharField(_("backend"), max_length=256)
     action = models.CharField(_("action"), max_length=64)
     content_type = models.ForeignKey(ContentType)
     object_id = models.PositiveIntegerField()
@@ -94,11 +93,11 @@ class BackendOperation(models.Model):
         verbose_name_plural = _("Operations")
     
     def __unicode__(self):
-        return '%s.%s(%s)' % (self.backend_class, self.action, self.instance)
+        return '%s.%s(%s)' % (self.backend, self.action, self.instance)
     
     def __hash__(self):
         """ set() """
-        backend = getattr(self, 'backend', self.backend_class)
+        backend = getattr(self, 'backend', self.backend)
         return hash(backend) + hash(self.instance) + hash(self.action)
     
     def __eq__(self, operation):
@@ -107,7 +106,7 @@ class BackendOperation(models.Model):
     
     @classmethod
     def create(cls, backend, instance, action):
-        op = cls(backend_class=backend.get_name(), instance=instance, action=action)
+        op = cls(backend=backend.get_name(), instance=instance, action=action)
         op.backend = backend
         return op
     
@@ -115,6 +114,9 @@ class BackendOperation(models.Model):
     def execute(cls, operations):
         return manager.execute(operations)
     
+    def backend_class(self):
+        return ServiceBackend.get_backend(self.backend)
+
 
 autodiscover('backends')
 

orchestra/apps/resources/admin.py

@@ -1,8 +1,9 @@
-from django.contrib import admin
+from django.contrib import admin, messages
 from django.contrib.contenttypes import generic
 from django.utils.functional import cached_property
 from django.utils.translation import ugettext_lazy as _
 
+from orchestra.admin import ExtendedModelAdmin
 from orchestra.admin.filters import UsedContentTypeFilter
 from orchestra.admin.utils import insertattr, get_modeladmin
 from orchestra.core import services
@@ -12,14 +13,34 @@ from .forms import ResourceForm
 from .models import Resource, ResourceData, MonitorData
 
 
-class ResourceAdmin(admin.ModelAdmin):
+class ResourceAdmin(ExtendedModelAdmin):
-    # TODO warning message server/celery should be restarted when creating things
-    
     list_display = (
         'name', 'verbose_name', 'content_type', 'period', 'ondemand',
-        'default_allocation', 'disable_trigger'
+        'default_allocation', 'disable_trigger', 'crontab',
     )
     list_filter = (UsedContentTypeFilter, 'period', 'ondemand', 'disable_trigger')
+    fieldsets = (
+        (None, {
+            'fields': ('name', 'content_type', 'period'),
+        }),
+        (_("Configuration"), {
+            'fields': ('verbose_name', 'default_allocation', 'ondemand',
+                       'disable_trigger', 'is_active'),
+        }),
+        (_("Monitoring"), {
+            'fields': ('monitors', 'crontab'),
+        }),
+    )
+    change_readonly_fields = ('name', 'content_type', 'period')
+    
+    def add_view(self, request, **kwargs):
+        """ Warning user if the node is not fully configured """
+        if request.method == 'GET':
+            messages.warning(request, _(
+                "Restarting orchestra and celery is required to fully apply changes. "
+                "Remember that allocated values will be applied when objects are saved"
+            ))
+        return super(ResourceAdmin, self).add_view(request, **kwargs)
     
     def save_model(self, request, obj, form, change):
         super(ResourceAdmin, self).save_model(request, obj, form, change)

orchestra/apps/resources/backends.py

@@ -1,4 +1,9 @@
+import datetime
+
+from django.contrib.contenttypes.models import ContentType
+
 from orchestra.apps.orchestration import ServiceBackend
+from orchestra.utils.functional import cached
 
 
 class ServiceMonitor(ServiceBackend):
@@ -14,14 +19,34 @@ class ServiceMonitor(ServiceBackend):
         """ filter monitor classes """
         return [plugin for plugin in cls.plugins if ServiceMonitor in plugin.__mro__]
     
-    def store(self, stdout):
+    @cached
+    def get_last_date(self, obj):
+        from .models import MonitorData
+        try:
+            # TODO replace
+            #return MonitorData.objects.filter(content_object=obj).latest().date
+            ct = ContentType.objects.get(app_label=obj._meta.app_label, model=obj._meta.model_name)
+            return MonitorData.objects.filter(content_type=ct, object_id=obj.pk).latest().date
+        except MonitorData.DoesNotExist:
+            return self.get_current_date() - datetime.timedelta(days=1)
+    
+    @cached
+    def get_current_date(self):
+        return datetime.datetime.now()
+    
+    def store(self, log):
         """ object_id value """
-        for line in stdout.readlines():
+        from .models import MonitorData
+        name = self.get_name()
+        app_label, model_name = self.model.split('.')
+        ct = ContentType.objects.get(app_label=app_label, model=model_name.lower())
+        for line in log.stdout.splitlines():
             line = line.strip()
             object_id, value = line.split()
-            # TODO date
+            MonitorData.objects.create(monitor=name, object_id=object_id,
-            MonitorHistory.store(self.model, object_id, value, date)
+                    content_type=ct, value=value, date=self.get_current_date())
     
     def execute(self, server):
-        log = super(MonitorBackend, self).execute(server)
+        log = super(ServiceMonitor, self).execute(server)
+        self.store(log)
         return log

orchestra/apps/resources/forms.py

@@ -21,7 +21,6 @@ class ResourceForm(forms.ModelForm):
         super(ResourceForm, self).__init__(*args, **kwargs)
         if self.resource:
             self.fields['verbose_name'].initial = self.resource.verbose_name
-            self.fields['used'].initial = self.resource.get_used() # TODO
             if self.resource.ondemand:
                 self.fields['allocated'].required = False
                 self.fields['allocated'].widget = ReadOnlyWidget(None, '')

orchestra/apps/resources/models.py

@@ -1,7 +1,7 @@
 import datetime
 
 from django.db import models
-from django.contrib.contenttypes import generic
+from django.contrib.contenttypes.fields import GenericForeignKey, GenericRelation
 from django.contrib.contenttypes.models import ContentType
 from django.core import validators
 from django.utils.translation import ugettext_lazy as _
@@ -34,17 +34,22 @@ class Resource(models.Model):
             validators=[validators.RegexValidator(r'^[a-z0-9_\-]+$',
                         _('Enter a valid name.'), 'invalid')])
     verbose_name = models.CharField(_("verbose name"), max_length=256, unique=True)
-    content_type = models.ForeignKey(ContentType) # TODO filter by servicE?
+    content_type = models.ForeignKey(ContentType,
-    period = models.CharField(_("period"), max_length=16, choices=PERIODS,
+            help_text=_("Model where this resource will be hooked"))
-            default=LAST)
+    period = models.CharField(_("period"), max_length=16, choices=PERIODS, default=LAST,
-    ondemand = models.BooleanField(_("on demand"), default=False)
+            help_text=_("Operation used for aggregating this resource monitored data."))
+    ondemand = models.BooleanField(_("on demand"), default=False,
+            help_text=_("If enabled the resource will not be pre-allocated, "
+                        "but allocated under the application demand"))
     default_allocation = models.PositiveIntegerField(_("default allocation"),
+            help_text=_("Default allocation value used when this is not an "
+                        "on demand resource"),
             null=True, blank=True)
     is_active = models.BooleanField(_("is active"), default=True)
-    disable_trigger = models.BooleanField(_("disable trigger"), default=False)
+    disable_trigger = models.BooleanField(_("disable trigger"), default=False,
+            help_text=_("Disables monitor's resource exeeded and recovery triggers"))
     crontab = models.ForeignKey(CrontabSchedule, verbose_name=_("crontab"),
             help_text=_("Crontab for periodic execution"))
-    # TODO create custom field that returns backend python objects
     monitors = MultiSelectField(_("monitors"), max_length=256,
             choices=ServiceMonitor.get_choices())
     
@@ -58,10 +63,13 @@ class Resource(models.Model):
         try:
             task = PeriodicTask.objects.get(name=name)
         except PeriodicTask.DoesNotExist:
+            if self.is_active:
                 PeriodicTask.objects.create(name=name, task='resources.Monitor',
                                             args=[self.pk], crontab=self.crontab)
         else:
-            if task.crontab != self.crontab:
+            if not self.is_active:
+                task.delete()
+            elif task.crontab != self.crontab:
                 task.crontab = self.crontab
                 task.save()
     
@@ -97,7 +105,7 @@ class ResourceData(models.Model):
     last_update = models.DateTimeField(null=True)
     allocated = models.PositiveIntegerField(null=True)
     
-    content_object = generic.GenericForeignKey()
+    content_object = GenericForeignKey()
     
     class Meta:
         unique_together = ('resource', 'content_type', 'object_id')
@@ -159,7 +167,7 @@ class MonitorData(models.Model):
     date = models.DateTimeField(auto_now_add=True)
     value = models.PositiveIntegerField()
     
-    content_object = generic.GenericForeignKey()
+    content_object = GenericForeignKey()
     
     class Meta:
         get_latest_by = 'date'
@@ -170,7 +178,7 @@ class MonitorData(models.Model):
 
 
 def create_resource_relation():
-    relation = generic.GenericRelation('resources.ResourceData')
+    relation = GenericRelation('resources.ResourceData')
     for resources in Resource.group_by_content_type():
         model = resources[0].content_type.model_class()
         model.add_to_class('resources', relation)

orchestra/apps/resources/serializers.py

@@ -7,6 +7,9 @@ from .models import Resource, ResourceData
 
 
 class ResourceSerializer(serializers.ModelSerializer):
+    # TODO required allocation serializers (like resource form)
+    # TODO create missing ResourceData (like resource form)
+    # TODO make default allocation available on OPTIONS (like resource form)
     name = serializers.SerializerMethodField('get_name')
     
     class Meta:

orchestra/apps/users/roles/mail/backends.py

@@ -12,6 +12,7 @@ from . import settings
 class MailSystemUserBackend(ServiceController):
     verbose_name = _("Mail system user")
     model = 'mail.Mailbox'
+    # TODO related_models = ('resources__content_type') ??
     
     DEFAULT_GROUP = 'postfix'
     

orchestra/apps/websites/backends/apache.py

@@ -185,47 +185,48 @@ class Apache2Traffic(ServiceMonitor):
         context = self.get_context(site)
         self.append("""
             awk 'BEGIN {
-                ini = "%(start_date)s";
+                ini = "%(last_date)s";
-                end = "%(end_date)s";
+                end = "%(current_date)s";
                 
-                months["Jan"]="01";
+                months["Jan"] = "01";
-                months["Feb"]="02";
+                months["Feb"] = "02";
-                months["Mar"]="03";
+                months["Mar"] = "03";
-                months["Apr"]="04";
+                months["Apr"] = "04";
-                months["May"]="05";
+                months["May"] = "05";
-                months["Jun"]="06";
+                months["Jun"] = "06";
-                months["Jul"]="07";
+                months["Jul"] = "07";
-                months["Aug"]="08";
+                months["Aug"] = "08";
-                months["Sep"]="09";
+                months["Sep"] = "09";
-                months["Oct"]="10";
+                months["Oct"] = "10";
-                months["Nov"]="11";
+                months["Nov"] = "11";
-                months["Dec"]="12";
+                months["Dec"] = "12";
             } {
-                date = substr($4,2)
+                date = substr($4, 2)
-                year = substr(date,8,4)
+                year = substr(date, 8, 4)
-                month = months[substr(date,4,3)];
+                month = months[substr(date, 4, 3)];
-                day = substr(date,1,2)
+                day = substr(date, 1, 2)
-                hour = substr(date,13,2)
+                hour = substr(date, 13, 2)
-                minute = substr(date,16,2)
+                minute = substr(date, 16, 2)
-                second = substr(date,19,2);
+                second = substr(date, 19, 2);
                 line_date = year month day hour minute second
                 if ( line_date > ini && line_date < end)
                     if ( $10 == "" )
-                        sum+=$9
+                        sum += $9
                     else
-                        sum+=$10;
+                        sum += $10;
             } END {
                 print sum;
             }' %(log_file)s | {
                 read value
-                echo %(site_name)s $value
+                echo %(site_id)s $value
             }
             """ % context)
     
     def get_context(self, site):
+        # TODO log timezone!!
         return {
             'log_file': os.path.join(settings.WEBSITES_BASE_APACHE_LOGS, site.unique_name),
-            'start_date': '',
+            'last_date': self.get_last_date(site).strftime("%Y%m%d%H%M%S"),
-            'end_date': '',
+            'current_date': self.get_current_date().strftime("%Y%m%d%H%M%S"),
-            'site_name': '',
+            'site_id': site.pk,
         }

orchestra/models/fields.py

@@ -52,6 +52,12 @@ class MultiSelectField(models.CharField):
         return [ value for value,__ in arr_choices ]
 
 
+class NullableCharField(models.CharField):
+     def get_db_prep_value(self, value, connection=None, prepared=False):
+         return value or None
+
+
 if isinstalled('south'):
     from south.modelsinspector import add_introspection_rules
     add_introspection_rules([], ["^orchestra\.models\.fields\.MultiSelectField"])
+    add_introspection_rules([], ["^orchestra\.models\.fields\.NullableCharField"])