Powermeter

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

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

# Copyright 2012 Leonidas Poulopoulos

#

# Licensed under the Apache License, Version 2.0 (the "License");

# you may not use this file except in compliance with the License.

# You may obtain a copy of the License at

#

#    http://www.apache.org/licenses/LICENSE-2.0

#

# Unless required by applicable law or agreed to in writing, software

# distributed under the License is distributed on an "AS IS" BASIS,

# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

# See the License for the specific language governing permissions and

# limitations under the License.

from django.core.management.base import BaseCommand, CommandError

import urllib2

import pycurl

from django.conf import settings

from dcapp.models import *

import csv

import cStringIO

import time, datetime

class Command(BaseCommand):

    help = "Populates the devices' series measurements"

    args = ""

    label = ''

    def getUserData(self, dev_id):

        response = cStringIO.StringIO()

        c = pycurl.Curl()

        past = datetime.datetime.now() + datetime.timedelta(minutes=-30)

        start = time.mktime(past.timetuple())

        c.setopt(pycurl.URL, '%s?_=%s&device=%s&absolute=%s&tstart=%s&tfinish=2147483647&series=%s' %(settings.POWERMETER_DATA_URI, int(time.time()),dev_id, settings.POWERMETER_ABSOLUTE_MEASSUREMENT, start, ','.join(map(str, range(40))) ))

        c.setopt(pycurl.HTTPHEADER, ['Accept: text/plain, */*'])

        c.setopt(pycurl.VERBOSE, 0)

        c.setopt(pycurl.CONNECTTIMEOUT, 30)

        c.setopt(pycurl.TIMEOUT, 30)

        c.setopt(pycurl.USERPWD, '%s:%s' %(settings.POWERMETER_USER,settings.POWERMETER_PASS))

        c.setopt(c.WRITEFUNCTION, response.write)

        c.perform()

        return response

    def handle(self, *args, **options):

        devices = Device.objects.all()

        for device in devices:

            print "Now parsing: %s\n" %device

            data = self.getUserData(device.mapped_id)

            d = data.getvalue()

            html_text = urllib2.unquote(d)

            f = cStringIO.StringIO(html_text)

            reader = csv.DictReader(f)

            monitoring_list = []

            for i in reader:

                monitoring_list.append(i)

            device.graphCounters(monitoring_list)

            print "DONE with device...\n\n"