resources.py 44 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423
  1. """
  2. DataTypes used by this provider
  3. """
  4. import inspect
  5. import ipaddress
  6. import logging
  7. import os
  8. import cloudbridge.cloud.base.helpers as cb_helpers
  9. from cloudbridge.cloud.base.resources import BaseAttachmentInfo
  10. from cloudbridge.cloud.base.resources import BaseBucket
  11. from cloudbridge.cloud.base.resources import BaseBucketContainer
  12. from cloudbridge.cloud.base.resources import BaseBucketObject
  13. from cloudbridge.cloud.base.resources import BaseFloatingIP
  14. from cloudbridge.cloud.base.resources import BaseFloatingIPContainer
  15. from cloudbridge.cloud.base.resources import BaseGatewayContainer
  16. from cloudbridge.cloud.base.resources import BaseInstance
  17. from cloudbridge.cloud.base.resources import BaseInternetGateway
  18. from cloudbridge.cloud.base.resources import BaseKeyPair
  19. from cloudbridge.cloud.base.resources import BaseMachineImage
  20. from cloudbridge.cloud.base.resources import BaseNetwork
  21. from cloudbridge.cloud.base.resources import BasePlacementZone
  22. from cloudbridge.cloud.base.resources import BaseRegion
  23. from cloudbridge.cloud.base.resources import BaseRouter
  24. from cloudbridge.cloud.base.resources import BaseSnapshot
  25. from cloudbridge.cloud.base.resources import BaseSubnet
  26. from cloudbridge.cloud.base.resources import BaseVMFirewall
  27. from cloudbridge.cloud.base.resources import BaseVMFirewallRule
  28. from cloudbridge.cloud.base.resources import BaseVMFirewallRuleContainer
  29. from cloudbridge.cloud.base.resources import BaseVMType
  30. from cloudbridge.cloud.base.resources import BaseVolume
  31. from cloudbridge.cloud.base.resources import ClientPagedResultList
  32. from cloudbridge.cloud.interfaces.exceptions import InvalidValueException
  33. from cloudbridge.cloud.interfaces.resources import GatewayState
  34. from cloudbridge.cloud.interfaces.resources import InstanceState
  35. from cloudbridge.cloud.interfaces.resources import MachineImageState
  36. from cloudbridge.cloud.interfaces.resources import NetworkState
  37. from cloudbridge.cloud.interfaces.resources import RouterState
  38. from cloudbridge.cloud.interfaces.resources import SnapshotState
  39. from cloudbridge.cloud.interfaces.resources import SubnetState
  40. from cloudbridge.cloud.interfaces.resources import TrafficDirection
  41. from cloudbridge.cloud.interfaces.resources import VolumeState
  42. from cloudbridge.cloud.providers.openstack import helpers as oshelpers
  43. from keystoneclient.v3.regions import Region
  44. from neutronclient.common.exceptions import PortNotFoundClient
  45. import novaclient.exceptions as novaex
  46. from openstack.exceptions import HttpException
  47. from openstack.exceptions import ResourceNotFound
  48. import swiftclient
  49. from swiftclient.service import SwiftService, SwiftUploadObject
  50. ONE_GIG = 1048576000 # in bytes
  51. FIVE_GIG = ONE_GIG * 5 # in bytes
  52. log = logging.getLogger(__name__)
  53. class OpenStackMachineImage(BaseMachineImage):
  54. # ref: http://docs.openstack.org/developer/glance/statuses.html
  55. IMAGE_STATE_MAP = {
  56. 'queued': MachineImageState.PENDING,
  57. 'saving': MachineImageState.PENDING,
  58. 'active': MachineImageState.AVAILABLE,
  59. 'killed': MachineImageState.ERROR,
  60. 'deleted': MachineImageState.ERROR,
  61. 'pending_delete': MachineImageState.ERROR,
  62. 'deactivated': MachineImageState.ERROR
  63. }
  64. def __init__(self, provider, os_image):
  65. super(OpenStackMachineImage, self).__init__(provider)
  66. if isinstance(os_image, OpenStackMachineImage):
  67. # pylint:disable=protected-access
  68. self._os_image = os_image._os_image
  69. else:
  70. self._os_image = os_image
  71. @property
  72. def id(self):
  73. """
  74. Get the image identifier.
  75. """
  76. return self._os_image.id
  77. @property
  78. def name(self):
  79. """
  80. Get the image name.
  81. """
  82. return self._os_image.name
  83. @property
  84. def description(self):
  85. """
  86. Get the image description.
  87. """
  88. return None
  89. @property
  90. def min_disk(self):
  91. """
  92. Returns the minimum size of the disk that's required to
  93. boot this image (in GB)
  94. :rtype: ``int``
  95. :return: The minimum disk size needed by this image
  96. """
  97. return self._os_image.min_disk
  98. def delete(self):
  99. """
  100. Delete this image
  101. """
  102. self._os_image.delete(self._provider.os_conn.session)
  103. @property
  104. def state(self):
  105. return OpenStackMachineImage.IMAGE_STATE_MAP.get(
  106. self._os_image.status, MachineImageState.UNKNOWN)
  107. def refresh(self):
  108. """
  109. Refreshes the state of this instance by re-querying the cloud provider
  110. for its latest state.
  111. """
  112. log.debug("Refreshing OpenStack Machine Image")
  113. image = self._provider.compute.images.get(self.id)
  114. if image:
  115. self._os_image = image._os_image # pylint:disable=protected-access
  116. else:
  117. # The image no longer exists and cannot be refreshed.
  118. # set the status to unknown
  119. self._os_image.status = 'unknown'
  120. class OpenStackPlacementZone(BasePlacementZone):
  121. def __init__(self, provider, zone, region):
  122. super(OpenStackPlacementZone, self).__init__(provider)
  123. if isinstance(zone, OpenStackPlacementZone):
  124. # pylint:disable=protected-access
  125. self._os_zone = zone._os_zone
  126. # pylint:disable=protected-access
  127. self._os_region = zone._os_region
  128. else:
  129. self._os_zone = zone
  130. self._os_region = region
  131. @property
  132. def id(self):
  133. """
  134. Get the zone id
  135. :rtype: ``str``
  136. :return: ID for this zone as returned by the cloud middleware.
  137. """
  138. return self._os_zone
  139. @property
  140. def name(self):
  141. """
  142. Get the zone name.
  143. :rtype: ``str``
  144. :return: Name for this zone as returned by the cloud middleware.
  145. """
  146. # return self._os_zone.zoneName
  147. return self._os_zone
  148. @property
  149. def region_name(self):
  150. """
  151. Get the region that this zone belongs to.
  152. :rtype: ``str``
  153. :return: Name of this zone's region as returned by the cloud middleware
  154. """
  155. return self._os_region
  156. class OpenStackVMType(BaseVMType):
  157. def __init__(self, provider, os_flavor):
  158. super(OpenStackVMType, self).__init__(provider)
  159. self._os_flavor = os_flavor
  160. @property
  161. def id(self):
  162. return self._os_flavor.id
  163. @property
  164. def name(self):
  165. return self._os_flavor.name
  166. @property
  167. def family(self):
  168. # TODO: This may not be standardised across OpenStack
  169. # but NeCTAR is using it this way
  170. return self.extra_data.get('flavor_class:name')
  171. @property
  172. def vcpus(self):
  173. return self._os_flavor.vcpus
  174. @property
  175. def ram(self):
  176. return self._os_flavor.ram
  177. @property
  178. def size_root_disk(self):
  179. return self._os_flavor.disk
  180. @property
  181. def size_ephemeral_disks(self):
  182. return 0 if self._os_flavor.ephemeral == 'N/A' else \
  183. self._os_flavor.ephemeral
  184. @property
  185. def num_ephemeral_disks(self):
  186. return 0 if self._os_flavor.ephemeral == 'N/A' else \
  187. self._os_flavor.ephemeral
  188. @property
  189. def extra_data(self):
  190. extras = self._os_flavor.get_keys()
  191. extras['rxtx_factor'] = self._os_flavor.rxtx_factor
  192. extras['swap'] = self._os_flavor.swap
  193. extras['is_public'] = self._os_flavor.is_public
  194. return extras
  195. class OpenStackInstance(BaseInstance):
  196. # ref: http://docs.openstack.org/developer/nova/v2/2.0_server_concepts.html
  197. # and http://developer.openstack.org/api-ref-compute-v2.html
  198. INSTANCE_STATE_MAP = {
  199. 'ACTIVE': InstanceState.RUNNING,
  200. 'BUILD': InstanceState.PENDING,
  201. 'DELETED': InstanceState.DELETED,
  202. 'ERROR': InstanceState.ERROR,
  203. 'HARD_REBOOT': InstanceState.REBOOTING,
  204. 'PASSWORD': InstanceState.PENDING,
  205. 'PAUSED': InstanceState.STOPPED,
  206. 'REBOOT': InstanceState.REBOOTING,
  207. 'REBUILD': InstanceState.CONFIGURING,
  208. 'RESCUE': InstanceState.CONFIGURING,
  209. 'RESIZE': InstanceState.CONFIGURING,
  210. 'REVERT_RESIZE': InstanceState.CONFIGURING,
  211. 'SOFT_DELETED': InstanceState.STOPPED,
  212. 'STOPPED': InstanceState.STOPPED,
  213. 'SUSPENDED': InstanceState.STOPPED,
  214. 'SHUTOFF': InstanceState.STOPPED,
  215. 'UNKNOWN': InstanceState.UNKNOWN,
  216. 'VERIFY_RESIZE': InstanceState.CONFIGURING
  217. }
  218. def __init__(self, provider, os_instance):
  219. super(OpenStackInstance, self).__init__(provider)
  220. self._os_instance = os_instance
  221. @property
  222. def id(self):
  223. """
  224. Get the instance identifier.
  225. """
  226. return self._os_instance.id
  227. @property
  228. # pylint:disable=arguments-differ
  229. def name(self):
  230. """
  231. Get the instance name.
  232. """
  233. return self._os_instance.name
  234. @name.setter
  235. # pylint:disable=arguments-differ
  236. def name(self, value):
  237. """
  238. Set the instance name.
  239. """
  240. self.assert_valid_resource_name(value)
  241. self._os_instance.name = value
  242. self._os_instance.update(name=value)
  243. @property
  244. def public_ips(self):
  245. """
  246. Get all the public IP addresses for this instance.
  247. """
  248. # OpenStack doesn't provide an easy way to figure our whether an IP is
  249. # public or private, since the returned IPs are grouped by an arbitrary
  250. # network label. Therefore, it's necessary to parse the address and
  251. # determine whether it's public or private
  252. return [address
  253. for _, addresses in self._os_instance.networks.items()
  254. for address in addresses
  255. if not ipaddress.ip_address(address).is_private]
  256. @property
  257. def private_ips(self):
  258. """
  259. Get all the private IP addresses for this instance.
  260. """
  261. return [address
  262. for _, addresses in self._os_instance.networks.items()
  263. for address in addresses
  264. if ipaddress.ip_address(address).is_private]
  265. @property
  266. def vm_type_id(self):
  267. """
  268. Get the VM type name.
  269. """
  270. return self._os_instance.flavor.get('id')
  271. @property
  272. def vm_type(self):
  273. """
  274. Get the VM type object.
  275. """
  276. flavor = self._provider.nova.flavors.get(
  277. self._os_instance.flavor.get('id'))
  278. return OpenStackVMType(self._provider, flavor)
  279. def reboot(self):
  280. """
  281. Reboot this instance (using the cloud middleware API).
  282. """
  283. self._os_instance.reboot()
  284. def delete(self):
  285. """
  286. Permanently delete this instance.
  287. """
  288. # delete the port we created when launching
  289. # Assumption: it's the first interface in the list
  290. iface_list = self._os_instance.interface_list()
  291. if iface_list:
  292. self._provider.neutron.delete_port(iface_list[0].port_id)
  293. self._os_instance.delete()
  294. @property
  295. def image_id(self):
  296. """
  297. Get the image ID for this instance.
  298. """
  299. # In OpenStack, the Machine Image of a running instance may
  300. # be deleted, so make sure the image exists before attempting to
  301. # retrieve its id
  302. return (self._os_instance.image.get("id")
  303. if self._os_instance.image else "")
  304. @property
  305. def zone_id(self):
  306. """
  307. Get the placement zone where this instance is running.
  308. """
  309. return getattr(self._os_instance, 'OS-EXT-AZ:availability_zone', None)
  310. @property
  311. def vm_firewalls(self):
  312. return [
  313. self._provider.security.vm_firewalls.get(group.id)
  314. for group in self._os_instance.list_security_group()
  315. ]
  316. @property
  317. def vm_firewall_ids(self):
  318. """
  319. Get the VM firewall IDs associated with this instance.
  320. """
  321. return [fw.id for fw in self.vm_firewalls]
  322. @property
  323. def key_pair_name(self):
  324. """
  325. Get the name of the key pair associated with this instance.
  326. """
  327. return self._os_instance.key_name
  328. def create_image(self, name):
  329. """
  330. Create a new image based on this instance.
  331. """
  332. log.debug("Creating OpenStack Image with the name %s", name)
  333. self.assert_valid_resource_name(name)
  334. image_id = self._os_instance.create_image(name)
  335. return OpenStackMachineImage(
  336. self._provider, self._provider.compute.images.get(image_id))
  337. def _get_fip(self, floating_ip):
  338. """Get a floating IP object based on the supplied ID."""
  339. return OpenStackFloatingIP(
  340. self._provider,
  341. self._provider.os_conn.network.get_ip(floating_ip))
  342. def add_floating_ip(self, floating_ip):
  343. """
  344. Add a floating IP address to this instance.
  345. """
  346. log.debug("Adding floating IP adress: %s", floating_ip)
  347. fip = (floating_ip if isinstance(floating_ip, OpenStackFloatingIP)
  348. else self._get_fip(floating_ip))
  349. self._os_instance.add_floating_ip(fip.public_ip)
  350. def remove_floating_ip(self, floating_ip):
  351. """
  352. Remove a floating IP address from this instance.
  353. """
  354. log.debug("Removing floating IP adress: %s", floating_ip)
  355. fip = (floating_ip if isinstance(floating_ip, OpenStackFloatingIP)
  356. else self._get_fip(floating_ip))
  357. self._os_instance.remove_floating_ip(fip.public_ip)
  358. def add_vm_firewall(self, firewall):
  359. """
  360. Add a VM firewall to this instance
  361. """
  362. log.debug("Adding firewall: %s", firewall)
  363. self._os_instance.add_security_group(firewall.id)
  364. def remove_vm_firewall(self, firewall):
  365. """
  366. Remove a VM firewall from this instance
  367. """
  368. log.debug("Removing firewall: %s", firewall)
  369. self._os_instance.remove_security_group(firewall.id)
  370. @property
  371. def state(self):
  372. return OpenStackInstance.INSTANCE_STATE_MAP.get(
  373. self._os_instance.status, InstanceState.UNKNOWN)
  374. def refresh(self):
  375. """
  376. Refreshes the state of this instance by re-querying the cloud provider
  377. for its latest state.
  378. """
  379. instance = self._provider.compute.instances.get(
  380. self.id)
  381. if instance:
  382. # pylint:disable=protected-access
  383. self._os_instance = instance._os_instance
  384. else:
  385. # The instance no longer exists and cannot be refreshed.
  386. # set the status to unknown
  387. self._os_instance.status = 'unknown'
  388. class OpenStackRegion(BaseRegion):
  389. def __init__(self, provider, os_region):
  390. super(OpenStackRegion, self).__init__(provider)
  391. self._os_region = os_region
  392. @property
  393. def id(self):
  394. return (self._os_region.id if type(self._os_region) == Region else
  395. self._os_region)
  396. @property
  397. def name(self):
  398. return (self._os_region.id if type(self._os_region) == Region else
  399. self._os_region)
  400. @property
  401. def zones(self):
  402. # ``detailed`` param must be set to ``False`` because the (default)
  403. # ``True`` value requires Admin privileges
  404. if self.name == self._provider.region_name: # optimisation
  405. zones = self._provider.nova.availability_zones.list(detailed=False)
  406. else:
  407. try:
  408. # pylint:disable=protected-access
  409. region_nova = self._provider._connect_nova_region(self.name)
  410. zones = region_nova.availability_zones.list(detailed=False)
  411. except novaex.EndpointNotFound:
  412. # This region may not have a compute endpoint. If so just
  413. # return an empty list
  414. zones = []
  415. return [OpenStackPlacementZone(self._provider, z.zoneName, self.name)
  416. for z in zones]
  417. class OpenStackVolume(BaseVolume):
  418. # Ref: http://developer.openstack.org/api-ref-blockstorage-v2.html
  419. VOLUME_STATE_MAP = {
  420. 'creating': VolumeState.CREATING,
  421. 'available': VolumeState.AVAILABLE,
  422. 'attaching': VolumeState.CONFIGURING,
  423. 'in-use': VolumeState.IN_USE,
  424. 'deleting': VolumeState.CONFIGURING,
  425. 'error': VolumeState.ERROR,
  426. 'error_deleting': VolumeState.ERROR,
  427. 'backing-up': VolumeState.CONFIGURING,
  428. 'restoring-backup': VolumeState.CONFIGURING,
  429. 'error_restoring': VolumeState.ERROR,
  430. 'error_extending': VolumeState.ERROR
  431. }
  432. def __init__(self, provider, volume):
  433. super(OpenStackVolume, self).__init__(provider)
  434. self._volume = volume
  435. @property
  436. def id(self):
  437. return self._volume.id
  438. @property
  439. # pylint:disable=arguments-differ
  440. def name(self):
  441. """
  442. Get the volume name.
  443. """
  444. return self._volume.name
  445. @name.setter
  446. # pylint:disable=arguments-differ
  447. def name(self, value):
  448. """
  449. Set the volume name.
  450. """
  451. self.assert_valid_resource_name(value)
  452. self._volume.name = value
  453. self._volume.update(name=value)
  454. @property
  455. def description(self):
  456. return self._volume.description
  457. @description.setter
  458. def description(self, value):
  459. self._volume.description = value
  460. self._volume.update(description=value)
  461. @property
  462. def size(self):
  463. return self._volume.size
  464. @property
  465. def create_time(self):
  466. return self._volume.created_at
  467. @property
  468. def zone_id(self):
  469. return self._volume.availability_zone
  470. @property
  471. def source(self):
  472. if self._volume.snapshot_id:
  473. return self._provider.storage.snapshots.get(
  474. self._volume.snapshot_id)
  475. return None
  476. @property
  477. def attachments(self):
  478. if self._volume.attachments:
  479. return BaseAttachmentInfo(
  480. self,
  481. self._volume.attachments[0].get('server_id'),
  482. self._volume.attachments[0].get('device'))
  483. else:
  484. return None
  485. def attach(self, instance, device):
  486. """
  487. Attach this volume to an instance.
  488. """
  489. log.debug("Attaching %s to %s instance", device, instance)
  490. instance_id = instance.id if isinstance(
  491. instance,
  492. OpenStackInstance) else instance
  493. self._volume.attach(instance_id, device)
  494. def detach(self, force=False):
  495. """
  496. Detach this volume from an instance.
  497. """
  498. self._volume.detach()
  499. def create_snapshot(self, name, description=None):
  500. """
  501. Create a snapshot of this Volume.
  502. """
  503. log.debug("Creating snapchat of volume: %s with the "
  504. "description: %s", name, description)
  505. return self._provider.storage.snapshots.create(
  506. name, self, description=description)
  507. def delete(self):
  508. """
  509. Delete this volume.
  510. """
  511. self._volume.delete()
  512. @property
  513. def state(self):
  514. return OpenStackVolume.VOLUME_STATE_MAP.get(
  515. self._volume.status, VolumeState.UNKNOWN)
  516. def refresh(self):
  517. """
  518. Refreshes the state of this volume by re-querying the cloud provider
  519. for its latest state.
  520. """
  521. vol = self._provider.storage.volumes.get(
  522. self.id)
  523. if vol:
  524. self._volume = vol._volume # pylint:disable=protected-access
  525. else:
  526. # The volume no longer exists and cannot be refreshed.
  527. # set the status to unknown
  528. self._volume.status = 'unknown'
  529. class OpenStackSnapshot(BaseSnapshot):
  530. # Ref: http://developer.openstack.org/api-ref-blockstorage-v2.html
  531. SNAPSHOT_STATE_MAP = {
  532. 'creating': SnapshotState.PENDING,
  533. 'available': SnapshotState.AVAILABLE,
  534. 'deleting': SnapshotState.CONFIGURING,
  535. 'error': SnapshotState.ERROR,
  536. 'error_deleting': SnapshotState.ERROR
  537. }
  538. def __init__(self, provider, snapshot):
  539. super(OpenStackSnapshot, self).__init__(provider)
  540. self._snapshot = snapshot
  541. @property
  542. def id(self):
  543. return self._snapshot.id
  544. @property
  545. # pylint:disable=arguments-differ
  546. def name(self):
  547. """
  548. Get the snapshot name.
  549. """
  550. return self._snapshot.name
  551. @name.setter
  552. # pylint:disable=arguments-differ
  553. def name(self, value):
  554. """
  555. Set the snapshot name.
  556. """
  557. self.assert_valid_resource_name(value)
  558. self._snapshot.name = value
  559. self._snapshot.update(name=value)
  560. @property
  561. def description(self):
  562. return self._snapshot.description
  563. @description.setter
  564. def description(self, value):
  565. self._snapshot.description = value
  566. self._snapshot.update(description=value)
  567. @property
  568. def size(self):
  569. return self._snapshot.size
  570. @property
  571. def volume_id(self):
  572. return self._snapshot.volume_id
  573. @property
  574. def create_time(self):
  575. return self._snapshot.created_at
  576. @property
  577. def state(self):
  578. return OpenStackSnapshot.SNAPSHOT_STATE_MAP.get(
  579. self._snapshot.status, SnapshotState.UNKNOWN)
  580. def refresh(self):
  581. """
  582. Refreshes the state of this snapshot by re-querying the cloud provider
  583. for its latest state.
  584. """
  585. snap = self._provider.storage.snapshots.get(
  586. self.id)
  587. if snap:
  588. self._snapshot = snap._snapshot # pylint:disable=protected-access
  589. else:
  590. # The snapshot no longer exists and cannot be refreshed.
  591. # set the status to unknown
  592. self._snapshot.status = 'unknown'
  593. def delete(self):
  594. """
  595. Delete this snapshot.
  596. """
  597. self._snapshot.delete()
  598. def create_volume(self, placement, size=None, volume_type=None, iops=None):
  599. """
  600. Create a new Volume from this Snapshot.
  601. """
  602. vol_name = "from_snap_{0}".format(self.id or self.name)
  603. size = size if size else self._snapshot.size
  604. os_vol = self._provider.cinder.volumes.create(
  605. size, name=vol_name, availability_zone=placement,
  606. snapshot_id=self._snapshot.id)
  607. cb_vol = OpenStackVolume(self._provider, os_vol)
  608. cb_vol.name = vol_name
  609. return cb_vol
  610. class OpenStackGatewayContainer(BaseGatewayContainer):
  611. """For OpenStack, an internet gateway is a just an 'external' network."""
  612. def __init__(self, provider, network):
  613. super(OpenStackGatewayContainer, self).__init__(provider, network)
  614. def get_or_create_inet_gateway(self, name=None):
  615. """For OS, inet gtw is any net that has `external` property set."""
  616. if name:
  617. OpenStackInternetGateway.assert_valid_resource_name(name)
  618. for n in self._provider.networking.networks:
  619. if n.external:
  620. return OpenStackInternetGateway(self._provider, n)
  621. return None
  622. def delete(self, gateway):
  623. log.debug("Deleting OpenStack Gateway: %s", gateway)
  624. gateway.delete()
  625. def list(self, limit=None, marker=None):
  626. log.debug("OpenStack listing of all current internet gateways")
  627. igl = [OpenStackInternetGateway(self._provider, n)
  628. for n in self._provider.networking.networks if n.external]
  629. return ClientPagedResultList(self._provider, igl, limit=limit,
  630. marker=marker)
  631. class OpenStackNetwork(BaseNetwork):
  632. # Ref: https://github.com/openstack/neutron/blob/master/neutron/plugins/
  633. # common/constants.py
  634. _NETWORK_STATE_MAP = {
  635. 'PENDING_CREATE': NetworkState.PENDING,
  636. 'PENDING_UPDATE': NetworkState.PENDING,
  637. 'PENDING_DELETE': NetworkState.PENDING,
  638. 'CREATED': NetworkState.PENDING,
  639. 'INACTIVE': NetworkState.PENDING,
  640. 'DOWN': NetworkState.DOWN,
  641. 'ERROR': NetworkState.ERROR,
  642. 'ACTIVE': NetworkState.AVAILABLE
  643. }
  644. def __init__(self, provider, network):
  645. super(OpenStackNetwork, self).__init__(provider)
  646. self._network = network
  647. self._gateway_service = OpenStackGatewayContainer(provider, self)
  648. @property
  649. def id(self):
  650. return self._network.get('id', None)
  651. @property
  652. def name(self):
  653. return self._network.get('name', None)
  654. @name.setter
  655. def name(self, value): # pylint:disable=arguments-differ
  656. """
  657. Set the network name.
  658. """
  659. self.assert_valid_resource_name(value)
  660. self._provider.neutron.update_network(self.id,
  661. {'network': {'name': value}})
  662. self.refresh()
  663. @property
  664. def external(self):
  665. return self._network.get('router:external', False)
  666. @property
  667. def state(self):
  668. self.refresh()
  669. return OpenStackNetwork._NETWORK_STATE_MAP.get(
  670. self._network.get('status', None),
  671. NetworkState.UNKNOWN)
  672. @property
  673. def cidr_block(self):
  674. # OpenStack does not define a CIDR block for networks
  675. return ''
  676. def delete(self):
  677. if not self.external and self.id in str(
  678. self._provider.neutron.list_networks()):
  679. # If there are ports associated with the network, it won't delete
  680. ports = self._provider.neutron.list_ports(
  681. network_id=self.id).get('ports', [])
  682. for port in ports:
  683. try:
  684. self._provider.neutron.delete_port(port.get('id'))
  685. except PortNotFoundClient:
  686. # Ports could have already been deleted if instances
  687. # are terminated etc. so exceptions can be safely ignored
  688. pass
  689. self._provider.neutron.delete_network(self.id)
  690. @property
  691. def subnets(self):
  692. subnets = (self._provider.neutron.list_subnets(network_id=self.id)
  693. .get('subnets', []))
  694. return [OpenStackSubnet(self._provider, subnet) for subnet in subnets]
  695. def refresh(self):
  696. """Refresh the state of this network by re-querying the provider."""
  697. network = self._provider.networking.networks.get(self.id)
  698. if network:
  699. # pylint:disable=protected-access
  700. self._network = network._network
  701. else:
  702. # subnet no longer exists
  703. self._network.state = NetworkState.UNKNOWN
  704. @property
  705. def gateways(self):
  706. return self._gateway_service
  707. class OpenStackSubnet(BaseSubnet):
  708. def __init__(self, provider, subnet):
  709. super(OpenStackSubnet, self).__init__(provider)
  710. self._subnet = subnet
  711. self._state = None
  712. @property
  713. def id(self):
  714. return self._subnet.get('id', None)
  715. @property
  716. def name(self):
  717. return self._subnet.get('name', None)
  718. @name.setter
  719. def name(self, value): # pylint:disable=arguments-differ
  720. """
  721. Set the subnet name.
  722. """
  723. self.assert_valid_resource_name(value)
  724. self._provider.neutron.update_subnet(
  725. self.id, {'subnet': {'name': value}})
  726. self._subnet['name'] = value
  727. @property
  728. def cidr_block(self):
  729. return self._subnet.get('cidr', None)
  730. @property
  731. def network_id(self):
  732. return self._subnet.get('network_id', None)
  733. @property
  734. def zone(self):
  735. """
  736. OpenStack does not have a notion of placement zone for subnets.
  737. Default to None.
  738. """
  739. return None
  740. def delete(self):
  741. if self.id in str(self._provider.neutron.list_subnets()):
  742. self._provider.neutron.delete_subnet(self.id)
  743. @property
  744. def state(self):
  745. return SubnetState.UNKNOWN if self._state == SubnetState.UNKNOWN \
  746. else SubnetState.AVAILABLE
  747. def refresh(self):
  748. subnet = self._provider.networking.subnets.get(self.id)
  749. if subnet:
  750. # pylint:disable=protected-access
  751. self._subnet = subnet._subnet
  752. self._state = SubnetState.AVAILABLE
  753. else:
  754. # subnet no longer exists
  755. self._state = SubnetState.UNKNOWN
  756. class OpenStackFloatingIPContainer(BaseFloatingIPContainer):
  757. def __init__(self, provider, gateway):
  758. super(OpenStackFloatingIPContainer, self).__init__(provider, gateway)
  759. def get(self, fip_id):
  760. try:
  761. return OpenStackFloatingIP(
  762. self._provider, self._provider.os_conn.network.get_ip(fip_id))
  763. except ResourceNotFound:
  764. return None
  765. def list(self, limit=None, marker=None):
  766. fips = [OpenStackFloatingIP(self._provider, fip)
  767. for fip in self._provider.os_conn.network.ips(
  768. floating_network_id=self.gateway.id
  769. )]
  770. return ClientPagedResultList(self._provider, fips,
  771. limit=limit, marker=marker)
  772. def create(self):
  773. return OpenStackFloatingIP(
  774. self._provider, self._provider.os_conn.network.create_ip(
  775. floating_network_id=self.gateway.id))
  776. class OpenStackFloatingIP(BaseFloatingIP):
  777. def __init__(self, provider, floating_ip):
  778. super(OpenStackFloatingIP, self).__init__(provider)
  779. self._ip = floating_ip
  780. @property
  781. def id(self):
  782. return self._ip.id
  783. @property
  784. def public_ip(self):
  785. return self._ip.floating_ip_address
  786. @property
  787. def private_ip(self):
  788. return self._ip.fixed_ip_address
  789. @property
  790. def in_use(self):
  791. return bool(self._ip.port_id)
  792. def delete(self):
  793. self._ip.delete(self._provider.os_conn.session)
  794. def refresh(self):
  795. net = self._provider.networking.networks.get(
  796. self._ip.floating_network_id)
  797. gw = self._provider.networking.gateways.get_or_create_inet_gateway(
  798. net)
  799. fip = gw.floating_ips.get(self.id)
  800. # pylint:disable=protected-access
  801. self._ip = fip._ip
  802. class OpenStackRouter(BaseRouter):
  803. def __init__(self, provider, router):
  804. super(OpenStackRouter, self).__init__(provider)
  805. self._router = router
  806. @property
  807. def id(self):
  808. return self._router.get('id', None)
  809. @property
  810. def name(self):
  811. return self._router.get('name', None)
  812. @name.setter
  813. def name(self, value): # pylint:disable=arguments-differ
  814. """
  815. Set the router name.
  816. """
  817. self.assert_valid_resource_name(value)
  818. self._provider.neutron.update_router(
  819. self.id, {'router': {'name': value}})
  820. self.refresh()
  821. def refresh(self):
  822. self._router = self._provider.neutron.show_router(self.id)['router']
  823. @property
  824. def state(self):
  825. if self._router.get('external_gateway_info'):
  826. return RouterState.ATTACHED
  827. return RouterState.DETACHED
  828. @property
  829. def network_id(self):
  830. if self.state == RouterState.ATTACHED:
  831. return self._router.get('external_gateway_info', {}).get(
  832. 'network_id', None)
  833. return None
  834. def delete(self):
  835. self._provider.neutron.delete_router(self.id)
  836. def attach_subnet(self, subnet):
  837. router_interface = {'subnet_id': subnet.id}
  838. ret = self._provider.neutron.add_interface_router(
  839. self.id, router_interface)
  840. if subnet.id in ret.get('subnet_ids', ""):
  841. return True
  842. return False
  843. def detach_subnet(self, subnet):
  844. router_interface = {'subnet_id': subnet.id}
  845. ret = self._provider.neutron.remove_interface_router(
  846. self.id, router_interface)
  847. if subnet.id in ret.get('subnet_ids', ""):
  848. return True
  849. return False
  850. def attach_gateway(self, gateway):
  851. self._provider.neutron.add_gateway_router(
  852. self.id, {'network_id': gateway.id})
  853. def detach_gateway(self, gateway):
  854. self._provider.neutron.remove_gateway_router(
  855. self.id).get('router', self._router)
  856. class OpenStackInternetGateway(BaseInternetGateway):
  857. GATEWAY_STATE_MAP = {
  858. NetworkState.AVAILABLE: GatewayState.AVAILABLE,
  859. NetworkState.DOWN: GatewayState.ERROR,
  860. NetworkState.ERROR: GatewayState.ERROR,
  861. NetworkState.PENDING: GatewayState.CONFIGURING,
  862. NetworkState.UNKNOWN: GatewayState.UNKNOWN
  863. }
  864. def __init__(self, provider, gateway_net):
  865. super(OpenStackInternetGateway, self).__init__(provider)
  866. if isinstance(gateway_net, OpenStackNetwork):
  867. # pylint:disable=protected-access
  868. gateway_net = gateway_net._network
  869. self._gateway_net = gateway_net
  870. self._fips_container = OpenStackFloatingIPContainer(provider, self)
  871. @property
  872. def id(self):
  873. return self._gateway_net.get('id', None)
  874. @property
  875. def name(self):
  876. return self._gateway_net.get('name', None)
  877. @name.setter
  878. # pylint:disable=arguments-differ
  879. def name(self, value):
  880. self.assert_valid_resource_name(value)
  881. self._provider.neutron.update_network(self.id,
  882. {'network': {'name': value}})
  883. self.refresh()
  884. @property
  885. def network_id(self):
  886. return self._gateway_net.get('id')
  887. def refresh(self):
  888. """Refresh the state of this network by re-querying the provider."""
  889. network = self._provider.networking.networks.get(self.id)
  890. if network:
  891. # pylint:disable=protected-access
  892. self._gateway_net = network._network
  893. else:
  894. # subnet no longer exists
  895. self._gateway_net.state = NetworkState.UNKNOWN
  896. @property
  897. def state(self):
  898. return self.GATEWAY_STATE_MAP.get(
  899. self._gateway_net.state, GatewayState.UNKNOWN)
  900. def delete(self):
  901. """Do nothing on openstack"""
  902. pass
  903. @property
  904. def floating_ips(self):
  905. return self._fips_container
  906. class OpenStackKeyPair(BaseKeyPair):
  907. def __init__(self, provider, key_pair):
  908. super(OpenStackKeyPair, self).__init__(provider, key_pair)
  909. class OpenStackVMFirewall(BaseVMFirewall):
  910. def __init__(self, provider, vm_firewall):
  911. super(OpenStackVMFirewall, self).__init__(provider, vm_firewall)
  912. self._rule_svc = OpenStackVMFirewallRuleContainer(provider, self)
  913. @property
  914. def network_id(self):
  915. """
  916. OpenStack does not associate a SG with a network so default to None.
  917. :return: Always return ``None``.
  918. """
  919. return None
  920. @property
  921. def rules(self):
  922. return self._rule_svc
  923. def delete(self):
  924. return self._vm_firewall.delete(self._provider.os_conn.session)
  925. def refresh(self):
  926. self._vm_firewall = self._provider.os_conn.network.get_security_group(
  927. self.id)
  928. def to_json(self):
  929. attr = inspect.getmembers(self, lambda a: not(inspect.isroutine(a)))
  930. js = {k: v for(k, v) in attr if not k.startswith('_')}
  931. json_rules = [r.to_json() for r in self.rules]
  932. js['rules'] = json_rules
  933. return js
  934. class OpenStackVMFirewallRuleContainer(BaseVMFirewallRuleContainer):
  935. def __init__(self, provider, firewall):
  936. super(OpenStackVMFirewallRuleContainer, self).__init__(
  937. provider, firewall)
  938. def list(self, limit=None, marker=None):
  939. # pylint:disable=protected-access
  940. rules = [OpenStackVMFirewallRule(self.firewall, r)
  941. for r in self.firewall._vm_firewall.security_group_rules]
  942. return ClientPagedResultList(self._provider, rules,
  943. limit=limit, marker=marker)
  944. def create(self, direction, protocol=None, from_port=None,
  945. to_port=None, cidr=None, src_dest_fw=None):
  946. src_dest_fw_id = (src_dest_fw.id if isinstance(src_dest_fw,
  947. OpenStackVMFirewall)
  948. else src_dest_fw)
  949. try:
  950. if direction == TrafficDirection.INBOUND:
  951. os_direction = 'ingress'
  952. elif direction == TrafficDirection.OUTBOUND:
  953. os_direction = 'egress'
  954. else:
  955. raise InvalidValueException("direction", direction)
  956. # pylint:disable=protected-access
  957. rule = self._provider.os_conn.network.create_security_group_rule(
  958. security_group_id=self.firewall.id,
  959. direction=os_direction,
  960. port_range_max=to_port,
  961. port_range_min=from_port,
  962. protocol=protocol,
  963. remote_ip_prefix=cidr,
  964. remote_group_id=src_dest_fw_id)
  965. self.firewall.refresh()
  966. return OpenStackVMFirewallRule(self.firewall, rule.to_dict())
  967. except HttpException as e:
  968. self.firewall.refresh()
  969. # 409=Conflict, raised for duplicate rule
  970. if e.http_status == 409:
  971. existing = self.find(direction=direction, protocol=protocol,
  972. from_port=from_port, to_port=to_port,
  973. cidr=cidr, src_dest_fw_id=src_dest_fw_id)
  974. return existing[0]
  975. else:
  976. raise e
  977. class OpenStackVMFirewallRule(BaseVMFirewallRule):
  978. def __init__(self, parent_fw, rule):
  979. super(OpenStackVMFirewallRule, self).__init__(parent_fw, rule)
  980. @property
  981. def id(self):
  982. return self._rule.get('id')
  983. @property
  984. def direction(self):
  985. direction = self._rule.get('direction')
  986. if direction == 'ingress':
  987. return TrafficDirection.INBOUND
  988. elif direction == 'egress':
  989. return TrafficDirection.OUTBOUND
  990. else:
  991. return None
  992. @property
  993. def protocol(self):
  994. return self._rule.get('protocol')
  995. @property
  996. def from_port(self):
  997. return self._rule.get('port_range_min')
  998. @property
  999. def to_port(self):
  1000. return self._rule.get('port_range_max')
  1001. @property
  1002. def cidr(self):
  1003. return self._rule.get('remote_ip_prefix')
  1004. @property
  1005. def src_dest_fw_id(self):
  1006. fw = self.src_dest_fw
  1007. if fw:
  1008. return fw.id
  1009. return None
  1010. @property
  1011. def src_dest_fw(self):
  1012. fw_id = self._rule.get('remote_group_id')
  1013. if fw_id:
  1014. return self._provider.security.vm_firewalls.get(fw_id)
  1015. return None
  1016. def delete(self):
  1017. self._provider.os_conn.network.delete_security_group_rule(self.id)
  1018. self.firewall.refresh()
  1019. class OpenStackBucketObject(BaseBucketObject):
  1020. def __init__(self, provider, cbcontainer, obj):
  1021. super(OpenStackBucketObject, self).__init__(provider)
  1022. self.cbcontainer = cbcontainer
  1023. self._obj = obj
  1024. @property
  1025. def id(self):
  1026. return self._obj.get("name")
  1027. @property
  1028. def name(self):
  1029. """Get this object's name."""
  1030. return self._obj.get("name")
  1031. @property
  1032. def size(self):
  1033. return self._obj.get("bytes")
  1034. @property
  1035. def last_modified(self):
  1036. return self._obj.get("last_modified")
  1037. def iter_content(self):
  1038. """Returns this object's content as an iterable."""
  1039. _, content = self._provider.swift.get_object(
  1040. self.cbcontainer.name, self.name, resp_chunk_size=65536)
  1041. return content
  1042. def upload(self, data):
  1043. """
  1044. Set the contents of this object to the data read from the source
  1045. string.
  1046. .. warning:: Will fail if the data is larger than 5 Gig.
  1047. """
  1048. self._provider.swift.put_object(self.cbcontainer.name, self.name,
  1049. data)
  1050. def upload_from_file(self, path):
  1051. """
  1052. Stores the contents of the file pointed by the ``path`` variable.
  1053. If the file is bigger than 5 Gig, it will be broken into segments.
  1054. :type path: ``str``
  1055. :param path: Absolute path to the file to be uploaded to Swift.
  1056. :rtype: ``bool``
  1057. :return: ``True`` if successful, ``False`` if not.
  1058. .. note::
  1059. * The size of the segments chosen (or any of the other upload
  1060. options) is not under user control.
  1061. * If called this method will remap the
  1062. ``swiftclient.service.get_conn`` factory method to
  1063. ``self._provider._connect_swift``
  1064. .. seealso:: https://github.com/gvlproject/cloudbridge/issues/35#issuecomment-297629661 # noqa
  1065. """
  1066. upload_options = {}
  1067. if 'segment_size' not in upload_options:
  1068. if os.path.getsize(path) >= FIVE_GIG:
  1069. upload_options['segment_size'] = FIVE_GIG
  1070. # remap the swift service's connection factory method
  1071. # pylint:disable=protected-access
  1072. swiftclient.service.get_conn = self._provider._connect_swift
  1073. result = True
  1074. with SwiftService() as swift:
  1075. upload_object = SwiftUploadObject(path, object_name=self.name)
  1076. for up_res in swift.upload(self.cbcontainer.name,
  1077. [upload_object, ],
  1078. options=upload_options):
  1079. result = result and up_res['success']
  1080. return result
  1081. def delete(self):
  1082. """
  1083. Delete this object.
  1084. :rtype: ``bool``
  1085. :return: True if successful
  1086. .. note:: If called this method will remap the
  1087. ``swiftclient.service.get_conn`` factory method to
  1088. ``self._provider._connect_swift``
  1089. """
  1090. # remap the swift service's connection factory method
  1091. # pylint:disable=protected-access
  1092. swiftclient.service.get_conn = self._provider._connect_swift
  1093. result = True
  1094. with SwiftService() as swift:
  1095. for del_res in swift.delete(self.cbcontainer.name, [self.name, ]):
  1096. result = result and del_res['success']
  1097. return result
  1098. def generate_url(self, expires_in=0):
  1099. """
  1100. Generates a URL to this object.
  1101. If the object is public, `expires_in` argument is not necessary, but if
  1102. the object is private, the life time of URL is set using `expires_in`
  1103. argument.
  1104. See here for implementation details:
  1105. http://stackoverflow.com/a/37057172
  1106. """
  1107. raise NotImplementedError("This functionality is not implemented yet.")
  1108. class OpenStackBucket(BaseBucket):
  1109. def __init__(self, provider, bucket):
  1110. super(OpenStackBucket, self).__init__(provider)
  1111. self._bucket = bucket
  1112. self._object_container = OpenStackBucketContainer(provider, self)
  1113. @property
  1114. def id(self):
  1115. return self._bucket.get("name")
  1116. @property
  1117. def name(self):
  1118. return self._bucket.get("name")
  1119. @property
  1120. def objects(self):
  1121. return self._object_container
  1122. def delete(self, delete_contents=False):
  1123. self._provider.swift.delete_container(self.name)
  1124. class OpenStackBucketContainer(BaseBucketContainer):
  1125. def __init__(self, provider, bucket):
  1126. super(OpenStackBucketContainer, self).__init__(provider, bucket)
  1127. def get(self, name):
  1128. """
  1129. Retrieve a given object from this bucket.
  1130. FIXME: If multiple objects match the name as their name prefix,
  1131. all will be returned by the provider but this method will only
  1132. return the first element.
  1133. """
  1134. _, object_list = self._provider.swift.get_container(
  1135. self.bucket.name, prefix=name)
  1136. if object_list:
  1137. return OpenStackBucketObject(self._provider, self.bucket,
  1138. object_list[0])
  1139. else:
  1140. return None
  1141. def list(self, limit=None, marker=None, prefix=None):
  1142. """
  1143. List all objects within this bucket.
  1144. :rtype: BucketObject
  1145. :return: List of all available BucketObjects within this bucket.
  1146. """
  1147. _, object_list = self._provider.swift.get_container(
  1148. self.bucket.name,
  1149. limit=oshelpers.os_result_limit(self._provider, limit),
  1150. marker=marker, prefix=prefix)
  1151. cb_objects = [OpenStackBucketObject(
  1152. self._provider, self.bucket, obj) for obj in object_list]
  1153. return oshelpers.to_server_paged_list(
  1154. self._provider,
  1155. cb_objects,
  1156. limit)
  1157. def find(self, **kwargs):
  1158. obj_list = self
  1159. filters = ['name']
  1160. matches = cb_helpers.generic_find(filters, kwargs, obj_list)
  1161. return ClientPagedResultList(self._provider, list(matches))
  1162. def create(self, object_name):
  1163. self._provider.swift.put_object(self.bucket.name, object_name, None)
  1164. return self.get(object_name)