blob: 08a3ff27468932e05d47476cc322a9226b61572c [file] [log] [blame]
import os
from datetime import datetime
from django.db import models
from plstackapi.openstack.driver import OpenStackDriver
# Create your models here.
class PlCoreBase(models.Model):
created = models.DateTimeField(auto_now_add=True)
updated = models.DateTimeField(auto_now=True)
class Meta:
abstract = True
def save(self):
if not self.id:
self.created = datetime.date.today()
self.updated = datetime.datetime.today()
super(PlCoreBase, self).save()
class Role(PlCoreBase):
ROLE_CHOICES = (('admin', 'Admin'), ('pi', 'Principle Investigator'), ('user','User'))
role_id = models.CharField(max_length=256, unique=True)
role_type = models.CharField(max_length=80, unique=True, choices=ROLE_CHOICES)
def __unicode__(self): return u'%s' % (self.role_type)
def save(self):
if not self.id:
self.created = datetime.date.today()
self.updated = datetime.datetime.today()
super(Role, self).save()
class Site(PlCoreBase):
tenant_id = models.CharField(max_length=200, help_text="Keystone tenant id")
name = models.CharField(max_length=200, help_text="Name for this Site")
site_url = models.URLField(null=True, blank=True, max_length=512, help_text="Site's Home URL Page")
enabled = models.BooleanField(default=True, help_text="Status for this Site")
longitude = models.FloatField(null=True, blank=True)
latitude = models.FloatField(null=True, blank=True)
login_base = models.CharField(max_length=50, help_text="Prefix for Slices associated with this Site")
is_public = models.BooleanField(default=True, help_text="Indicates the visibility of this site to other members")
abbreviated_name = models.CharField(max_length=80)
def __unicode__(self): return u'%s' % (self.name)
def save(self, *args, **kwargs):
driver = OpenStackDriver()
if not self.id:
tenant = driver.create_tenant(tenant_name=self.login_base,
description=self.name,
enabled=self.enabled)
self.tenant_id = tenant.id
else:
# update record
self.driver.update_tenant(self.tenant_id, name=self.login_base,
description=self.name, enabled=self.enabled)
super(Site, self).save(*args, **kwargs)
def delete(self, *args, **kwds):
# delete keystone tenant
driver = OpenStackDriver()
driver.delete_tenant(self.tenant_id)
super(Site, self).delete(*args, **kwargs)
class User(PlCoreBase):
user_id = models.CharField(max_length=256, unique=True)
firstname = models.CharField(help_text="person's given name", max_length=200)
lastname = models.CharField(help_text="person's surname", max_length=200)
email = models.EmailField(help_text="e-mail address")
phone = models.CharField(null=True, blank=True, help_text="phone number contact", max_length=100)
user_url = models.URLField(null=True, blank=True)
is_admin = models.BooleanField(default=False)
site = models.ForeignKey(Site, related_name='site_user', verbose_name="Site this user will be homed too")
def __unicode__(self): return u'%s' % (self.email)
def save(self, *args, **kwds):
driver = OpenStackDriver()
name = self.email[:self.email.find('@')]
fields = {'name': name, 'email': self.email, 'password': self.password,
'enabled': self.enabled}
if not self.id:
user = driver.create_user(**fields)
else:
driver.update_user(self.user_id, **fields)
super(User, self).save(*args, **kwds)
def delete(self, *args, **kwds):
driver = OpenStackDriver()
driver.delete_user(self.user_id)
super(User, self).delete(*args, **kwds)
class SitePrivilege(PlCoreBase):
user = models.ForeignKey('User')
site = models.ForeignKey('Site')
role = models.ForeignKey('Role')
def __unicode__(self): return u'%s %s %s' % (self.site, self.user, self.role)
def save(self, *args, **kwds):
driver = OpenStackDriver()
driver.add_user_role(user_id=user.user_id,
tenant_id=site.tenant_id,
role_name=role.name)
super(SitePrivilege, self).save(*args, **kwds)
def delete(self, *args, **kwds):
driver = OpenStackDriver()
driver.delete_user_role(user_id=user.user_id,
tenant_id=site.tenant_id,
role_name=role.name)
super(SitePrivilege, self).delete(*args, **kwds)
class DeploymentNetwork(PlCoreBase):
name = models.CharField(max_length=200, unique=True, help_text="Name of the Deployment Network")
def __unicode__(self): return u'%s' % (self.name)
class SiteDeploymentNetwork(PlCoreBase):
class Meta:
unique_together = ['site', 'deploymentNetwork']
site = models.ForeignKey(Site, related_name='deploymentNetworks')
deploymentNetwork = models.ForeignKey(DeploymentNetwork, related_name='sites')
name = models.CharField(default="Blah", max_length=100)
def __unicode__(self): return u'%s::%s' % (self.site, self.deploymentNetwork)
class Slice(PlCoreBase):
tenant_id = models.CharField(max_length=200, help_text="Keystone tenant id")
name = models.CharField(help_text="The Name of the Slice", max_length=80)
enabled = models.BooleanField(default=True, help_text="Status for this Slice")
SLICE_CHOICES = (('plc', 'PLC'), ('delegated', 'Delegated'), ('controller','Controller'), ('none','None'))
instantiation = models.CharField(help_text="The instantiation type of the slice", max_length=80, choices=SLICE_CHOICES)
omf_friendly = models.BooleanField()
description=models.TextField(blank=True,help_text="High level description of the slice and expected activities", max_length=1024)
slice_url = models.URLField(blank=True, max_length=512)
site = models.ForeignKey(Site, related_name='site_slice', help_text="The Site this Node belongs too")
network_id = models.CharField(max_length=256, help_text="Quantum network")
router_id = models.CharField(max_length=256, help_text="Quantum router id")
def __unicode__(self): return u'%s' % (self.name)
def save(self, *args, **kwds):
# sync keystone tenant
driver = OpenStackDriver()
if not self.id:
tenant = driver.create_tenant(tenant_name=self.name,
description=self.description,
enabled=self.enabled)
self.tenant_id = tenant.id
# create a network
network = driver.create_network(name=self.name)
self.network_id = network['id']
# create router
router = driver.create_router(name=self.name)
self.router_id = router['id']
else:
# update record
self.driver.update_tenant(self.tenant_id, name=self.name,
description=self.description, enabled=self.enabled)
super(Slice, self).save(*args, **kwds)
def delete(self, *args, **kwds):
# delete keystone tenant
driver = OpenStackDriver()
driver.delete_tenant(self.tenant_id)
super(Slice, self).delete(*args, **kwds)
class SliceMembership(PlCoreBase):
user = models.ForeignKey('User')
slice = models.ForeignKey('Slice')
role = models.ForeignKey('Role')
def __unicode__(self): return u'%s %s %s' % (self.slice, self.user, self.role)
def save(self, *args, **kwds):
driver = OpenStackDriver()
driver.add_user_role(user_id=user.user_id,
tenant_id=slice.tenant_id,
role_name=role.name)
super(SliceMembership, self).save(*args, **kwds)
def delete(self, *args, **kwds):
driver = OpenStackDriver()
driver.delete_user_role(user_id=user.user_id,
tenant_id=slice.tenant_id,
role_name=role.name)
super(SliceMembership, self).delete(*args, **kwds)
class SubNet(PlCoreBase):
subnet_id = models.CharField(max_length=256, unique=True)
cidr = models.CharField(max_length=20)
ip_version = models.IntegerField()
start = models.IPAddressField()
end = models.IPAddressField()
slice = models.ForeignKey(Slice, related_name='slice_subnet')
def __unicode__(self): return u'%s' % (self.name)
def save(self, *args, **kwargs):
driver = OpenStackDriver()
if not self.id:
subnet = driver.create_subnet(network_name=self.slice.name,
cidr_ip = self.cidr,
ip_version=self.ip_version,
start = self.start,
end = self.end)
self.subnet_id = subnet.id
# add subnet as interface to slice router
driver.add_router_interface(self.slice.router_id, subnet.id)
super(SubNet, self).save(*args, **kwargs)
def delete(self, *args, **kwargs):
# delete quantum network
driver = OpenStackDriver()
driver.delete_subnet(self.subnet_id)
driver.delete_router_interface(self.slice.router_id, self.subnet.id)
super(SubNet, self).delete(*args, **kwargs)
class Node(PlCoreBase):
name = models.CharField(max_length=200, unique=True, help_text="Name of the Node")
siteDeploymentNetwork = models.ForeignKey(SiteDeploymentNetwork, help_text="The Site and Deployment Network this Node belongs too.")
def __unicode__(self): return u'%s' % (self.name)
class Image(PlCoreBase):
image_id = models.CharField(max_length=256, unique=True)
name = models.CharField(max_length=256, unique=True)
disk_format = models.CharField(max_length=256)
container_format = models.CharField(max_length=256)
def __unicode__(self): return u'%s' % (self.name)
class Flavor(PlCoreBase):
flavor_id = models.IntegerField(unique=True)
name = models.CharField(max_length=256, unique=True)
memory_mb = models.IntegerField()
disk_gb = models.IntegerField()
vcpus = models.IntegerField()
def __unicode__(self): return u'%s' % (self.name)
class Key(PlCoreBase):
name = models.CharField(max_length=256, unique=True)
key = models.CharField(max_length=512)
type = models.CharField(max_length=256)
blacklisted = models.BooleanField()
user = models.ForeignKey(User)
def __unicode__(self): return u'%s' % (self.name)
def save(self, *args, **kwds):
driver = OpenStackDriver()
if not self.id:
keypair = driver.create_keypair(name=self.name, key=self.key)
super(Key, self).save(*args, **kwds)
def delete(self, *args, **kwds):
driver = OpenStackDriver()
driver.delete_keypair(self.name)
super(Key, self).delete(*args, **kwds)
class Sliver(PlCoreBase):
instance_id = models.CharField(max_length=200, help_text="Nova instance id")
name = models.CharField(max_length=200, help_text="Sliver name")
flavor = models.ForeignKey(Flavor, related_name='sliver_flavor')
image = models.ForeignKey(Image, related_name='sliver_image')
key = models.ForeignKey(Key, related_name='sliver_key')
slice = models.ForeignKey(Slice, related_name='sliver_slice')
siteDeploymentNetwork = models.ForeignKey(SiteDeploymentNetwork, related_name='sliver_deployment')
node = models.ForeignKey(Node, related_name='sliver_node')
def __unicode__(self): return u'%s::%s' % (self.slice, self.siteDeploymentNetwork)
def save(self, *args, **kwds):
driver = OpenStackDriver()
instance = driver.spawn_instances(name=self.name,
keyname=self.name,
hostnames=self.node.name,
flavor=self.flavor.name,
image=self.image.name)
self.instance_id = instance.id
super(Sliver, self).save(*args, **kwds)
def delete(self, *args, **kwds):
driver = OpenStackDriver()
driver.destroy_instance(name=self.name, id=self.instance_id)
super(Sliver, self).delete(*args, **kwds)