planetstack/observer/steps/sync_slice_deployments.py

   1 import os
   2 import base64
   3 from collections import defaultdict
   4 from netaddr import IPAddress, IPNetwork
   5 from django.db.models import F, Q
   6 from planetstack.config import Config
   7 from observer.openstacksyncstep import OpenStackSyncStep
   8 from core.models.site import SiteDeployments
   9 from core.models.slice import Slice, SliceDeployments
  10 from core.models.user import UserDeployments
  11 from util.logger import Logger, logging
  12
  13 logger = Logger(level=logging.INFO)
  14
  15 class SyncSliceDeployments(OpenStackSyncStep):
  16     provides=[Slice, SliceDeployments]
  17     requested_interval=0
  18
  19     def fetch_pending(self):
  20         # slice deployments are not visible to users. We must ensure
  21         # slices are deployed at all deploymets available to their site.
  22         site_deployments = SiteDeployment.objects.all()
  23         site_deploy_lookup = defaultdict(list)
  24         for site_deployment in site_deployments:
  25             site_deploy_lookup[site_deployment.site].append(site_deployment.deployment)
  26
  27         slice_deployments = SliceDeployment.objects.all()
  28         slice_deploy_lookup = defaultdict(list)
  29         for slice_deployment in slice_deployments:
  30             slice_deploy_lookup[slice_deployment.slice].append(slice_deployment.deployment)
  31
  32         for slice in Slice.objects.all():
  33             expected_deployments = site_deploy_lookup[slice.site]
  34             for expected_deployment in expected_deployments:
  35                 if slice not in slice_deploy_lookup or \
  36                    expected_deployment not in slice_deploy_lookup[slice]:
  37                     sd = SliceDeployments(slice=slice, deployment=expected_deployment)
  38                     sd.save()
  39
  40         # now we can return all slice deployments that need to be enacted
  41         return SliceDeployments.objects.filter(Q(enacted__lt=F('updated')) | Q(enacted=None))
  42
  43     def get_next_subnet(self, deployment=None):
  44         # limit ourself to 10.0.x.x for now
  45         valid_subnet = lambda net: net.startswith('10.0')
  46         driver = self.driver.admin_driver(deployment=deployment)
  47         subnets = driver.shell.quantum.list_subnets()['subnets']
  48         ints = [int(IPNetwork(subnet['cidr']).ip) for subnet in subnets \
  49                 if valid_subnet(subnet['cidr'])]
  50         ints.sort()
  51         last_ip = IPAddress(ints[-1])
  52         last_network = IPNetwork(str(last_ip) + "/24")
  53         next_network = IPNetwork(str(IPAddress(last_network) + last_network.size) + "/24")
  54         return next_network
  55
  56     def sync_record(self, slice_deployment):
  57         logger.info("sync'ing slice deployment %s" % slice_deployment)
  58         if not slice_deployment.tenant_id:
  59             nova_fields = {'tenant_name': slice_deployment.slice.name,
  60                    'description': slice_deployment.slice.description,
  61                    'enabled': slice_deployment.slice.enabled}
  62             driver = self.driver.admin_driver(deployment=slice_deployment.deployment.name)
  63             tenant = driver.create_tenant(**nova_fields)
  64             slice_deployment.tenant_id = tenant.id
  65
  66             # XXX give caller an admin role at the tenant they've created
  67             deployment_users = UserDeployments.objects.filter(user=slice_deployment.slice.creator,
  68                                                              deployment=slice_deployment.deployment)
  69             if not deployment_users or not deployment_users[0].kuser_id:
  70                 logger.info("slice createor %s has not accout at deployment %s" % (slice_deployment.slice.creator, slice_deployment.deployment.name)
  71             else:
  72                 driver.add_user_role(slice_deployment.slice.creator.kuser_id, tenant.id, 'admin')
  73
  74                 # refresh credentials using this tenant
  75                 client_driver = self.driver.client_driver(tenant=tenant.name,
  76                                                           deployment=slice_deployment.deployment.name)
  77
  78                 # create network
  79                 network = client_driver.create_network(slice.name)
  80                 slice_deployment.network_id = network['id']
  81
  82                 # create router
  83                 router = client_driver.create_router(slice.name)
  84                 slice_deployment.router_id = router['id']
  85
  86                 # create subnet for slice's private network
  87                 next_subnet = self.get_next_subnet(deployment=slice_deployment.deployment.name)
  88                 cidr = str(next_subnet.cidr)
  89                 ip_version = next_subnet.version
  90                 start = str(next_subnet[2])
  91                 end = str(next_subnet[-2])
  92                 subnet = client_driver.create_subnet(name=slice.name,
  93                                                    network_id = network['id'],
  94                                                    cidr_ip = cidr,
  95                                                    ip_version = ip_version,
  96                                                    start = start,
  97                                                    end = end)
  98                 slice_deployment.subnet_id = subnet['id']
  99                 # add subnet as interface to slice's router
 100                 client_driver.add_router_interface(router['id'], subnet['id'])
 101                 # add external route
 102                 client_driver.add_external_route(subnet)
 103
 104
 105         if slice_deployment.id and slice_deployment.tenant_id:
 106             driver = self.driver.admin_driver(deployment=slice_deployment.deployment.name)
 107             driver.update_tenant(slice_deployment.tenant_id,
 108                                  description=slice_deployment.slice.description,
 109                                  enabled=slice_deployment.slice.enabled)
 110
 111         slice_deployment.save()