2 from types import StringTypes
3 from plstackapi.openstack.client import OpenStackClient
4 from plstackapi.openstack.driver import OpenStackDriver
5 from plstackapi.core.api.auth import auth_check
6 from plstackapi.core.models import Slice
7 from plstackapi.core.api.sites import _get_sites
9 def validate_name(name):
10 # N.B.: Responsibility of the caller to ensure that login_base
11 # portion of the slice name corresponds to a valid site, if
15 # 2. Begins with login_base (letters or numbers).
16 # 3. Then single underscore after login_base.
17 # 4. Then letters, numbers, or underscores.
18 good_name = r'^[a-z0-9]+_[a-zA-Z0-9_]+$'
20 not re.match(good_name, name):
21 raise Exception, "Invalid slice name: %s" % name
23 def _get_slices(filter):
24 if isinstance(filter, StringTypes) and filter.isdigit():
26 if isinstance(filter, int):
27 slices = Slice.objects.filter(id=filter)
28 elif isinstance(filter, StringTypes):
29 slices = Slice.objects.filter(name=filter)
30 elif isinstance(filter, dict):
31 slices = Slice.objects.filter(**filter)
37 def add_slice(auth, fields):
38 driver = OpenStackDriver(client = auth_check(auth))
39 validate_name(fields.get('name'))
40 login_base = fields['name'][:fields['name'].find('_')]
41 sites = _get_sites(login_base)
42 if sites: fields['site'] = sites[0]
43 slice = Slice(**fields)
46 nova_fields = {'tenant_name': slice.name,
47 'description': slice.description,
48 'enabled': slice.enabled}
49 tenant = driver.create_tenant(**nova_fields)
50 slice.tenant_id=tenant.id
53 network = driver.create_network(slice.name)
54 slice.network_id = network['id']
57 router = driver.create_router(slice.name)
58 slice.router_id = router['id']
63 def update_slice(auth, id, **fields):
64 driver = OpenStackDriver(client = auth_check(auth))
65 slices = _get_slices(id)
73 nova_fields['tenant_name'] = fields['name']
74 if 'description' in fields:
75 nova_fields['description'] = fields['description']
76 if 'enabled' in fields:
77 nova_fields['enabled'] = fields['enabled']
78 driver.update_tenant(slice.tenant_id, **nova_fields)
81 sites = _get_sites(fields.get('site'))
82 if sites: fields['site'] = sites[0]
83 slice.update(**fields)
87 def delete_slice(auth, filter={}):
88 driver = OpenStackDriver(client = auth_check(auth))
89 slices = _get_slices(id)
91 driver.delete_slice(id=slice.tenant_id)
95 def get_slices(auth, filter={}):
96 client = auth_check(auth)
98 sites = _get_sites(filter.get('site'))
99 if sites: filter['site'] = sites[0]
100 slices = _get_slices(filter)