from collections import namedtuple
from datetime import datetime

import psutil
from peewee import *
from playhouse.shortcuts import model_to_dict

import config
from database import BaseModel

from .plugin import Plugin


class DiskUsage(BaseModel):
    time = DateTimeField(index=True, default=datetime.now)
    partition = TextField(null=False)
    mountpoint = TextField(null=False)
    total = BigIntegerField(null=False)
    used = BigIntegerField(null=False)
    free = BigIntegerField(null=False)


class DiskIO(BaseModel):
    time = DateTimeField(index=True, default=datetime.now)
    disk = TextField(null=True)
    read_count = FloatField(null=False)     # all values are per second
    write_count = FloatField(null=False)
    read_speed = FloatField(null=False)
    write_speed = FloatField(null=False)


class DiskPlugin(Plugin):
    models = [DiskUsage, DiskIO]

    def __init__(self):
        self.__i = 0
        self.__previous_io = {}

    def store_io(self, disk, current):
        previous = self.__previous_io.get(disk, current)

        entry = DiskIO()
        entry.disk = disk
        entry.read_count = (current.read_count - previous.read_count) / config.INTERVAL
        entry.write_count = (current.write_count - previous.write_count) / config.INTERVAL
        entry.read_speed = (current.read_bytes - previous.read_bytes) / config.INTERVAL
        entry.write_speed = (current.write_bytes - previous.write_bytes) / config.INTERVAL
        entry.save()

        self.__previous_io[disk] = current

    def execute(self):

        # Collect disk usage
        if (self.__i % config.DISK_SPACE_FREQUENCY) == 0:
            for partition in psutil.disk_partitions():
                usage = psutil.disk_usage(partition.mountpoint)
                
                entry = DiskUsage()
                entry.partition = partition.device
                entry.mountpoint = partition.mountpoint
                entry.total = usage.total
                entry.used = usage.used
                entry.free = usage.free
                entry.save()

        # Collect IO
        self.store_io(None, psutil.disk_io_counters(perdisk=False))

        io_reads = psutil.disk_io_counters(perdisk=True)
        for disk, current in io_reads.items():
            self.store_io(disk, current)

        self.__i += 1