Inital code commit with upload funcationality

Author: maiiku

__init__.py

@@ -0,0 +1 @@
+__author__ = 'ubuntu'

apps/__init__.py

@@ -0,0 +1,34 @@
+# -*- coding: utf-8 -*-
+from django import forms
+from apps.pinger.models import Haystack
+from apps.util.forms import BootstrapErrorList
+
+
+class HaystackCreateModelForm(forms.ModelForm):
+    error_class=BootstrapErrorList
+
+    def __init__(self, *args, **kwargs):
+        super(HaystackCreateModelForm, self).__init__(*args, **kwargs)
+        self.error_class=BootstrapErrorList
+
+    class Meta:
+        model = Haystack
+        fields = ['url', 'search_phrase', 'name']
+
+
+class UploadFileForm(forms.Form):
+    file = forms.FileField()
+
+    def __init__(self, *args, **kwargs):
+        super(UploadFileForm, self).__init__(*args, **kwargs)
+        self.error_class=BootstrapErrorList
+
+    def clean_file(self):
+        '''
+        Check if uploaded file features correct syntax
+        '''
+        _file = self.cleaned_data["file"]
+        for (num, line) in enumerate(_file):
+            if len(line.split('|')) < 2:
+                raise forms.ValidationError("Invalid file format in line %d: %s" % (num+1, line))
+        return _file

apps/pinger/__init__.py

@@ -0,0 +1,68 @@
+# -*- coding: utf-8 -*-
+from django.core.urlresolvers import reverse
+from django.db import models
+from datetime import datetime as dt
+
+class BaseModel(models.Model):
+    '''
+    Model with common fields for all models
+    All models should inherit this class
+    '''
+    created = models.DateTimeField(auto_now=True)
+    modified = models.DateTimeField(auto_now=True)
+
+    def save(self, *args, **kwargs):
+        self.modified = dt.utcnow()
+        super(BaseModel, self).save(*args, **kwargs)
+
+
+class Haystack(BaseModel):
+    '''
+    Model holding urls to be checked
+    '''
+    name = models.CharField(max_length=255, blank=True, null=True)
+    url = models.CharField(max_length=255, blank=False, null=False)
+    search_phrase = models.CharField(max_length=255, blank=True, null=True)
+    last_checked = models.DateTimeField(null=True)
+    last_ok = models.DateTimeField(null=True)
+    last_error = models.DateTimeField(null=True)
+    last_time_ms = models.IntegerField(default=0)
+
+    def __unicode__(self):
+        return str(self.url)
+
+    class Meta:
+        unique_together = ("url", "search_phrase")
+
+    def get_absolute_url(self):
+        return reverse('hay_update', kwargs={'pk': self.pk})
+
+    def get_delete_url(self):
+        return reverse('hay_delete', kwargs={'pk': self.pk})
+
+class PingHistory(BaseModel):
+    '''
+    Model holding health check history
+    '''
+    haystack = models.ForeignKey('Haystack')
+    last_time_ms = models.IntegerField(default=0)
+    http_code=models.IntegerField(default=0)
+    time_ms = models.IntegerField(default=0)
+    phrase_found = models.BooleanField(default=False)
+
+    @property
+    def is_error(self):
+        '''
+        Everything else then 200 OK is not ok
+        '''
+        return True if self.http_code != 200 and self.phrase_found else False
+
+    def save(self, *args, **kwargs):
+        self.haystack.last_time_ms = self.time_ms
+        self.haystack.last_checked = self.created
+        if not self.is_error:
+            self.haystack.last_ok = dt.utcnow()
+        else:
+            self.haystack.last_error = dt.utcnow()
+        super(self, PingHistory).save(*args, **kwargs)
+

apps/pinger/forms.py

@@ -0,0 +1,16 @@
+"""
+This file demonstrates writing tests using the unittest module. These will pass
+when you run "manage.py test".
+
+Replace this with more appropriate tests for your application.
+"""
+
+from django.test import TestCase
+
+
+class SimpleTest(TestCase):
+    def test_basic_addition(self):
+        """
+        Tests that 1 + 1 always equals 2.
+        """
+        self.assertEqual(1 + 1, 2)

apps/pinger/models.py

