| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178 |
- """
- DataTypes used by this provider
- """
- import inspect
- import ipaddress
- import json
- from cloudbridge.cloud.base.resources import BaseAttachmentInfo
- from cloudbridge.cloud.base.resources import BaseBucket
- from cloudbridge.cloud.base.resources import BaseBucketObject
- from cloudbridge.cloud.base.resources import BaseFloatingIP
- from cloudbridge.cloud.base.resources import BaseInstance
- from cloudbridge.cloud.base.resources import BaseInstanceType
- from cloudbridge.cloud.base.resources import BaseKeyPair
- from cloudbridge.cloud.base.resources import BaseMachineImage
- from cloudbridge.cloud.base.resources import BaseNetwork
- from cloudbridge.cloud.base.resources import BasePlacementZone
- from cloudbridge.cloud.base.resources import BaseRegion
- from cloudbridge.cloud.base.resources import BaseRouter
- from cloudbridge.cloud.base.resources import BaseSecurityGroup
- from cloudbridge.cloud.base.resources import BaseSecurityGroupRule
- from cloudbridge.cloud.base.resources import BaseSnapshot
- from cloudbridge.cloud.base.resources import BaseSubnet
- from cloudbridge.cloud.base.resources import BaseVolume
- from cloudbridge.cloud.interfaces.resources import InstanceState
- from cloudbridge.cloud.interfaces.resources import MachineImageState
- from cloudbridge.cloud.interfaces.resources import NetworkState
- from cloudbridge.cloud.interfaces.resources import RouterState
- from cloudbridge.cloud.interfaces.resources import SecurityGroup
- from cloudbridge.cloud.interfaces.resources import SnapshotState
- from cloudbridge.cloud.interfaces.resources import VolumeState
- from cloudbridge.cloud.providers.openstack import helpers as oshelpers
- from keystoneclient.v3.regions import Region
- import novaclient.exceptions as novaex
- import swiftclient.exceptions as swiftex
- class OpenStackMachineImage(BaseMachineImage):
- # ref: http://docs.openstack.org/developer/glance/statuses.html
- IMAGE_STATE_MAP = {
- 'QUEUED': MachineImageState.PENDING,
- 'SAVING': MachineImageState.PENDING,
- 'ACTIVE': MachineImageState.AVAILABLE,
- 'KILLED': MachineImageState.ERROR,
- 'DELETED': MachineImageState.ERROR,
- 'PENDING_DELETE': MachineImageState.ERROR
- }
- def __init__(self, provider, os_image):
- super(OpenStackMachineImage, self).__init__(provider)
- if isinstance(os_image, OpenStackMachineImage):
- # pylint:disable=protected-access
- self._os_image = os_image._os_image
- else:
- self._os_image = os_image
- @property
- def id(self):
- """
- Get the image identifier.
- """
- return self._os_image.id
- @property
- def name(self):
- """
- Get the image name.
- """
- return self._os_image.name
- @property
- def description(self):
- """
- Get the image description.
- """
- return None
- @property
- def min_disk(self):
- """
- Returns the minimum size of the disk that's required to
- boot this image (in GB)
- :rtype: ``int``
- :return: The minimum disk size needed by this image
- """
- return self._os_image.minDisk
- def delete(self):
- """
- Delete this image
- """
- self._os_image.delete()
- @property
- def state(self):
- return OpenStackMachineImage.IMAGE_STATE_MAP.get(
- self._os_image.status, MachineImageState.UNKNOWN)
- def refresh(self):
- """
- Refreshes the state of this instance by re-querying the cloud provider
- for its latest state.
- """
- image = self._provider.compute.images.get(self.id)
- if image:
- self._os_image = image._os_image # pylint:disable=protected-access
- else:
- # The image no longer exists and cannot be refreshed.
- # set the status to unknown
- self._os_image.status = 'unknown'
- class OpenStackPlacementZone(BasePlacementZone):
- def __init__(self, provider, zone, region):
- super(OpenStackPlacementZone, self).__init__(provider)
- if isinstance(zone, OpenStackPlacementZone):
- self._os_zone = zone._os_zone # pylint:disable=protected-access
- self._os_region = zone._os_region
- else:
- self._os_zone = zone
- self._os_region = region
- @property
- def id(self):
- """
- Get the zone id
- :rtype: ``str``
- :return: ID for this zone as returned by the cloud middleware.
- """
- return self._os_zone
- @property
- def name(self):
- """
- Get the zone name.
- :rtype: ``str``
- :return: Name for this zone as returned by the cloud middleware.
- """
- # return self._os_zone.zoneName
- return self._os_zone
- @property
- def region_name(self):
- """
- Get the region that this zone belongs to.
- :rtype: ``str``
- :return: Name of this zone's region as returned by the cloud middleware
- """
- return self._os_region
- class OpenStackInstanceType(BaseInstanceType):
- def __init__(self, provider, os_flavor):
- super(OpenStackInstanceType, self).__init__(provider)
- self._os_flavor = os_flavor
- @property
- def id(self):
- return self._os_flavor.id
- @property
- def name(self):
- return self._os_flavor.name
- @property
- def family(self):
- # TODO: This may not be standardised across OpenStack
- # but NeCTAR is using it this way
- return self.extra_data.get('flavor_class:name')
- @property
- def vcpus(self):
- return self._os_flavor.vcpus
- @property
- def ram(self):
- return self._os_flavor.ram
- @property
- def size_root_disk(self):
- return self._os_flavor.disk
- @property
- def size_ephemeral_disks(self):
- return 0 if self._os_flavor.ephemeral == 'N/A' else \
- self._os_flavor.ephemeral
- @property
- def num_ephemeral_disks(self):
- return 0 if self._os_flavor.ephemeral == 'N/A' else \
- self._os_flavor.ephemeral
- @property
- def extra_data(self):
- extras = self._os_flavor.get_keys()
- extras['rxtx_factor'] = self._os_flavor.rxtx_factor
- extras['swap'] = self._os_flavor.swap
- extras['is_public'] = self._os_flavor.is_public
- return extras
- class OpenStackInstance(BaseInstance):
- # ref: http://docs.openstack.org/developer/nova/v2/2.0_server_concepts.html
- # and http://developer.openstack.org/api-ref-compute-v2.html
- INSTANCE_STATE_MAP = {
- 'ACTIVE': InstanceState.RUNNING,
- 'BUILD': InstanceState.PENDING,
- 'DELETED': InstanceState.TERMINATED,
- 'ERROR': InstanceState.ERROR,
- 'HARD_REBOOT': InstanceState.REBOOTING,
- 'PASSWORD': InstanceState.PENDING,
- 'PAUSED': InstanceState.STOPPED,
- 'REBOOT': InstanceState.REBOOTING,
- 'REBUILD': InstanceState.CONFIGURING,
- 'RESCUE': InstanceState.CONFIGURING,
- 'RESIZE': InstanceState.CONFIGURING,
- 'REVERT_RESIZE': InstanceState.CONFIGURING,
- 'SOFT_DELETED': InstanceState.STOPPED,
- 'STOPPED': InstanceState.STOPPED,
- 'SUSPENDED': InstanceState.STOPPED,
- 'SHUTOFF': InstanceState.STOPPED,
- 'UNKNOWN': InstanceState.UNKNOWN,
- 'VERIFY_RESIZE': InstanceState.CONFIGURING
- }
- def __init__(self, provider, os_instance):
- super(OpenStackInstance, self).__init__(provider)
- self._os_instance = os_instance
- @property
- def id(self):
- """
- Get the instance identifier.
- """
- return self._os_instance.id
- @property
- def name(self):
- """
- Get the instance name.
- """
- return self._os_instance.name
- @name.setter
- # pylint:disable=arguments-differ
- def name(self, value):
- """
- Set the instance name.
- """
- self._os_instance.name = value
- self._os_instance.update()
- @property
- def public_ips(self):
- """
- Get all the public IP addresses for this instance.
- """
- # OpenStack doesn't provide an easy way to figure our whether an IP is
- # public or private, since the returned IPs are grouped by an arbitrary
- # network label. Therefore, it's necessary to parse the address and
- # determine whether it's public or private
- return [address
- for _, addresses in self._os_instance.networks.items()
- for address in addresses
- if not ipaddress.ip_address(address).is_private]
- @property
- def private_ips(self):
- """
- Get all the private IP addresses for this instance.
- """
- return [address
- for _, addresses in self._os_instance.networks.items()
- for address in addresses
- if ipaddress.ip_address(address).is_private]
- @property
- def instance_type_id(self):
- """
- Get the instance type name.
- """
- return self._os_instance.flavor.get('id')
- @property
- def instance_type(self):
- """
- Get the instance type object.
- """
- flavor = self._provider.nova.flavors.get(
- self._os_instance.flavor.get('id'))
- return OpenStackInstanceType(self._provider, flavor)
- def reboot(self):
- """
- Reboot this instance (using the cloud middleware API).
- """
- self._os_instance.reboot()
- def terminate(self):
- """
- Permanently terminate this instance.
- """
- self._os_instance.delete()
- @property
- def image_id(self):
- """
- Get the image ID for this instance.
- """
- # In OpenStack, the Machine Image of a running instance may
- # be deleted, so make sure the image exists before attempting to
- # retrieve its id
- return (self._os_instance.image.get("id")
- if self._os_instance.image else "")
- @property
- def zone_id(self):
- """
- Get the placement zone where this instance is running.
- """
- return getattr(self._os_instance, 'OS-EXT-AZ:availability_zone', None)
- @property
- def security_groups(self):
- """
- Get the security groups associated with this instance.
- """
- return [self._provider.security.security_groups.find(group['name'])[0]
- for group in self._os_instance.security_groups]
- @property
- def security_group_ids(self):
- """
- Get the security groups IDs associated with this instance.
- """
- return [group.id for group in self.security_groups]
- @property
- def key_pair_name(self):
- """
- Get the name of the key pair associated with this instance.
- """
- return self._os_instance.key_name
- def create_image(self, name):
- """
- Create a new image based on this instance.
- """
- image_id = self._os_instance.create_image(name)
- return OpenStackMachineImage(
- self._provider, self._provider.compute.images.get(image_id))
- def add_floating_ip(self, ip_address):
- """
- Add a floating IP address to this instance.
- """
- self._os_instance.add_floating_ip(ip_address)
- def remove_floating_ip(self, ip_address):
- """
- Remove a floating IP address from this instance.
- """
- self._os_instance.remove_floating_ip(ip_address)
- def add_security_group(self, sg):
- """
- Add a security group to this instance
- """
- self._os_instance.add_security_group(sg.id)
- def remove_security_group(self, sg):
- """
- Remove a security group from this instance
- """
- self._os_instance.remove_security_group(sg.id)
- @property
- def state(self):
- return OpenStackInstance.INSTANCE_STATE_MAP.get(
- self._os_instance.status, InstanceState.UNKNOWN)
- def refresh(self):
- """
- Refreshes the state of this instance by re-querying the cloud provider
- for its latest state.
- """
- instance = self._provider.compute.instances.get(
- self.id)
- if instance:
- # pylint:disable=protected-access
- self._os_instance = instance._os_instance
- else:
- # The instance no longer exists and cannot be refreshed.
- # set the status to unknown
- self._os_instance.status = 'unknown'
- class OpenStackRegion(BaseRegion):
- def __init__(self, provider, os_region):
- super(OpenStackRegion, self).__init__(provider)
- self._os_region = os_region
- @property
- def id(self):
- return (self._os_region.id if type(self._os_region) == Region else
- self._os_region)
- @property
- def name(self):
- return (self._os_region.id if type(self._os_region) == Region else
- self._os_region)
- @property
- def zones(self):
- # ``detailed`` param must be set to ``False`` because the (default)
- # ``True`` value requires Admin privileges
- if self.name == self._provider.region_name: # optimisation
- zones = self._provider.nova.availability_zones.list(detailed=False)
- else:
- try:
- region_nova = self._provider._connect_nova_region(self.name)
- zones = region_nova.availability_zones.list(detailed=False)
- except novaex.EndpointNotFound:
- # This region may not have a compute endpoint. If so just
- # return an empty list
- zones = []
- return [OpenStackPlacementZone(self._provider, z.zoneName, self.name)
- for z in zones]
- class OpenStackVolume(BaseVolume):
- # Ref: http://developer.openstack.org/api-ref-blockstorage-v2.html
- VOLUME_STATE_MAP = {
- 'creating': VolumeState.CREATING,
- 'available': VolumeState.AVAILABLE,
- 'attaching': VolumeState.CONFIGURING,
- 'in-use': VolumeState.IN_USE,
- 'deleting': VolumeState.CONFIGURING,
- 'error': VolumeState.ERROR,
- 'error_deleting': VolumeState.ERROR,
- 'backing-up': VolumeState.CONFIGURING,
- 'restoring-backup': VolumeState.CONFIGURING,
- 'error_restoring': VolumeState.ERROR,
- 'error_extending': VolumeState.ERROR
- }
- def __init__(self, provider, volume):
- super(OpenStackVolume, self).__init__(provider)
- self._volume = volume
- @property
- def id(self):
- return self._volume.id
- @property
- def name(self):
- """
- Get the volume name.
- """
- return self._volume.name
- @name.setter
- def name(self, value): # pylint:disable=arguments-differ
- """
- Set the volume name.
- """
- self._volume.name = value
- self._volume.update(name=value)
- @property
- def description(self):
- return self._volume.description
- @description.setter
- def description(self, value):
- self._volume.description = value
- self._volume.update(description=value)
- @property
- def size(self):
- return self._volume.size
- @property
- def create_time(self):
- return self._volume.created_at
- @property
- def zone_id(self):
- return self._volume.availability_zone
- @property
- def source(self):
- if self._volume.snapshot_id:
- return self._provider.block_store.snapshots.get(
- self._volume.snapshot_id)
- return None
- @property
- def attachments(self):
- if self._volume.attachments:
- return BaseAttachmentInfo(
- self,
- self._volume.attachments[0].get('server_id'),
- self._volume.attachments[0].get('device'))
- else:
- return None
- def attach(self, instance, device):
- """
- Attach this volume to an instance.
- """
- instance_id = instance.id if isinstance(
- instance,
- OpenStackInstance) else instance
- self._volume.attach(instance_id, device)
- def detach(self, force=False):
- """
- Detach this volume from an instance.
- """
- self._volume.detach()
- def create_snapshot(self, name, description=None):
- """
- Create a snapshot of this Volume.
- """
- return self._provider.block_store.snapshots.create(
- name, self, description=description)
- def delete(self):
- """
- Delete this volume.
- """
- self._volume.delete()
- @property
- def state(self):
- return OpenStackVolume.VOLUME_STATE_MAP.get(
- self._volume.status, VolumeState.UNKNOWN)
- def refresh(self):
- """
- Refreshes the state of this volume by re-querying the cloud provider
- for its latest state.
- """
- vol = self._provider.block_store.volumes.get(
- self.id)
- if vol:
- self._volume = vol._volume # pylint:disable=protected-access
- else:
- # The volume no longer exists and cannot be refreshed.
- # set the status to unknown
- self._volume.status = 'unknown'
- class OpenStackSnapshot(BaseSnapshot):
- # Ref: http://developer.openstack.org/api-ref-blockstorage-v2.html
- SNAPSHOT_STATE_MAP = {
- 'creating': SnapshotState.PENDING,
- 'available': SnapshotState.AVAILABLE,
- 'deleting': SnapshotState.CONFIGURING,
- 'error': SnapshotState.ERROR,
- 'error_deleting': SnapshotState.ERROR
- }
- def __init__(self, provider, snapshot):
- super(OpenStackSnapshot, self).__init__(provider)
- self._snapshot = snapshot
- @property
- def id(self):
- return self._snapshot.id
- @property
- def name(self):
- """
- Get the snapshot name.
- """
- return self._snapshot.name
- @name.setter
- def name(self, value): # pylint:disable=arguments-differ
- """
- Set the snapshot name.
- """
- self._snapshot.name = value
- self._snapshot.update(name=value)
- @property
- def description(self):
- return self._snapshot.description
- @description.setter
- def description(self, value):
- self._snapshot.description = value
- self._snapshot.update(description=value)
- @property
- def size(self):
- return self._snapshot.size
- @property
- def volume_id(self):
- return self._snapshot.volume_id
- @property
- def create_time(self):
- return self._snapshot.created_at
- @property
- def state(self):
- return OpenStackSnapshot.SNAPSHOT_STATE_MAP.get(
- self._snapshot.status, SnapshotState.UNKNOWN)
- def refresh(self):
- """
- Refreshes the state of this snapshot by re-querying the cloud provider
- for its latest state.
- """
- snap = self._provider.block_store.snapshots.get(
- self.id)
- if snap:
- self._snapshot = snap._snapshot # pylint:disable=protected-access
- else:
- # The snapshot no longer exists and cannot be refreshed.
- # set the status to unknown
- self._snapshot.status = 'unknown'
- def delete(self):
- """
- Delete this snapshot.
- """
- self._snapshot.delete()
- def create_volume(self, placement, size=None, volume_type=None, iops=None):
- """
- Create a new Volume from this Snapshot.
- """
- vol_name = "Created from {0} ({1})".format(self.id, self.name)
- size = size if size else self._snapshot.size
- os_vol = self._provider.cinder.volumes.create(
- size, name=vol_name, availability_zone=placement,
- snapshot_id=self._snapshot.id)
- cb_vol = OpenStackVolume(self._provider, os_vol)
- cb_vol.name = vol_name
- return cb_vol
- class OpenStackNetwork(BaseNetwork):
- # Ref: https://github.com/openstack/neutron/blob/master/neutron/plugins/
- # common/constants.py
- _NETWORK_STATE_MAP = {
- 'PENDING_CREATE': NetworkState.PENDING,
- 'PENDING_UPDATE': NetworkState.PENDING,
- 'PENDING_DELETE': NetworkState.PENDING,
- 'CREATED': NetworkState.PENDING,
- 'INACTIVE': NetworkState.PENDING,
- 'DOWN': NetworkState.DOWN,
- 'ERROR': NetworkState.ERROR,
- 'ACTIVE': NetworkState.AVAILABLE
- }
- def __init__(self, provider, network):
- super(OpenStackNetwork, self).__init__(provider)
- self._network = network
- @property
- def id(self):
- return self._network.get('id', None)
- @property
- def name(self):
- return self._network.get('name', None)
- @property
- def external(self):
- return self._network.get('router:external', False)
- @property
- def state(self):
- self.refresh()
- return OpenStackNetwork._NETWORK_STATE_MAP.get(
- self._network.get('status', None),
- NetworkState.UNKNOWN)
- @property
- def cidr_block(self):
- # OpenStack does not define a CIDR block for networks
- return ''
- def delete(self):
- if self.id in str(self._provider.neutron.list_networks()):
- self._provider.neutron.delete_network(self.id)
- # Adhere to the interface docs
- if self.id not in str(self._provider.neutron.list_networks()):
- return True
- def subnets(self):
- subnets = (self._provider.neutron.list_subnets(network_id=self.id)
- .get('subnets', []))
- return [OpenStackSubnet(self._provider, subnet) for subnet in subnets]
- def create_subnet(self, cidr_block, name='', zone=None):
- """OpenStack has no support for subnet zones so the value is ignored"""
- subnet_info = {'name': name, 'network_id': self.id,
- 'cidr': cidr_block, 'ip_version': 4}
- subnet = (self._provider.neutron.create_subnet({'subnet': subnet_info})
- .get('subnet'))
- return OpenStackSubnet(self._provider, subnet)
- def refresh(self):
- """Refresh the state of this network by re-querying the provider."""
- net = self._provider.neutron.list_networks(id=self.id).get('networks')
- self._network = net[0] if net else {}
- class OpenStackSubnet(BaseSubnet):
- def __init__(self, provider, subnet):
- super(OpenStackSubnet, self).__init__(provider)
- self._subnet = subnet
- @property
- def id(self):
- return self._subnet.get('id', None)
- @property
- def name(self):
- return self._subnet.get('name', None)
- @property
- def cidr_block(self):
- return self._subnet.get('cidr', None)
- @property
- def network_id(self):
- return self._subnet.get('network_id', None)
- @property
- def zone(self):
- """
- OpenStack does not have a notion of placement zone for subnets.
- Default to None.
- """
- return None
- def delete(self):
- if self.id in str(self._provider.neutron.list_subnets()):
- self._provider.neutron.delete_subnet(self.id)
- # Adhere to the interface docs
- if self.id not in str(self._provider.neutron.list_subnets()):
- return True
- class OpenStackFloatingIP(BaseFloatingIP):
- def __init__(self, provider, floating_ip):
- super(OpenStackFloatingIP, self).__init__(provider)
- self._ip = floating_ip
- @property
- def id(self):
- return self._ip.get('id', None)
- @property
- def public_ip(self):
- return self._ip.get('floating_ip_address', None)
- @property
- def private_ip(self):
- return self._ip.get('fixed_ip_address', None)
- def in_use(self):
- return True if self._ip.get('status', None) == 'ACTIVE' else False
- def delete(self):
- self._provider.neutron.delete_floatingip(self.id)
- # Adhere to the interface docs
- if self.id not in str(self._provider.neutron.list_floatingips()):
- return True
- class OpenStackRouter(BaseRouter):
- def __init__(self, provider, router):
- super(OpenStackRouter, self).__init__(provider)
- self._router = router
- @property
- def id(self):
- return self._router.get('id', None)
- @property
- def name(self):
- return self._router.get('name', None)
- def refresh(self):
- self._router = self._provider.neutron.show_router(self.id)['router']
- @property
- def state(self):
- if self._router.get('external_gateway_info'):
- return RouterState.ATTACHED
- return RouterState.DETACHED
- @property
- def network_id(self):
- if self.state == RouterState.ATTACHED:
- return self._router.get('external_gateway_info', {}).get(
- 'network_id', None)
- return None
- def delete(self):
- self._provider.neutron.delete_router(self.id)
- # Adhere to the interface docs
- if self.id not in str(self._provider.neutron.list_routers()):
- return True
- def attach_network(self, network_id):
- self._router = self._provider.neutron.add_gateway_router(
- self.id, {'network_id': network_id}).get('router', self._router)
- if self.network_id and self.network_id == network_id:
- return True
- return False
- def detach_network(self):
- self._router = self._provider.neutron.remove_gateway_router(
- self.id).get('router', self._router)
- if not self.network_id:
- return True
- return False
- def add_route(self, subnet_id):
- router_interface = {'subnet_id': subnet_id}
- ret = self._provider.neutron.add_interface_router(
- self.id, router_interface)
- if subnet_id in ret.get('subnet_ids', ""):
- return True
- return False
- def remove_route(self, subnet_id):
- router_interface = {'subnet_id': subnet_id}
- ret = self._provider.neutron.remove_interface_router(
- self.id, router_interface)
- if subnet_id in ret.get('subnet_ids', ""):
- return True
- return False
- class OpenStackKeyPair(BaseKeyPair):
- def __init__(self, provider, key_pair):
- super(OpenStackKeyPair, self).__init__(provider, key_pair)
- @property
- def material(self):
- """
- Unencrypted private key.
- :rtype: str
- :return: Unencrypted private key or ``None`` if not available.
- """
- return getattr(self._key_pair, 'private_key', None)
- class OpenStackSecurityGroup(BaseSecurityGroup):
- def __init__(self, provider, security_group):
- super(OpenStackSecurityGroup, self).__init__(provider, security_group)
- @property
- def network_id(self):
- """
- OpenStack does not associate a SG with a network so default to None.
- :return: Always return ``None``.
- """
- return None
- @property
- def rules(self):
- # Update SG object; otherwise, recently added rules do now show
- self._security_group = self._provider.nova.security_groups.get(
- self._security_group)
- return [OpenStackSecurityGroupRule(self._provider, r, self)
- for r in self._security_group.rules]
- def add_rule(self, ip_protocol=None, from_port=None, to_port=None,
- cidr_ip=None, src_group=None):
- """
- Create a security group rule.
- You need to pass in either ``src_group`` OR ``ip_protocol`` AND
- ``from_port``, ``to_port``, ``cidr_ip``. In other words, either
- you are authorizing another group or you are authorizing some
- ip-based rule.
- :type ip_protocol: str
- :param ip_protocol: Either ``tcp`` | ``udp`` | ``icmp``
- :type from_port: int
- :param from_port: The beginning port number you are enabling
- :type to_port: int
- :param to_port: The ending port number you are enabling
- :type cidr_ip: str or list of strings
- :param cidr_ip: The CIDR block you are providing access to.
- :type src_group: ``object`` of :class:`.SecurityGroup`
- :param src_group: The Security Group you are granting access to.
- :rtype: :class:``.SecurityGroupRule``
- :return: Rule object if successful or ``None``.
- """
- if src_group:
- if not isinstance(src_group, SecurityGroup):
- src_group = self._provider.security.security_groups.get(
- src_group)
- existing_rule = self.get_rule(ip_protocol=ip_protocol,
- from_port=from_port,
- to_port=to_port,
- src_group=src_group)
- if existing_rule:
- return existing_rule
- rule = self._provider.nova.security_group_rules.create(
- parent_group_id=self._security_group.id,
- ip_protocol=ip_protocol,
- from_port=from_port,
- to_port=to_port,
- group_id=src_group.id)
- if rule:
- # We can only return one Rule so default to TCP (ie, last in
- # the for loop above).
- return OpenStackSecurityGroupRule(self._provider,
- rule.to_dict(), self)
- else:
- existing_rule = self.get_rule(ip_protocol=ip_protocol,
- from_port=from_port,
- to_port=to_port,
- cidr_ip=cidr_ip)
- if existing_rule:
- return existing_rule
- rule = self._provider.nova.security_group_rules.create(
- parent_group_id=self._security_group.id,
- ip_protocol=ip_protocol,
- from_port=from_port,
- to_port=to_port,
- cidr=cidr_ip)
- if rule:
- return OpenStackSecurityGroupRule(self._provider,
- rule.to_dict(), self)
- return None
- def get_rule(self, ip_protocol=None, from_port=None, to_port=None,
- cidr_ip=None, src_group=None):
- # Update SG object; otherwise, recently added rules do not show
- self._security_group = self._provider.nova.security_groups.get(
- self._security_group)
- for rule in self._security_group.rules:
- if (rule['ip_protocol'] == ip_protocol and
- rule['from_port'] == from_port and
- rule['to_port'] == to_port and
- (rule['ip_range'].get('cidr') == cidr_ip or
- (rule['group'].get('name') == src_group.name if src_group
- else False))):
- return OpenStackSecurityGroupRule(self._provider, rule, self)
- return None
- def to_json(self):
- attr = inspect.getmembers(self, lambda a: not(inspect.isroutine(a)))
- js = {k: v for(k, v) in attr if not k.startswith('_')}
- json_rules = [r.to_json() for r in self.rules]
- js['rules'] = [json.loads(r) for r in json_rules]
- return json.dumps(js, sort_keys=True)
- class OpenStackSecurityGroupRule(BaseSecurityGroupRule):
- def __init__(self, provider, rule, parent):
- super(OpenStackSecurityGroupRule, self).__init__(
- provider, rule, parent)
- @property
- def id(self):
- return self._rule.get('id')
- @property
- def ip_protocol(self):
- return self._rule.get('ip_protocol')
- @property
- def from_port(self):
- return int(self._rule.get('from_port') or 0)
- @property
- def to_port(self):
- return int(self._rule.get('to_port') or 0)
- @property
- def cidr_ip(self):
- return self._rule.get('ip_range', {}).get('cidr')
- @property
- def group(self):
- cg = self._rule.get('group', {}).get('name')
- if cg:
- security_groups = self._provider.nova.security_groups.list()
- for sg in security_groups:
- if sg.name == cg:
- return OpenStackSecurityGroup(self._provider, sg)
- return None
- def to_json(self):
- attr = inspect.getmembers(self, lambda a: not(inspect.isroutine(a)))
- js = {k: v for(k, v) in attr if not k.startswith('_')}
- js['group'] = self.group.id if self.group else ''
- js['parent'] = self.parent.id if self.parent else ''
- return json.dumps(js, sort_keys=True)
- def delete(self):
- return self._provider.nova.security_group_rules.delete(self.id)
- class OpenStackBucketObject(BaseBucketObject):
- def __init__(self, provider, cbcontainer, obj):
- super(OpenStackBucketObject, self).__init__(provider)
- self.cbcontainer = cbcontainer
- self._obj = obj
- @property
- def id(self):
- return self._obj.get("name")
- @property
- def name(self):
- """Get this object's name."""
- return self._obj.get("name")
- @property
- def size(self):
- return self._obj.get("bytes")
- @property
- def last_modified(self):
- return self._obj.get("last_modified")
- def iter_content(self):
- """Returns this object's content as an iterable."""
- _, content = self._provider.swift.get_object(
- self.cbcontainer.name, self.name, resp_chunk_size=65536)
- return content
- def upload(self, data):
- """
- Set the contents of this object to the data read from the source
- string.
- """
- self._provider.swift.put_object(self.cbcontainer.name, self.name,
- data)
- def upload_from_file(self, path):
- """
- Stores the contents of the file pointed by the "path" variable.
- """
- with open(path, 'r') as f:
- self.upload(f.read())
- def delete(self):
- """
- Delete this object.
- :rtype: ``bool``
- :return: True if successful
- """
- try:
- self._provider.swift.delete_object(self.cbcontainer.name,
- self.name)
- except swiftex.ClientException as err:
- if err.http_status == 404:
- return True
- return False
- def generate_url(self, expires_in=0):
- """
- Generates a URL to this object.
- If the object is public, `expires_in` argument is not necessary, but if
- the object is private, the life time of URL is set using `expires_in`
- argument.
- See here for implementation details:
- http://stackoverflow.com/a/37057172
- """
- raise NotImplementedError("This functionality is not implemented yet.")
- class OpenStackBucket(BaseBucket):
- def __init__(self, provider, bucket):
- super(OpenStackBucket, self).__init__(provider)
- self._bucket = bucket
- @property
- def id(self):
- return self._bucket.get("name")
- @property
- def name(self):
- """
- Get this bucket's name.
- """
- return self._bucket.get("name")
- def get(self, name):
- """
- Retrieve a given object from this bucket.
- FIXME: If multiple objects match the name as their name prefix,
- all will be returned by the provider but this method will only
- return the first element.
- """
- _, object_list = self._provider.swift.get_container(
- self.name, prefix=name)
- if object_list:
- return OpenStackBucketObject(self._provider, self,
- object_list[0])
- else:
- return None
- def list(self, limit=None, marker=None, prefix=None):
- """
- List all objects within this bucket.
- :rtype: BucketObject
- :return: List of all available BucketObjects within this bucket.
- """
- _, object_list = self._provider.swift.get_container(
- self.name, limit=oshelpers.os_result_limit(self._provider, limit),
- marker=marker, prefix=prefix)
- cb_objects = [OpenStackBucketObject(
- self._provider, self, obj) for obj in object_list]
- return oshelpers.to_server_paged_list(
- self._provider,
- cb_objects,
- limit)
- def delete(self, delete_contents=False):
- """
- Delete this bucket.
- """
- self._provider.swift.delete_container(self.name)
- def create_object(self, object_name):
- self._provider.swift.put_object(self.name, object_name, None)
- return self.get(object_name)
|