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