Migrate Node Network Attach Functionality Into esisdk

esi/lease/v1/node.py

@@ -41,6 +41,9 @@ class Node(resource.Resource):
  owner = resource.Body("owner")
  lessee = resource.Body("lessee")
  provision_state = resource.Body("provision_state")
+ target_provision_state = resource.Body("target_provision_state")
+ power_state = resource.Body("power_state")
+ target_power_state = resource.Body("target_power_state")
  maintenance = resource.Body("maintenance")
  offer_uuid = resource.Body("offer_uuid")
  lease_uuid = resource.Body("lease_uuid")

esi/lib/nodes.py

@@ -13,9 +13,14 @@
 
 import concurrent.futures
 
+from openstack import exceptions
+
 from esi.lib import networks
 
 
+OPENSTACK_IRONIC_API_VERSION = '1.69'
+
+
 def node_and_port_list(connection, filter_node=None):
  """Get lists baremetal nodes and ports
 
@@ -138,3 +143,107 @@ def network_list(connection, filter_node=None, filter_network=None):
  })
 
  return data
+
+
+def network_attach(connection, node, attach_info):
+ """Attaches a node's bare metal port to a network port
+
+ :param connection: An OpenStack connection
+ :type connection: :class:`~openstack.connection.Connection`
+ :param node: The name or ID of a node
+ :param attach_info: A dictionary. Possible entrys are:
+ * 'network': <network name or ID>
+ * 'port': <port name or ID> (The network port to attach)
+ * 'trunk': <trunk name or ID>
+ * 'mac_address': <MAC addresses> (The MAC address of the bare metal port to attach)
+
+ :returns: a dictionary with the resulting node and network information
+ {
+ 'node': openstack.baremetal.v1.node.Node,
+ 'ports': [openstack.network.v2.port.Port]
+ 'networks': [openstack.network.v2.network.Network]
+ }
+ """
+
+ network = attach_info.get('network')
+ port = attach_info.get('port')
+ trunk = attach_info.get('trunk')
+ mac_address = attach_info.get('mac_address')
+
+ if (network and port) or (network and trunk) or (port and trunk):
+ raise exceptions.InvalidRequest('Specify only one of network, port, or trunk')
+ if not network and not port and not trunk:
+ raise exceptions.InvalidRequest('You must specify either network, port, or trunk')
+
+ if network:
+ parent_network = connection.network.find_network(network, ignore_missing=False)
+ network_port = None
+ elif port:
+ network_port = connection.network.find_port(port, ignore_missing=False)
+ elif trunk:
+ trunk_network = connection.network.find_trunk(trunk, ignore_missing=False)
+ network_port = None
+
+ with concurrent.futures.ThreadPoolExecutor() as executor:
+ f1 = executor.submit(connection.baremetal.get_node, node)
+ f2 = executor.submit(connection.session.get_endpoint,
+ service_type='baremetal',
+ service_name='ironic',
+ interface='public')
+ node = f1.result()
+ baremetal_endpoint = f2.result()
+
+ if mac_address:
+ baremetal_ports = list(connection.baremetal.ports(details=True, address=mac_address))
+ if len(baremetal_ports) == 0:
+ raise exceptions.ResourceFailure('MAC address {0} does not exist on node {1}'.format(mac_address, node.name))
+ else:
+ baremetal_ports = connection.baremetal.ports(details=True, node=node.id)
+ has_free_port = False
+ for bp in baremetal_ports:
+ if 'tenant_vif_port_id' not in bp.internal_info:
+ has_free_port = True
+ break
+
+ if not has_free_port:
+ raise exceptions.ResourceFailure('Node {0} has no free ports'.format(node.name))
+
+ if network:
+ port_name = 'esi-{0}-{1}'.format(node.name, parent_network.name)
+ network_ports = list(connection.network.ports(name=port_name, status='DOWN'))
+ if len(network_ports) > 0:
+ network_port = network_ports[0]
+ else:
+ network_port = connection.network.create_port(name=port_name,
+ network_id=parent_network.id,
+ device_owner='baremetal:none')
+ elif trunk:
+ network_port = connection.network.find_port(trunk_network.port_id, ignore_missing=False)
+
+ data = {'id': network_port.id}
+ if mac_address:
+ data['port_uuid'] = baremetal_ports[0].id
+
+ # TODO(ajamias) There should be a function in openstacksdk that specifies
+ # a bare metal port's MAC address to attach a network port to
+ connection.session.post(
+ '{0}/v1/nodes/{1}/vifs'.format(baremetal_endpoint, node.id),
+ headers={'X-OpenStack-Ironic-API-Version': OPENSTACK_IRONIC_API_VERSION},
+ json=data)
+ network_port = connection.network.find_port(network_port.id, ignore_missing=False)
+
+ networks_dict = {}
+ if network:
+ networks_dict[parent_network.id] = parent_network
+ elif trunk:
+ networks_dict[trunk_network.id] = trunk_network
+ parent_network, trunk_networks, trunk_ports, _ \
+ = networks.get_networks_from_port(connection,
+ network_port,
+ networks_dict=networks_dict)
+
+ return {
+ 'node': node,
+ 'ports': [network_port] + trunk_ports,
+ 'networks': [parent_network] + trunk_networks
+ }