staLUN: EMC LUN statistics

# -*- coding: utf-8 -*- vim:encoding=utf-8:

# vim: tabstop=4:shiftwidth=4:softtabstop=4:expandtab

from core.models import *

from graphs.models import *

from django.core.urlresolvers import reverse

from django.shortcuts import render_to_response,get_object_or_404

from django.http import HttpResponse,HttpResponseRedirect,Http404

from django.conf import settings

from django.views.decorators.cache import cache_page

from django.views.decorators.cache import never_cache

from django.core.cache import cache

from django.template import RequestContext

import json

import datetime

import time

import rrdtool

import tempfile

import os

import re

import random

import pprint

from gevent.pool import Pool

from gevent.timeout import Timeout

from util.analyze import analyze

def index(request):

    luns = Lun.objects.all()

    graphs = Graph.objects.all()

    nl = cache.get('lun:stats')

    if nl is None:

        nl = analyze(luns)

    urls = []

    for i in nl:

        lun = luns.get(pk=i['lunpk'])

        graph = graphs.get(pk=i['graph'])

        if int(i['avg']) == 0:

            continue 

        #for graph in lungraphs:

        graph_dict = {}

        graph.url = reverse(thumb, args=(graph.pk,))

        graph_dict['url'] = graph.url

        graph_dict['graph'] = graph

        graph_dict['lun'] = lun.name

        urls.append(graph_dict)

    ret = {

           'urls' :urls

           }

    return render_to_response('media.html', ret,

                context_instance=RequestContext(request))

def graphjson(request, graph_id):

    ret = {"graph_pk": graph_id}

    return render_to_response('graph.html', ret,  

                context_instance=RequestContext(request))

def graphperiods(request, graph_id):

    graph =  Graph.objects.get(pk=graph_id)

    return render_to_response('graphperiods.html', {'graph': graph, },

                context_instance=RequestContext(request))

def get_all_json(lazy=False, legend=False, start='-2d', end='-8m'):

    graphs = Graph.objects.all()

    kwargs = { 'start': str(start), 'end': str(end), 'lazy': lazy, 'legend': legend, 'jsonexport':True }

    results = []

    for graph in graphs:

        results.extend(graph.create_graph(path=None, **kwargs))

#     p = Pool(40)

#     

#     def _get_graphs_json(graph):

#         t = Timeout(5)

#         t.start()

#         try:

#             results.extend(graph.create_graph(path=None, **kwargs))

#         except (Timeout):

#             pass

#         finally:

#             t.cancel()

#     p.imap(_get_graphs_json, graphs)

    return results

def getpngdata(path,static):

    file = open(path,'r')

    data = file.read()

    file.close()

    if static == False:

        os.remove(path)

    response = HttpResponse(mimetype='image/png')

    response.write(data)

    return response

def drawpng(request, graph_id, static=True, lazy=False, legend=False, start='-30h', end='-8m', jsonexport=False):

    graph = get_object_or_404(Graph,pk=graph_id)

    if static == True:

        lazy = True

        path = '%s/%s%s%s.png' % (settings.RRD_PNG_STATICPATH, graph_id, start, end)

        try:

            statinfo = os.stat(path)

            if time.time() - statinfo.st_mtime < 300 :

                return getpngdata(path,static)

        except OSError:

            pass

    else:

        path = tempfile.NamedTemporaryFile().name

    '''Removing unicode from strings while constructing kwargs'''

    kwargs = { 'start': str(start), 'end': str(end), 'lazy': lazy, 'legend': legend, 'jsonexport':jsonexport }

    result = graph.create_graph(path, **kwargs)

    if result == None:

        raise Http404

    if jsonexport:

        return HttpResponse(json.dumps(result), mimetype="application/json")

    return getpngdata(path, static)

def thumb(request,graph_id,legend=True,lazy=True,static=True):

    start='-1d'

    end='-300'

    graph = get_object_or_404(Graph,pk=graph_id)

    path = '%s/thumb-%s.png' % (settings.RRD_PNG_STATICPATH, graph_id)

    try:

        statinfo = os.stat(path)

        if time.time() - statinfo.st_mtime < 300 :

            return getpngdata(path,static)

    except OSError:

        pass

    '''Removing unicode from strings while constructing kwargs'''

    kwargs = { 'start': str(start), 'end': str(end), 'lazy': lazy, 'legend': legend, 'thumb': True, }

    result=graph.create_graph(path,**kwargs)

    return getpngdata(path,static)