plstackapi/core/models/subnet.py

   1 import os
   2 import commands
   3 from django.db import models
   4 from plstackapi.core.models import PlCoreBase
   5 from plstackapi.core.models import Slice
   6 from plstackapi.openstack.driver import OpenStackDriver
   7
   8 # Create your models here.
   9
  10 class Subnet(PlCoreBase):
  11     subnet_id = models.CharField(max_length=256, unique=True)
  12     cidr = models.CharField(max_length=20)
  13     ip_version = models.IntegerField()
  14     start = models.IPAddressField()
  15     end = models.IPAddressField()
  16     slice = models.ForeignKey(Slice, related_name='subnet')
  17
  18     def __unicode__(self):  return u'%s' % (self.slice.name)
  19
  20     def save(self, *args, **kwds):
  21
  22         driver = OpenStackDriver()
  23         if not self.subnet_id:
  24             quantum_subnet = driver.create_subnet(name= self.slice.name,
  25                                           network_id=self.slice.network_id,
  26                                           cidr_ip = self.cidr,
  27                                           ip_version=self.ip_version,
  28                                           start = self.start,
  29                                           end = self.end)
  30             self.subnet_id = quantum_subnet['id']
  31             # add subnet as interface to slice's router
  32             driver.add_router_interface(self.slice.router_id, self.subnet_id)
  33             add_route = 'route add -net %s dev br-ex gw 10.100.0.5' % self.cidr
  34             commands.getstatusoutput(add_route)
  35
  36         super(Subnet, self).save(*args, **kwds)
  37
  38     def delete(self, *args, **kwds):
  39         driver = OpenStackDriver()
  40         if self.subnet_id:
  41             driver.delete_router_interface(self.slice.router_id, self.subnet_id)
  42             driver.delete_subnet(self.subnet_id)
  43             del_route = 'route del -net %s' % self.cidr
  44         commands.getstatusoutput(del_route)
  45         super(Subnet, self).delete(*args, **kwds)