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 Deployment, SiteDeployments
9 from core.models.slice import Slice, SliceDeployments
10 from core.models.userdeployments import UserDeployments
11 from util.logger import Logger, logging
13 logger = Logger(level=logging.INFO)
15 class SyncSliceDeployments(OpenStackSyncStep):
16 provides=[SliceDeployments]
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 = SiteDeployments.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)
27 slice_deployments = SliceDeployments.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)
32 all_deployments = Deployment.objects.all()
33 for slice in Slice.objects.all():
34 # slices are added to all deployments for now
35 expected_deployments = all_deployments
36 #expected_deployments = site_deploy_lookup[slice.site]
37 for expected_deployment in expected_deployments:
38 if slice not in slice_deploy_lookup or \
39 expected_deployment not in slice_deploy_lookup[slice]:
40 sd = SliceDeployments(slice=slice, deployment=expected_deployment)
43 # now we can return all slice deployments that need to be enacted
44 return SliceDeployments.objects.filter(Q(enacted__lt=F('updated')) | Q(enacted=None))
46 def get_next_subnet(self, deployment=None):
47 # limit ourself to 10.0.x.x for now
48 valid_subnet = lambda net: net.startswith('10.0')
49 driver = self.driver.admin_driver(deployment=deployment)
50 subnets = driver.shell.quantum.list_subnets()['subnets']
51 ints = [int(IPNetwork(subnet['cidr']).ip) for subnet in subnets \
52 if valid_subnet(subnet['cidr'])]
55 last_ip = IPAddress(ints[-1])
57 last_ip = IPAddress('10.0.0.1')
58 last_ip = IPAddress(ints[-1])
59 last_network = IPNetwork(str(last_ip) + "/24")
60 next_network = IPNetwork(str(IPAddress(last_network) + last_network.size) + "/24")
64 def sync_record(self, slice_deployment):
65 logger.info("sync'ing slice deployment %s" % slice_deployment)
66 if not slice_deployment.tenant_id:
67 nova_fields = {'tenant_name': slice_deployment.slice.name,
68 'description': slice_deployment.slice.description,
69 'enabled': slice_deployment.slice.enabled}
70 driver = self.driver.admin_driver(deployment=slice_deployment.deployment.name)
71 tenant = driver.create_tenant(**nova_fields)
72 slice_deployment.tenant_id = tenant.id
74 # XXX give caller an admin role at the tenant they've created
75 deployment_users = UserDeployments.objects.filter(user=slice_deployment.slice.creator,
76 deployment=slice_deployment.deployment)
77 if not deployment_users:
78 logger.info("slice createor %s has not accout at deployment %s" % (slice_deployment.slice.creator, slice_deployment.deployment.name))
80 deployment_user = deployment_users[0]
81 # lookup user id at this deployment
82 kuser= driver.shell.keystone.users.find(email=slice_deployment.slice.creator.email)
84 # add required roles at the slice's tenant
85 driver.add_user_role(kuser.id, tenant.id, 'admin')
87 # refresh credentials using this tenant
88 client_driver = self.driver.client_driver(caller=deployment_user.user,
90 deployment=slice_deployment.deployment.name)
93 if slice_deployment.id and slice_deployment.tenant_id:
94 # update existing tenant
95 driver = self.driver.admin_driver(deployment=slice_deployment.deployment.name)
96 driver.update_tenant(slice_deployment.tenant_id,
97 description=slice_deployment.slice.description,
98 enabled=slice_deployment.slice.enabled)
100 if slice_deployment.tenant_id:
101 # update slice/tenant quota
102 driver = self.driver.client_driver(deployment=slice_deployment.deployment.name, tenant=slice_deployment.slice.name)
103 driver.shell.nova.quotas.update(tenant_id=slice_deployment.tenant_id, instances=int(slice_deployment.slice.max_slivers))
105 slice_deployment.save()