@@ -0,0 +1,13 @@
+# -*- coding: utf-8 -*-
+from django.conf.urls import patterns, url
+from apps.pinger.views import HaystackUploadView, HaystackCreateView, HaystackListView, HaystackUpdateView, HaystackDeleteView
+
+urlpatterns = patterns('',
+    url(r'upload/$', HaystackUploadView.as_view(), name='hay_upload'),
+    url(r'add/$', HaystackCreateView.as_view(), name='hay_add'),
+    url(r'list/$', HaystackListView.as_view(), name='hay_list'),
+    url(r'list/(?P<pk>\d+)/$', HaystackUpdateView.as_view(), name='hay_update'),
+    url(r'list/(?P<pk>\d+)/delete/$', HaystackDeleteView.as_view(), name='hay_delete'),
+
+    url(r'$', HaystackListView.as_view(), name='home'),
+)

apps/pinger/ping.py

@@ -0,0 +1,61 @@
+# -*- coding: utf-8 -*-
+from django.http import HttpResponseRedirect
+from django.views.generic import CreateView, ListView, UpdateView, DeleteView
+from django.views.generic.base import TemplateResponseMixin, ContextMixin, View
+from apps.pinger.forms import UploadFileForm, HaystackCreateModelForm
+from apps.pinger.models import Haystack
+from django.core.urlresolvers import reverse_lazy
+from apps.util.mixins import MessageMixin
+
+class HaystackUploadView(TemplateResponseMixin, ContextMixin, View):
+    form_class = UploadFileForm
+    template_name = 'pinger/haystack_upload.html'
+    success_url = reverse_lazy('hay_list')
+
+    def get(self, request, *args, **kwargs):
+        return self.render_to_response(
+            self.get_context_data(form=self.form_class()))
+
+    def post(self, request, *args, **kwargs):
+        form = self.form_class(request.POST, request.FILES)
+        if form.is_valid():
+            #if file is valid, transfer contents to db
+            _file = form.cleaned_data["file"]
+            for line in _file:
+                self.line_to_haystack(line)
+            return HttpResponseRedirect(self.success_url)
+        else:
+            return self.render_to_response(self.get_context_data(form=form))
+
+    def line_to_haystack(self, line):
+        try:
+            (url, phrase, name) = line.split('|')
+        except ValueError:
+            (url, phrase) = line.split('|')
+            name = None
+        straw = Haystack(
+            url=url,
+            search_phrase=phrase,
+            name=name,
+        )
+        straw.save()
+
+class HaystackListView(ListView):
+    model = Haystack
+
+class HaystackCreateView(MessageMixin, CreateView):
+    model = Haystack
+    form_class = HaystackCreateModelForm
+    success_message='Added new item!'
+    success_url = reverse_lazy('hay_list')
+
+class HaystackUpdateView(MessageMixin, UpdateView):
+    model = Haystack
+    form_class = HaystackCreateModelForm
+    success_message='Data was successfully edited'
+    success_url = reverse_lazy('hay_list')
+
+class HaystackDeleteView(MessageMixin, DeleteView):
+    model = Haystack
+    success_message='Link was removed from monitoring system.'
+    success_url = reverse_lazy('hay_list')

apps/pinger/tests.py

@@ -0,0 +1,17 @@
+# -*- coding: utf-8 -*-
+from django.forms.util import ErrorList
+from django.utils.encoding import force_text
+from django.utils.html import format_html, format_html_join
+
+
+class BootstrapErrorList(ErrorList):
+    """
+    A collection of errors that knows how to display itself in various formats.
+    """
+    def as_ul(self):
+        if not self: return ''
+        return format_html('<ul class="errorlist">{0}</ul>',
+                           format_html_join('', '<li class="text-error">{0}</li>',
+                                            ((force_text(e),) for e in self)
+                                            )
+                           )

apps/pinger/urls.py

@@ -0,0 +1,14 @@
+from django.contrib import messages
+
+
+class MessageMixin(object):
+    """
+    Display notification messages using Class Based Views.
+    """
+    def delete(self, request, *args, **kwargs):
+        messages.success(self.request, self.success_message)
+        return super(MessageMixin, self).delete(request, *args, **kwargs)
+
+    def form_valid(self, form):
+        messages.success(self.request, self.success_message)
+        return super(MessageMixin, self).form_valid(form)

apps/pinger/views.py

@@ -0,0 +1,10 @@
+#!/usr/bin/env python
+import os
+import sys
+
+if __name__ == "__main__":
+    os.environ.setdefault("DJANGO_SETTINGS_MODULE", "hping.settings")
+
+    from django.core.management import execute_from_command_line
+
+    execute_from_command_line(sys.argv)

apps/util/__init__.py

@@ -0,0 +1 @@
+__author__ = 'ubuntu'