resources.py 48 KB

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