services.py 37 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983
  1. """
  2. Services implemented by the OpenStack provider.
  3. """
  4. import logging
  5. from cinderclient.exceptions import NotFound as CinderNotFound
  6. from neutronclient.common.exceptions import NeutronClientException
  7. from novaclient.exceptions import NotFound as NovaNotFound
  8. from openstack.exceptions import NotFoundException
  9. from openstack.exceptions import ResourceNotFound
  10. from swiftclient import ClientException as SwiftClientException
  11. import cloudbridge.cloud.base.helpers as cb_helpers
  12. from cloudbridge.cloud.base.resources import BaseLaunchConfig
  13. from cloudbridge.cloud.base.resources import ClientPagedResultList
  14. from cloudbridge.cloud.base.services import BaseBucketObjectService
  15. from cloudbridge.cloud.base.services import BaseBucketService
  16. from cloudbridge.cloud.base.services import BaseComputeService
  17. from cloudbridge.cloud.base.services import BaseImageService
  18. from cloudbridge.cloud.base.services import BaseInstanceService
  19. from cloudbridge.cloud.base.services import BaseKeyPairService
  20. from cloudbridge.cloud.base.services import BaseNetworkService
  21. from cloudbridge.cloud.base.services import BaseNetworkingService
  22. from cloudbridge.cloud.base.services import BaseRegionService
  23. from cloudbridge.cloud.base.services import BaseRouterService
  24. from cloudbridge.cloud.base.services import BaseSecurityService
  25. from cloudbridge.cloud.base.services import BaseSnapshotService
  26. from cloudbridge.cloud.base.services import BaseStorageService
  27. from cloudbridge.cloud.base.services import BaseSubnetService
  28. from cloudbridge.cloud.base.services import BaseVMFirewallService
  29. from cloudbridge.cloud.base.services import BaseVMTypeService
  30. from cloudbridge.cloud.base.services import BaseVolumeService
  31. from cloudbridge.cloud.interfaces.exceptions \
  32. import DuplicateResourceException
  33. from cloudbridge.cloud.interfaces.resources import KeyPair
  34. from cloudbridge.cloud.interfaces.resources import MachineImage
  35. from cloudbridge.cloud.interfaces.resources import Network
  36. from cloudbridge.cloud.interfaces.resources import PlacementZone
  37. from cloudbridge.cloud.interfaces.resources import Snapshot
  38. from cloudbridge.cloud.interfaces.resources import Subnet
  39. from cloudbridge.cloud.interfaces.resources import VMFirewall
  40. from cloudbridge.cloud.interfaces.resources import VMType
  41. from cloudbridge.cloud.interfaces.resources import Volume
  42. from cloudbridge.cloud.providers.openstack import helpers as oshelpers
  43. from .resources import OpenStackBucket
  44. from .resources import OpenStackBucketObject
  45. from .resources import OpenStackInstance
  46. from .resources import OpenStackKeyPair
  47. from .resources import OpenStackMachineImage
  48. from .resources import OpenStackNetwork
  49. from .resources import OpenStackRegion
  50. from .resources import OpenStackRouter
  51. from .resources import OpenStackSnapshot
  52. from .resources import OpenStackSubnet
  53. from .resources import OpenStackVMFirewall
  54. from .resources import OpenStackVMType
  55. from .resources import OpenStackVolume
  56. log = logging.getLogger(__name__)
  57. class OpenStackSecurityService(BaseSecurityService):
  58. def __init__(self, provider):
  59. super(OpenStackSecurityService, self).__init__(provider)
  60. # Initialize provider services
  61. self._key_pairs = OpenStackKeyPairService(provider)
  62. self._vm_firewalls = OpenStackVMFirewallService(provider)
  63. @property
  64. def key_pairs(self):
  65. """
  66. Provides access to key pairs for this provider.
  67. :rtype: ``object`` of :class:`.KeyPairService`
  68. :return: a KeyPairService object
  69. """
  70. return self._key_pairs
  71. @property
  72. def vm_firewalls(self):
  73. """
  74. Provides access to VM firewalls for this provider.
  75. :rtype: ``object`` of :class:`.VMFirewallService`
  76. :return: a VMFirewallService object
  77. """
  78. return self._vm_firewalls
  79. def get_or_create_ec2_credentials(self):
  80. """
  81. A provider specific method than returns the ec2 credentials for the
  82. current user, or creates a new pair if one doesn't exist.
  83. """
  84. keystone = self.provider.keystone
  85. if hasattr(keystone, 'ec2'):
  86. user_id = keystone.session.get_user_id()
  87. user_creds = [cred for cred in keystone.ec2.list(user_id) if
  88. cred.tenant_id == keystone.session.get_project_id()]
  89. if user_creds:
  90. return user_creds[0]
  91. else:
  92. return keystone.ec2.create(
  93. user_id, keystone.session.get_project_id())
  94. return None
  95. def get_ec2_endpoints(self):
  96. """
  97. A provider specific method than returns the ec2 endpoints if
  98. available.
  99. """
  100. keystone = self.provider.keystone
  101. ec2_url = keystone.session.get_endpoint(service_type='ec2')
  102. s3_url = keystone.session.get_endpoint(service_type='s3')
  103. return {'ec2_endpoint': ec2_url,
  104. 's3_endpoint': s3_url}
  105. class OpenStackKeyPairService(BaseKeyPairService):
  106. def __init__(self, provider):
  107. super(OpenStackKeyPairService, self).__init__(provider)
  108. def get(self, key_pair_id):
  109. """
  110. Returns a KeyPair given its id.
  111. """
  112. log.debug("Returning KeyPair with the id %s", key_pair_id)
  113. try:
  114. return OpenStackKeyPair(
  115. self.provider, self.provider.nova.keypairs.get(key_pair_id))
  116. except NovaNotFound:
  117. log.debug("KeyPair %s was not found.", key_pair_id)
  118. return None
  119. def list(self, limit=None, marker=None):
  120. """
  121. List all key pairs associated with this account.
  122. :rtype: ``list`` of :class:`.KeyPair`
  123. :return: list of KeyPair objects
  124. """
  125. keypairs = self.provider.nova.keypairs.list()
  126. results = [OpenStackKeyPair(self.provider, kp)
  127. for kp in keypairs]
  128. log.debug("Listing all key pairs associated with OpenStack "
  129. "Account: %s", results)
  130. return ClientPagedResultList(self.provider, results,
  131. limit=limit, marker=marker)
  132. def find(self, **kwargs):
  133. name = kwargs.pop('name', None)
  134. # All kwargs should have been popped at this time.
  135. if len(kwargs) > 0:
  136. raise TypeError("Unrecognised parameters for search: %s."
  137. " Supported attributes: %s" % (kwargs, 'name'))
  138. keypairs = self.provider.nova.keypairs.findall(name=name)
  139. results = [OpenStackKeyPair(self.provider, kp)
  140. for kp in keypairs]
  141. log.debug("Searching for %s in: %s", name, keypairs)
  142. return ClientPagedResultList(self.provider, results)
  143. def create(self, name, public_key_material=None):
  144. log.debug("Creating a new key pair with the name: %s", name)
  145. OpenStackKeyPair.assert_valid_resource_name(name)
  146. existing_kp = self.find(name=name)
  147. if existing_kp:
  148. raise DuplicateResourceException(
  149. 'Keypair already exists with name {0}'.format(name))
  150. private_key = None
  151. if not public_key_material:
  152. public_key_material, private_key = cb_helpers.generate_key_pair()
  153. kp = self.provider.nova.keypairs.create(name,
  154. public_key=public_key_material)
  155. cb_kp = OpenStackKeyPair(self.provider, kp)
  156. cb_kp.material = private_key
  157. return cb_kp
  158. class OpenStackVMFirewallService(BaseVMFirewallService):
  159. def __init__(self, provider):
  160. super(OpenStackVMFirewallService, self).__init__(provider)
  161. def get(self, firewall_id):
  162. log.debug("Getting OpenStack VM Firewall with the id: %s", firewall_id)
  163. try:
  164. return OpenStackVMFirewall(
  165. self.provider,
  166. self.provider.os_conn.network.get_security_group(firewall_id))
  167. except (ResourceNotFound, NotFoundException):
  168. log.debug("Firewall %s not found.", firewall_id)
  169. return None
  170. def list(self, limit=None, marker=None):
  171. firewalls = [
  172. OpenStackVMFirewall(self.provider, fw)
  173. for fw in self.provider.os_conn.network.security_groups()]
  174. return ClientPagedResultList(self.provider, firewalls,
  175. limit=limit, marker=marker)
  176. @cb_helpers.deprecated_alias(network_id='network')
  177. def create(self, label, network, description=None):
  178. OpenStackVMFirewall.assert_valid_resource_label(label)
  179. log.debug("Creating OpenStack VM Firewall with the params: "
  180. "[label: %s network id: %s description: %s]", label,
  181. network, description)
  182. net = network.id if isinstance(network, Network) else network
  183. # We generally simulate a network being associated with a firewall
  184. # by storing the supplied value in the firewall description field that
  185. # is not modifiable after creation; however, because of some networking
  186. # specificity in Nectar, we must also allow an empty network id value.
  187. if not net:
  188. net = ""
  189. if not description:
  190. description = ""
  191. description += " [{}{}]".format(OpenStackVMFirewall._network_id_tag,
  192. net)
  193. sg = self.provider.os_conn.network.create_security_group(
  194. name=label, description=description)
  195. if sg:
  196. return OpenStackVMFirewall(self.provider, sg)
  197. return None
  198. def delete(self, group_id):
  199. log.debug("Deleting OpenStack Firewall with the id: %s", group_id)
  200. firewall = self.get(group_id)
  201. if firewall:
  202. firewall.delete()
  203. return True
  204. class OpenStackStorageService(BaseStorageService):
  205. def __init__(self, provider):
  206. super(OpenStackStorageService, self).__init__(provider)
  207. # Initialize provider services
  208. self._volume_svc = OpenStackVolumeService(self.provider)
  209. self._snapshot_svc = OpenStackSnapshotService(self.provider)
  210. self._bucket_svc = OpenStackBucketService(self.provider)
  211. self._bucket_obj_svc = OpenStackBucketObjectService(self.provider)
  212. @property
  213. def volumes(self):
  214. return self._volume_svc
  215. @property
  216. def snapshots(self):
  217. return self._snapshot_svc
  218. @property
  219. def buckets(self):
  220. return self._bucket_svc
  221. @property
  222. def bucket_objects(self):
  223. return self._bucket_obj_svc
  224. class OpenStackVolumeService(BaseVolumeService):
  225. def __init__(self, provider):
  226. super(OpenStackVolumeService, self).__init__(provider)
  227. def get(self, volume_id):
  228. """
  229. Returns a volume given its id.
  230. """
  231. log.debug("Getting OpenStack Volume with the id: %s", volume_id)
  232. try:
  233. return OpenStackVolume(
  234. self.provider, self.provider.cinder.volumes.get(volume_id))
  235. except CinderNotFound:
  236. log.debug("Volume %s was not found.", volume_id)
  237. return None
  238. def find(self, **kwargs):
  239. label = kwargs.pop('label', None)
  240. # All kwargs should have been popped at this time.
  241. if len(kwargs) > 0:
  242. raise TypeError("Unrecognised parameters for search: %s."
  243. " Supported attributes: %s" % (kwargs, 'label'))
  244. log.debug("Searching for an OpenStack Volume with the label %s", label)
  245. search_opts = {'name': label}
  246. cb_vols = [
  247. OpenStackVolume(self.provider, vol)
  248. for vol in self.provider.cinder.volumes.list(
  249. search_opts=search_opts,
  250. limit=oshelpers.os_result_limit(self.provider),
  251. marker=None)]
  252. return oshelpers.to_server_paged_list(self.provider, cb_vols)
  253. def list(self, limit=None, marker=None):
  254. """
  255. List all volumes.
  256. """
  257. cb_vols = [
  258. OpenStackVolume(self.provider, vol)
  259. for vol in self.provider.cinder.volumes.list(
  260. limit=oshelpers.os_result_limit(self.provider, limit),
  261. marker=marker)]
  262. return oshelpers.to_server_paged_list(self.provider, cb_vols, limit)
  263. def create(self, label, size, zone, snapshot=None, description=None):
  264. """
  265. Creates a new volume.
  266. """
  267. log.debug("Creating a new volume with the params: "
  268. "[label: %s size: %s zone: %s snapshot: %s description: %s]",
  269. label, size, zone, snapshot, description)
  270. OpenStackVolume.assert_valid_resource_label(label)
  271. zone_id = zone.id if isinstance(zone, PlacementZone) else zone
  272. snapshot_id = snapshot.id if isinstance(
  273. snapshot, OpenStackSnapshot) and snapshot else snapshot
  274. os_vol = self.provider.cinder.volumes.create(
  275. size, name=label, description=description,
  276. availability_zone=zone_id, snapshot_id=snapshot_id)
  277. return OpenStackVolume(self.provider, os_vol)
  278. class OpenStackSnapshotService(BaseSnapshotService):
  279. def __init__(self, provider):
  280. super(OpenStackSnapshotService, self).__init__(provider)
  281. def get(self, snapshot_id):
  282. """
  283. Returns a snapshot given its id.
  284. """
  285. log.debug("Getting OpenStack snapshot with the id: %s", snapshot_id)
  286. try:
  287. return OpenStackSnapshot(
  288. self.provider,
  289. self.provider.cinder.volume_snapshots.get(snapshot_id))
  290. except CinderNotFound:
  291. log.debug("Snapshot %s was not found.", snapshot_id)
  292. return None
  293. def find(self, **kwargs):
  294. label = kwargs.pop('label', None)
  295. # All kwargs should have been popped at this time.
  296. if len(kwargs) > 0:
  297. raise TypeError("Unrecognised parameters for search: %s."
  298. " Supported attributes: %s" % (kwargs, 'label'))
  299. search_opts = {'name': label, # TODO: Cinder is ignoring name
  300. 'limit': oshelpers.os_result_limit(self.provider),
  301. 'marker': None}
  302. log.debug("Searching for an OpenStack snapshot with the following "
  303. "params: %s", search_opts)
  304. cb_snaps = [
  305. OpenStackSnapshot(self.provider, snap) for
  306. snap in self.provider.cinder.volume_snapshots.list(search_opts)
  307. if snap.name == label]
  308. return oshelpers.to_server_paged_list(self.provider, cb_snaps)
  309. def list(self, limit=None, marker=None):
  310. """
  311. List all snapshot.
  312. """
  313. cb_snaps = [
  314. OpenStackSnapshot(self.provider, snap) for
  315. snap in self.provider.cinder.volume_snapshots.list(
  316. search_opts={'limit': oshelpers.os_result_limit(self.provider,
  317. limit),
  318. 'marker': marker})]
  319. return oshelpers.to_server_paged_list(self.provider, cb_snaps, limit)
  320. def create(self, label, volume, description=None):
  321. """
  322. Creates a new snapshot of a given volume.
  323. """
  324. log.debug("Creating a new snapshot of the %s volume.", label)
  325. OpenStackSnapshot.assert_valid_resource_label(label)
  326. volume_id = (volume.id if isinstance(volume, OpenStackVolume)
  327. else volume)
  328. os_snap = self.provider.cinder.volume_snapshots.create(
  329. volume_id, name=label,
  330. description=description)
  331. return OpenStackSnapshot(self.provider, os_snap)
  332. class OpenStackBucketService(BaseBucketService):
  333. def __init__(self, provider):
  334. super(OpenStackBucketService, self).__init__(provider)
  335. def _get(self, bucket_id):
  336. """
  337. Returns a bucket given its ID. Returns ``None`` if the bucket
  338. does not exist.
  339. """
  340. _, container_list = self.provider.swift.get_account(
  341. prefix=bucket_id)
  342. if container_list:
  343. return OpenStackBucket(self.provider,
  344. next((c for c in container_list
  345. if c['name'] == bucket_id), None))
  346. else:
  347. log.debug("Bucket %s was not found.", bucket_id)
  348. return None
  349. def _find(self, **kwargs):
  350. name = kwargs.pop('name', None)
  351. # All kwargs should have been popped at this time.
  352. if len(kwargs) > 0:
  353. raise TypeError("Unrecognised parameters for search: %s."
  354. " Supported attributes: %s" % (kwargs, 'name'))
  355. _, container_list = self.provider.swift.get_account()
  356. cb_buckets = [OpenStackBucket(self.provider, c)
  357. for c in container_list
  358. if name in c.get("name")]
  359. return oshelpers.to_server_paged_list(self.provider, cb_buckets)
  360. def _list(self, limit, marker):
  361. _, container_list = self.provider.swift.get_account(
  362. limit=oshelpers.os_result_limit(self.provider, limit),
  363. marker=marker)
  364. cb_buckets = [OpenStackBucket(self.provider, c)
  365. for c in container_list]
  366. return oshelpers.to_server_paged_list(self.provider, cb_buckets, limit)
  367. def _create(self, name, location):
  368. try:
  369. self.provider.swift.head_container(name)
  370. raise DuplicateResourceException(
  371. 'Bucket already exists with name {0}'.format(name))
  372. except SwiftClientException:
  373. self.provider.swift.put_container(name)
  374. return self.get(name)
  375. class OpenStackBucketObjectService(BaseBucketObjectService):
  376. def __init__(self, provider):
  377. super(OpenStackBucketObjectService, self).__init__(provider)
  378. def get(self, bucket, name):
  379. """
  380. Retrieve a given object from this bucket.
  381. """
  382. # Swift always returns a reference for the container first,
  383. # followed by a list containing references to objects.
  384. _, object_list = self.provider.swift.get_container(
  385. bucket.name, prefix=name)
  386. # Loop through list of objects looking for an exact name vs. a prefix
  387. for obj in object_list:
  388. if obj.get('name') == name:
  389. return OpenStackBucketObject(self.provider,
  390. bucket,
  391. obj)
  392. return None
  393. def list(self, bucket, limit=None, marker=None, prefix=None):
  394. """
  395. List all objects within this bucket.
  396. :rtype: BucketObject
  397. :return: List of all available BucketObjects within this bucket.
  398. """
  399. _, object_list = self.provider.swift.get_container(
  400. bucket.name,
  401. limit=oshelpers.os_result_limit(self.provider, limit),
  402. marker=marker, prefix=prefix)
  403. cb_objects = [OpenStackBucketObject(
  404. self.provider, bucket, obj) for obj in object_list]
  405. return oshelpers.to_server_paged_list(
  406. self.provider,
  407. cb_objects,
  408. limit)
  409. def find(self, bucket, **kwargs):
  410. _, obj_list = self.provider.swift.get_container(bucket.name)
  411. cb_objs = [OpenStackBucketObject(self.provider, bucket, obj)
  412. for obj in obj_list]
  413. filters = ['name']
  414. matches = cb_helpers.generic_find(filters, kwargs, cb_objs)
  415. return ClientPagedResultList(self.provider, list(matches))
  416. def create(self, bucket, object_name):
  417. self.provider.swift.put_object(bucket.name, object_name, None)
  418. return self.get(object_name)
  419. class OpenStackComputeService(BaseComputeService):
  420. def __init__(self, provider):
  421. super(OpenStackComputeService, self).__init__(provider)
  422. self._vm_type_svc = OpenStackVMTypeService(self.provider)
  423. self._instance_svc = OpenStackInstanceService(self.provider)
  424. self._region_svc = OpenStackRegionService(self.provider)
  425. self._images_svc = OpenStackImageService(self.provider)
  426. @property
  427. def images(self):
  428. return self._images_svc
  429. @property
  430. def vm_types(self):
  431. return self._vm_type_svc
  432. @property
  433. def instances(self):
  434. return self._instance_svc
  435. @property
  436. def regions(self):
  437. return self._region_svc
  438. class OpenStackImageService(BaseImageService):
  439. def __init__(self, provider):
  440. super(OpenStackImageService, self).__init__(provider)
  441. def get(self, image_id):
  442. """
  443. Returns an Image given its id
  444. """
  445. log.debug("Getting OpenStack Image with the id: %s", image_id)
  446. try:
  447. return OpenStackMachineImage(
  448. self.provider, self.provider.os_conn.image.get_image(image_id))
  449. except (NotFoundException, ResourceNotFound):
  450. log.debug("Image %s not found", image_id)
  451. return None
  452. def find(self, **kwargs):
  453. filters = ['label']
  454. obj_list = self
  455. return cb_helpers.generic_find(filters, kwargs, obj_list)
  456. def list(self, filter_by_owner=True, limit=None, marker=None):
  457. """
  458. List all images.
  459. """
  460. project_id = None
  461. if filter_by_owner:
  462. project_id = self.provider.os_conn.session.get_project_id()
  463. os_images = self.provider.os_conn.image.images(
  464. owner=project_id,
  465. limit=oshelpers.os_result_limit(self.provider, limit),
  466. marker=marker)
  467. cb_images = [
  468. OpenStackMachineImage(self.provider, img)
  469. for img in os_images]
  470. return oshelpers.to_server_paged_list(self.provider, cb_images, limit)
  471. class OpenStackInstanceService(BaseInstanceService):
  472. def __init__(self, provider):
  473. super(OpenStackInstanceService, self).__init__(provider)
  474. def create(self, label, image, vm_type, subnet, zone,
  475. key_pair=None, vm_firewalls=None, user_data=None,
  476. launch_config=None, **kwargs):
  477. """Create a new virtual machine instance."""
  478. OpenStackInstance.assert_valid_resource_label(label)
  479. image_id = image.id if isinstance(image, MachineImage) else image
  480. vm_size = vm_type.id if \
  481. isinstance(vm_type, VMType) else \
  482. self.provider.compute.vm_types.find(
  483. name=vm_type)[0].id
  484. if isinstance(subnet, Subnet):
  485. subnet_id = subnet.id
  486. net_id = subnet.network_id
  487. else:
  488. subnet_id = subnet
  489. net_id = (self.provider.networking.subnets
  490. .get(subnet_id).network_id
  491. if subnet_id else None)
  492. zone_id = zone.id if isinstance(zone, PlacementZone) else zone
  493. key_pair_name = key_pair.name if \
  494. isinstance(key_pair, KeyPair) else key_pair
  495. bdm = None
  496. if launch_config:
  497. bdm = self._to_block_device_mapping(launch_config)
  498. # Security groups must be passed in as a list of IDs and attached to a
  499. # port if a port is being created. Otherwise, the security groups must
  500. # be passed in as a list of names to the servers.create() call.
  501. # OpenStack will respect the port's security groups first and then
  502. # fall-back to the named security groups.
  503. sg_name_list = []
  504. nics = None
  505. if subnet_id:
  506. log.debug("Creating network port for %s in subnet: %s",
  507. label, subnet_id)
  508. sg_list = []
  509. if vm_firewalls:
  510. if isinstance(vm_firewalls, list) and \
  511. isinstance(vm_firewalls[0], VMFirewall):
  512. sg_list = vm_firewalls
  513. else:
  514. sg_list = (self.provider.security.vm_firewalls
  515. .find(label=sg) for sg in vm_firewalls)
  516. sg_list = (sg[0] for sg in sg_list if sg)
  517. sg_id_list = [sg.id for sg in sg_list]
  518. port_def = {
  519. "port": {
  520. "admin_state_up": True,
  521. "name": OpenStackInstance._generate_name_from_label(
  522. label, 'cb-port'),
  523. "network_id": net_id,
  524. "fixed_ips": [{"subnet_id": subnet_id}],
  525. "security_groups": sg_id_list
  526. }
  527. }
  528. port_id = self.provider.neutron.create_port(port_def)['port']['id']
  529. nics = [{'net-id': net_id, 'port-id': port_id}]
  530. else:
  531. if vm_firewalls:
  532. if isinstance(vm_firewalls, list) and \
  533. isinstance(vm_firewalls[0], VMFirewall):
  534. sg_name_list = [sg.name for sg in vm_firewalls]
  535. else:
  536. sg_list = (self.provider.security.vm_firewalls.get(sg)
  537. for sg in vm_firewalls)
  538. sg_name_list = (sg[0].name for sg in sg_list if sg)
  539. log.debug("Launching in subnet %s", subnet_id)
  540. os_instance = self.provider.nova.servers.create(
  541. label,
  542. None if self._has_root_device(launch_config) else image_id,
  543. vm_size,
  544. min_count=1,
  545. max_count=1,
  546. availability_zone=zone_id,
  547. key_name=key_pair_name,
  548. security_groups=sg_name_list,
  549. userdata=str(user_data) or None,
  550. block_device_mapping_v2=bdm,
  551. nics=nics)
  552. return OpenStackInstance(self.provider, os_instance)
  553. def _to_block_device_mapping(self, launch_config):
  554. """
  555. Extracts block device mapping information
  556. from a launch config and constructs a BlockDeviceMappingV2
  557. object.
  558. """
  559. bdm = []
  560. for device in launch_config.block_devices:
  561. bdm_dict = dict()
  562. if device.is_volume:
  563. bdm_dict['destination_type'] = 'volume'
  564. if device.is_root:
  565. bdm_dict['device_name'] = '/dev/sda'
  566. bdm_dict['boot_index'] = 0
  567. if isinstance(device.source, Snapshot):
  568. bdm_dict['source_type'] = 'snapshot'
  569. bdm_dict['uuid'] = device.source.id
  570. elif isinstance(device.source, Volume):
  571. bdm_dict['source_type'] = 'volume'
  572. bdm_dict['uuid'] = device.source.id
  573. elif isinstance(device.source, MachineImage):
  574. bdm_dict['source_type'] = 'image'
  575. bdm_dict['uuid'] = device.source.id
  576. else:
  577. bdm_dict['source_type'] = 'blank'
  578. if device.delete_on_terminate is not None:
  579. bdm_dict[
  580. 'delete_on_termination'] = device.delete_on_terminate
  581. if device.size:
  582. bdm_dict['volume_size'] = device.size
  583. else:
  584. bdm_dict['destination_type'] = 'local'
  585. bdm_dict['source_type'] = 'blank'
  586. bdm_dict['delete_on_termination'] = True
  587. bdm.append(bdm_dict)
  588. return bdm
  589. def _has_root_device(self, launch_config):
  590. if not launch_config:
  591. return False
  592. for device in launch_config.block_devices:
  593. if device.is_root:
  594. return True
  595. return False
  596. def create_launch_config(self):
  597. return BaseLaunchConfig(self.provider)
  598. def find(self, **kwargs):
  599. label = kwargs.pop('label', None)
  600. # All kwargs should have been popped at this time.
  601. if len(kwargs) > 0:
  602. raise TypeError("Unrecognised parameters for search: %s."
  603. " Supported attributes: %s" % (kwargs, 'label'))
  604. search_opts = {'name': label}
  605. cb_insts = [
  606. OpenStackInstance(self.provider, inst)
  607. for inst in self.provider.nova.servers.list(
  608. search_opts=search_opts,
  609. limit=oshelpers.os_result_limit(self.provider),
  610. marker=None)]
  611. return oshelpers.to_server_paged_list(self.provider, cb_insts)
  612. def list(self, limit=None, marker=None):
  613. """
  614. List all instances.
  615. """
  616. cb_insts = [
  617. OpenStackInstance(self.provider, inst)
  618. for inst in self.provider.nova.servers.list(
  619. limit=oshelpers.os_result_limit(self.provider, limit),
  620. marker=marker)]
  621. return oshelpers.to_server_paged_list(self.provider, cb_insts, limit)
  622. def get(self, instance_id):
  623. """
  624. Returns an instance given its id.
  625. """
  626. try:
  627. os_instance = self.provider.nova.servers.get(instance_id)
  628. return OpenStackInstance(self.provider, os_instance)
  629. except NovaNotFound:
  630. log.debug("Instance %s was not found.", instance_id)
  631. return None
  632. class OpenStackVMTypeService(BaseVMTypeService):
  633. def __init__(self, provider):
  634. super(OpenStackVMTypeService, self).__init__(provider)
  635. def list(self, limit=None, marker=None):
  636. cb_itypes = [
  637. OpenStackVMType(self.provider, obj)
  638. for obj in self.provider.nova.flavors.list(
  639. limit=oshelpers.os_result_limit(self.provider, limit),
  640. marker=marker)]
  641. return oshelpers.to_server_paged_list(self.provider, cb_itypes, limit)
  642. class OpenStackRegionService(BaseRegionService):
  643. def __init__(self, provider):
  644. super(OpenStackRegionService, self).__init__(provider)
  645. def get(self, region_id):
  646. log.debug("Getting OpenStack Region with the id: %s", region_id)
  647. region = (r for r in self if r.id == region_id)
  648. return next(region, None)
  649. def list(self, limit=None, marker=None):
  650. # pylint:disable=protected-access
  651. if self.provider._keystone_version == 3:
  652. os_regions = [OpenStackRegion(self.provider, region)
  653. for region in self.provider.keystone.regions.list()]
  654. return ClientPagedResultList(self.provider, os_regions,
  655. limit=limit, marker=marker)
  656. else:
  657. # Keystone v3 onwards supports directly listing regions
  658. # but for v2, this convoluted method is necessary.
  659. regions = (
  660. endpoint.get('region') or endpoint.get('region_id')
  661. for svc in self.provider.keystone.service_catalog.get_data()
  662. for endpoint in svc.get('endpoints', [])
  663. )
  664. regions = set(region for region in regions if region)
  665. os_regions = [OpenStackRegion(self.provider, region)
  666. for region in regions]
  667. return ClientPagedResultList(self.provider, os_regions,
  668. limit=limit, marker=marker)
  669. @property
  670. def current(self):
  671. nova_region = self.provider.nova.client.region_name
  672. return self.get(nova_region) if nova_region else None
  673. class OpenStackNetworkingService(BaseNetworkingService):
  674. def __init__(self, provider):
  675. super(OpenStackNetworkingService, self).__init__(provider)
  676. self._network_service = OpenStackNetworkService(self.provider)
  677. self._subnet_service = OpenStackSubnetService(self.provider)
  678. self._router_service = OpenStackRouterService(self.provider)
  679. @property
  680. def networks(self):
  681. return self._network_service
  682. @property
  683. def subnets(self):
  684. return self._subnet_service
  685. @property
  686. def routers(self):
  687. return self._router_service
  688. class OpenStackNetworkService(BaseNetworkService):
  689. def __init__(self, provider):
  690. super(OpenStackNetworkService, self).__init__(provider)
  691. def get(self, network_id):
  692. log.debug("Getting OpenStack Network with the id: %s", network_id)
  693. network = (n for n in self if n.id == network_id)
  694. return next(network, None)
  695. def list(self, limit=None, marker=None):
  696. networks = [OpenStackNetwork(self.provider, network)
  697. for network in self.provider.neutron.list_networks()
  698. .get('networks') if network]
  699. return ClientPagedResultList(self.provider, networks,
  700. limit=limit, marker=marker)
  701. def find(self, **kwargs):
  702. label = kwargs.pop('label', None)
  703. # All kwargs should have been popped at this time.
  704. if len(kwargs) > 0:
  705. raise TypeError("Unrecognised parameters for search: %s."
  706. " Supported attributes: %s" % (kwargs, 'label'))
  707. log.debug("Searching for OpenStack Network with label: %s", label)
  708. networks = [OpenStackNetwork(self.provider, network)
  709. for network in self.provider.neutron.list_networks(
  710. name=label)
  711. .get('networks') if network]
  712. return ClientPagedResultList(self.provider, networks)
  713. def create(self, label, cidr_block):
  714. log.debug("Creating OpenStack Network with the params: "
  715. "[label: %s Cinder Block: %s]", label, cidr_block)
  716. OpenStackNetwork.assert_valid_resource_label(label)
  717. net_info = {'name': label or ""}
  718. network = self.provider.neutron.create_network({'network': net_info})
  719. cb_net = OpenStackNetwork(self.provider, network.get('network'))
  720. if label:
  721. cb_net.label = label
  722. return cb_net
  723. class OpenStackSubnetService(BaseSubnetService):
  724. def __init__(self, provider):
  725. super(OpenStackSubnetService, self).__init__(provider)
  726. def get(self, subnet_id):
  727. log.debug("Getting OpenStack Subnet with the id: %s", subnet_id)
  728. subnet = (s for s in self if s.id == subnet_id)
  729. return next(subnet, None)
  730. def list(self, network=None, limit=None, marker=None):
  731. if network:
  732. network_id = (network.id if isinstance(network, OpenStackNetwork)
  733. else network)
  734. subnets = [subnet for subnet in self if network_id ==
  735. subnet.network_id]
  736. else:
  737. subnets = [OpenStackSubnet(self.provider, subnet) for subnet in
  738. self.provider.neutron.list_subnets().get('subnets', [])]
  739. return ClientPagedResultList(self.provider, subnets,
  740. limit=limit, marker=marker)
  741. def create(self, label, network, cidr_block, zone):
  742. """zone param is ignored."""
  743. log.debug("Creating OpenStack Subnet with the params: "
  744. "[Label: %s Network: %s Cinder Block: %s Zone: -ignored-]",
  745. label, network, cidr_block)
  746. OpenStackSubnet.assert_valid_resource_label(label)
  747. network_id = (network.id if isinstance(network, OpenStackNetwork)
  748. else network)
  749. subnet_info = {'name': label, 'network_id': network_id,
  750. 'cidr': cidr_block, 'ip_version': 4}
  751. subnet = (self.provider.neutron.create_subnet({'subnet': subnet_info})
  752. .get('subnet'))
  753. cb_subnet = OpenStackSubnet(self.provider, subnet)
  754. return cb_subnet
  755. def get_or_create_default(self, zone):
  756. """
  757. Subnet zone is not supported by OpenStack and is thus ignored.
  758. """
  759. try:
  760. sn = self.find(label=OpenStackSubnet.CB_DEFAULT_SUBNET_LABEL)
  761. if sn:
  762. return sn[0]
  763. # No default subnet look for default network, then create subnet
  764. net = self.provider.networking.networks.get_or_create_default()
  765. sn = self.provider.networking.subnets.create(
  766. label=OpenStackSubnet.CB_DEFAULT_SUBNET_LABEL,
  767. cidr_block=OpenStackSubnet.CB_DEFAULT_SUBNET_IPV4RANGE,
  768. network=net)
  769. router = self.provider.networking.routers.get_or_create_default(
  770. net)
  771. router.attach_subnet(sn)
  772. gateway = net.gateways.get_or_create_inet_gateway()
  773. router.attach_gateway(gateway)
  774. return sn
  775. except NeutronClientException:
  776. return None
  777. def delete(self, subnet):
  778. log.debug("Deleting subnet: %s", subnet)
  779. subnet_id = (subnet.id if isinstance(subnet, OpenStackSubnet)
  780. else subnet)
  781. self.provider.neutron.delete_subnet(subnet_id)
  782. # Adhere to the interface docs
  783. if subnet_id not in self:
  784. return True
  785. return False
  786. class OpenStackRouterService(BaseRouterService):
  787. def __init__(self, provider):
  788. super(OpenStackRouterService, self).__init__(provider)
  789. def get(self, router_id):
  790. log.debug("Getting OpenStack Router with the id: %s", router_id)
  791. router = (r for r in self if r.id == router_id)
  792. return next(router, None)
  793. def list(self, limit=None, marker=None):
  794. routers = self.provider.neutron.list_routers().get('routers')
  795. os_routers = [OpenStackRouter(self.provider, r) for r in routers]
  796. return ClientPagedResultList(self.provider, os_routers, limit=limit,
  797. marker=marker)
  798. def find(self, **kwargs):
  799. obj_list = self
  800. filters = ['label']
  801. matches = cb_helpers.generic_find(filters, kwargs, obj_list)
  802. return ClientPagedResultList(self._provider, list(matches))
  803. def create(self, label, network):
  804. """
  805. ``network`` is not used by OpenStack.
  806. However, the API seems to indicate it is a (required) param?!
  807. https://developer.openstack.org/api-ref/networking/v2/
  808. ?expanded=delete-router-detail,create-router-detail#create-router
  809. """
  810. log.debug("Creating OpenStack Router with the label: %s", label)
  811. OpenStackRouter.assert_valid_resource_label(label)
  812. body = {'router': {'name': label}} if label else None
  813. router = self.provider.neutron.create_router(body)
  814. return OpenStackRouter(self.provider, router.get('router'))