[WIP]Tick aggregate functions

seshdash/admin.py

@@ -95,3 +95,7 @@ class StatusRule(GuardedModelAdmin):
 @admin.register(Report)
 class ReportDisplay(GuardedModelAdmin):
     pass
+
+@admin.register(Tick_Script)
+class TickScript(GuardedModelAdmin):
+    pass

seshdash/migrations/0004_tick_script.py

@@ -0,0 +1,29 @@
+# -*- coding: utf-8 -*-
+# Generated by Django 1.9.1 on 2016-12-21 15:25
+from __future__ import unicode_literals
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('seshdash', '0003_auto_20161212_1209'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='Tick_Script',
+            fields=[
+                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('input_field_name', models.CharField(max_length=40)),
+                ('output_field_name', models.CharField(max_length=40)),
+                ('script', models.TextField()),
+                ('function', models.CharField(max_length=20)),
+                ('interval', models.CharField(max_length=10)),
+                ('type', models.CharField(choices=[(b'stream', b'Stream'), (b'batch', b'Batch')], max_length=10)),
+                ('site', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='seshdash.Sesh_Site')),
+            ],
+        ),
+    ]

seshdash/migrations/0005_auto_20161222_1026.py

@@ -0,0 +1,20 @@
+# -*- coding: utf-8 -*-
+# Generated by Django 1.9.1 on 2016-12-22 08:26
+from __future__ import unicode_literals
+
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('seshdash', '0004_tick_script'),
+    ]
+
+    operations = [
+        migrations.RenameField(
+            model_name='tick_script',
+            old_name='type',
+            new_name='script_type',
+        ),
+    ]

seshdash/models.py

@@ -811,3 +811,24 @@ class Data_Process_Rule(models.Model):
     duration = models.CharField(max_length=40, default="24h", choices=DURATION)
     interval = models.CharField(max_length=10, default="5m", choices=TIME_BUCKETS)
     output_field = models.ForeignKey(Trend_Data_Point)
+
+
+class Tick_Script(models.Model):
+    """ 
+    This is a model to hold data about the 
+    tick scripts
+    """
+    SCRIPT_TYPE_CHOICES = (
+        ('stream', 'Stream'),
+        ('batch', 'Batch'),
+    )
+
+    site = models.ForeignKey(Sesh_Site)
+    input_field_name = models.CharField(max_length=40)  # The influx measurement to operate on
+    output_field_name = models.CharField(max_length=40)  # The influx measurement to output results of the operation
+    script = models.TextField() 
+    function = models.CharField(max_length=20) # Influx function operation to use
+    interval = models.CharField(max_length=10) # The interval the task script runs on e.g 1min, 1h, 1d, 1w
+    script_type = models.CharField(max_length=10, choices=SCRIPT_TYPE_CHOICES)
+
+

seshdash/utils/reporting.py

@@ -1,12 +1,15 @@
 from seshdash.models import Sesh_Site,Site_Weather_Data,BoM_Data_Point, Alert_Rule, Sesh_Alert,Daily_Data_Point, Sesh_User
 from seshdash.utils.send_mail import send_mail
 from seshdash.utils.model_tools import get_measurement_unit
+from seshdash.data.db.kapacitor import Kapacitor
 
 from django.forms.models import model_to_dict
 from django.utils import timezone
 from django.db.models import Avg, Sum
 from django.apps import apps
 from django.core.exceptions import FieldError
+from django.conf import settings
+from django.template.loader import get_template
 
 from guardian.shortcuts import get_users_with_perms
 from datetime import datetime
@@ -213,3 +216,27 @@ def get_table_report_dict(report_table_name, operations):
                 report_table_attributes.append(report_attribute_dict)
 
     return report_table_attributes
+
+
+"""
+Kapacitor report utils
+"""
+def add_report_kap_task():
+    """  
+    This function adds a report task to kapacitor
+    """
+    data = {
+        'database': settings.INFLUX_DB,
+        'operator': 'mean',
+        'field': 'battery_voltage',
+        'output_field': 'mean_battery_voltage',
+        'duration': '5m',
+        'site_id': '1'
+    }
+
+    kap = Kapacitor()
+    t = get_template('seshdash/kapacitor_tasks/aggregate_report.tick')
+    print "About to send this template: %s" % (t.render(data))
+    response = kap.create_task('testing', t.render(data))
+
+    return response

templates/seshdash/kapacitor_tasks/aggregate_report.tick

@@ -0,0 +1,24 @@
+// Example report tick script
+// A user wants to get average(mean) for a field battery_voltage in a db called sesh
+// The result will be outputed in a field called mean_battery_voltage in the sesh db
+
+
+var database = '{{ database }}'
+
+var duration = '{{ duration }}'
+
+var output_field = '{{ output_field }}'
+
+var points = batch
+    |query ('''
+               SELECT {{operator}}(value)
+               FROM "{{database}}"."autogen"."{{ field }}"
+               WHERE site_id={{site_id}}
+            ''')
+        .period(duration)
+        .every(duration)
+
+points
+    |influxDBOut()
+        .database(database)
+        .measurement(output_field)