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 Controller, ControllerSites
9 from core.models.slice import Slice, ControllerSlices
10 from core.models.usercontrollers import ControllerUsers
11 from util.logger import Logger, logging
12 from observer.ansible import *
14 logger = Logger(level=logging.INFO)
16 class SyncControllerSlices(OpenStackSyncStep):
17 provides=[ControllerSlices]
20 def fetch_pending(self, deleted):
22 return ControllerSlices.deleted_objects.all()
24 return ControllerSlices.objects.filter(Q(enacted__lt=F('updated')) | Q(enacted=None))
26 def get_next_subnet(self, controller=None):
27 # limit ourself to 10.0.x.x for now
28 valid_subnet = lambda net: net.startswith('10.0')
29 driver = self.driver.admin_driver(controller=controller)
30 subnets = driver.shell.quantum.list_subnets()['subnets']
31 ints = [int(IPNetwork(subnet['cidr']).ip) for subnet in subnets \
32 if valid_subnet(subnet['cidr'])]
35 last_ip = IPAddress(ints[-1])
37 last_ip = IPAddress('10.0.0.1')
38 last_ip = IPAddress(ints[-1])
39 last_network = IPNetwork(str(last_ip) + "/24")
40 next_network = IPNetwork(str(IPAddress(last_network) + last_network.size) + "/24")
44 def sync_record(self, controller_slice):
45 logger.info("sync'ing slice controller %s" % controller_slice)
47 if not controller_slice.controller.admin_user:
48 logger.info("controller %r has no admin_user, skipping" % controller_slice.controller)
51 controller_users = ControllerUsers.objects.filter(user=controller_slice.slice.creator,
52 controller=controller_slice.controller)
53 if not controller_users:
54 logger.info("slice createor %s has not accout at controller %s" % (controller_slice.slice.creator, controller_slice.controller.name))
57 controller_user = controller_users[0]
60 max_instances=int(controller_slice.slice.max_slivers)
61 tenant_fields = {'endpoint':controller_slice.controller.auth_url,
62 'admin_user': controller_slice.controller.admin_user,
63 'admin_password': controller_slice.controller.admin_password,
64 'admin_tenant': 'admin',
65 'tenant': controller_slice.slice.name,
66 'tenant_description': controller_slice.slice.description,
68 'name':controller_user.user.email,
69 'max_instances':max_instances}
71 res = run_template('sync_controller_slices.yaml', tenant_fields)
72 expected_num = len(roles)+1
73 if (len(res)!=expected_num):
74 raise Exception('Could not sync tenants for slice %s'%controller_slice.slice.name)
76 tenant_id = res[0]['id']
77 if (not controller_slice.tenant_id):
78 handle = os.popen('nova quota-update --instances %d %s'%(max_instances,tenant_id))
79 output = handle.read()
80 result = handle.close()
82 logging.info('Could not update quota for %s'%controller_slice.slice.name)
83 controller_slice.tenant_id = tenant_id
84 controller_slice.save()
88 def delete_record(self, controller_slice):
89 user = User.objects.get(id=controller_slice.slice.creator.id)
90 driver = OpenStackDriver().admin_driver(controller=controller_slice.controller.name)
91 client_driver = driver.client_driver(caller=user,
92 tenant=controller_slice.slice.name,
93 controller=controller_slice.controller.name)
95 if controller_slice.router_id and controller_slice.subnet_id:
96 client_driver.delete_router_interface(controller_slice.router_id, controller_slice.subnet_id)
97 if controller_slice.subnet_id:
98 client_driver.delete_subnet(controller_slice.subnet_id)
99 if controller_slice.router_id:
100 client_driver.delete_router(controller_slice.router_id)
101 if controller_slice.network_id:
102 client_driver.delete_network(controller_slice.network_id)
103 if controller_slice.tenant_id:
104 driver.delete_tenant(controller_slice.tenant_id)