Tony Mack | 29c287f | 2013-04-11 21:07:16 -0400 | [diff] [blame^] | 1 | from types import StringTypes |
Tony Mack | 0b2b804 | 2013-04-07 19:55:27 -0400 | [diff] [blame] | 2 | from plstackapi.openstack.client import OpenStackClient |
| 3 | from plstackapi.openstack.driver import OpenStackDriver |
Tony Mack | f1c1224 | 2013-04-09 16:08:43 -0400 | [diff] [blame] | 4 | from plstackapi.core.api.auth import auth_check |
| 5 | from plstackapi.core.models import Role |
Tony Mack | 6242394 | 2013-04-08 15:14:52 -0400 | [diff] [blame] | 6 | |
Tony Mack | 0b2b804 | 2013-04-07 19:55:27 -0400 | [diff] [blame] | 7 | |
Tony Mack | 29c287f | 2013-04-11 21:07:16 -0400 | [diff] [blame^] | 8 | def _get_roles(filter): |
| 9 | if isinstance(filter, int): |
| 10 | roles = Role.objects.filter(id=filter) |
| 11 | elif isinstance(filter, StringTypes): |
| 12 | roles = Role.objects.filter(role_type=filter) |
| 13 | elif isinstance(filter, dict): |
| 14 | roles = Role.objects.filter(**filter) |
| 15 | else: |
| 16 | roles = [] |
| 17 | return roles |
| 18 | |
Tony Mack | 0b2b804 | 2013-04-07 19:55:27 -0400 | [diff] [blame] | 19 | def add_role(auth, name): |
Tony Mack | b5f339f | 2013-04-08 16:29:38 -0400 | [diff] [blame] | 20 | driver = OpenStackDriver(client = auth_check(auth)) |
| 21 | keystone_role = driver.create_role(name=name) |
Tony Mack | 0b2b804 | 2013-04-07 19:55:27 -0400 | [diff] [blame] | 22 | role = Role(role_type=name, role_id=keystone_role.id) |
| 23 | role.save() |
| 24 | return role |
| 25 | |
Tony Mack | dcca474 | 2013-04-08 21:54:57 -0400 | [diff] [blame] | 26 | def delete_role(auth, filter={}): |
Tony Mack | b5f339f | 2013-04-08 16:29:38 -0400 | [diff] [blame] | 27 | driver = OpenStackDriver(client = auth_check(auth)) |
Tony Mack | 29c287f | 2013-04-11 21:07:16 -0400 | [diff] [blame^] | 28 | roles = _get_roles(filter) |
Tony Mack | 97850a2 | 2013-04-08 21:57:24 -0400 | [diff] [blame] | 29 | for role in roles: |
| 30 | driver.delete_role({'id': role.role_id}) |
| 31 | role.delete() |
Tony Mack | 0b2b804 | 2013-04-07 19:55:27 -0400 | [diff] [blame] | 32 | return 1 |
| 33 | |
Tony Mack | 5271d6b | 2013-04-08 00:34:24 -0400 | [diff] [blame] | 34 | def get_roles(auth, filter={}): |
Tony Mack | 0b2b804 | 2013-04-07 19:55:27 -0400 | [diff] [blame] | 35 | client = auth_check(auth) |
Tony Mack | 29c287f | 2013-04-11 21:07:16 -0400 | [diff] [blame^] | 36 | return _get_roles(filter) |
Tony Mack | 0b2b804 | 2013-04-07 19:55:27 -0400 | [diff] [blame] | 37 | |