from django.db import models
from django.contrib.auth import get_user_model

from django.contrib.postgres.fields import JSONField

class VM(models.Model):
    vmid = models.IntegerField(primary_key=True)
    owner = models.ForeignKey(get_user_model(), on_delete=models.CASCADE)
    data = JSONField()


    def cores(self):
        return self.data['TEMPLATE']['VCPU']

    def ram_in_gb(self):
        return (int(self.data['TEMPLATE']['MEMORY'])/1024.)

    def disks(self):
        """
        If there is no disk then the key DISK does not exist.

        If there is only one disk, we have a dictionary in the database.

        If there are multiple disks, we have a list of dictionaries in the database.
        """

        if not 'DISK' in self.data['TEMPLATE']['DISK']:
            return []
        elif type(self.data['TEMPLATE']['DISK']) is dict:
            return [ self.data['TEMPLATE']['DISK'] ]
        else:
            return self.data['TEMPLATE']['DISK']