services.py 38 KB

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