# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.
import logging
from django.conf import settings
from django.utils.translation import ugettext_lazy as _
import six
from horizon import exceptions
from openstack_dashboard import api
LOG = logging.getLogger(__name__)
[docs]def flavor_list(request):
    """Utility method to retrieve a list of flavors."""
    try:
        return api.nova.flavor_list(request)
    except Exception:
        exceptions.handle(request,
                          _('Unable to retrieve instance flavors.'))
        return []
 
[docs]def sort_flavor_list(request, flavors):
    """Utility method to sort a list of flavors.
        By default, returns the available flavors, sorted by RAM
        usage (ascending). Override these behaviours with a
        CREATE_INSTANCE_FLAVOR_SORT dict
        in local_settings.py.
    """
    def get_key(flavor, sort_key):
        try:
            return getattr(flavor, sort_key)
        except AttributeError:
            LOG.warning('Could not find sort key "%s". Using the default '
                        '"ram" instead.', sort_key)
            return getattr(flavor, 'ram')
    try:
        flavor_sort = getattr(settings, 'CREATE_INSTANCE_FLAVOR_SORT', {})
        sort_key = flavor_sort.get('key', 'ram')
        rev = flavor_sort.get('reverse', False)
        if not callable(sort_key):
            key = lambda flavor: get_key(flavor, sort_key)
        else:
            key = sort_key
        flavor_list = [(flavor.id, '%s' % flavor.name)
                       for flavor in sorted(flavors, key=key, reverse=rev)]
        return flavor_list
    except Exception:
        exceptions.handle(request,
                          _('Unable to sort instance flavors.'))
        return []
 
[docs]def availability_zone_list(request):
    """Utility method to retrieve a list of availability zones."""
    try:
        return api.nova.availability_zone_list(request)
    except Exception:
        exceptions.handle(request,
                          _('Unable to retrieve Nova availability zones.'))
        return []
 
[docs]def server_group_list(request):
    """Utility method to retrieve a list of server groups."""
    try:
        return api.nova.server_group_list(request)
    except Exception:
        exceptions.handle(request,
                          _('Unable to retrieve Nova server groups.'))
        return []
 
[docs]def network_field_data(request, include_empty_option=False):
    """Returns a list of tuples of all networks.
    Generates a list of networks available to the user (request). And returns
    a list of (id, name) tuples.
    :param request: django http request object
    :param include_empty_option: flag to include a empty tuple in the front of
         the list
    :return: list of (id, name) tuples
    """
    tenant_id = request.user.tenant_id
    networks = []
    if api.base.is_service_enabled(request, 'network'):
        try:
            networks = api.neutron.network_list_for_tenant(request, tenant_id)
            networks = [(n.id, n.name_or_id) for n in networks if n['subnets']]
            networks.sort(key=lambda obj: obj[1])
        except Exception as e:
            msg = _('Failed to get network list {0}').format(six.text_type(e))
            exceptions.handle(request, msg)
    if not networks:
        if include_empty_option:
            return [("", _("No networks available")), ]
        return []
    if include_empty_option:
        return [("", _("Select Network")), ] + networks
    return networks
 
[docs]def keypair_field_data(request, include_empty_option=False):
    """Returns a list of tuples of all keypairs.
    Generates a list of keypairs available to the user (request). And returns
    a list of (id, name) tuples.
    :param request: django http request object
    :param include_empty_option: flag to include a empty tuple in the front of
        the list
    :return: list of (id, name) tuples
    """
    keypair_list = []
    try:
        keypairs = api.nova.keypair_list(request)
        keypair_list = [(kp.name, kp.name) for kp in keypairs]
    except Exception:
        exceptions.handle(request, _('Unable to retrieve key pairs.'))
    if not keypair_list:
        if include_empty_option:
            return [("", _("No key pairs available")), ]
        return []
    if include_empty_option:
        return [("", _("Select a key pair")), ] + keypair_list
    return keypair_list
 
[docs]def flavor_field_data(request, include_empty_option=False):
    """Returns a list of tuples of all image flavors.
    Generates a list of image flavors available. And returns a list of
    (id, name) tuples.
    :param request: django http request object
    :param include_empty_option: flag to include a empty tuple in the front of
        the list
    :return: list of (id, name) tuples
    """
    flavors = flavor_list(request)
    if flavors:
        flavors_list = sort_flavor_list(request, flavors)
        if include_empty_option:
            return [("", _("Select Flavor")), ] + flavors_list
        return flavors_list
    if include_empty_option:
        return [("", _("No flavors available")), ]
    return []
 
[docs]def port_field_data(request):
    """Returns a list of tuples of all ports available for the tenant.
    Generates a list of ports that have no device_owner based on the networks
    available to the tenant doing the request.
    :param request: django http request object
    :return: list of (id, name) tuples
    """
    def add_more_info_port_name(port):
        # add more info to the port for the display
        return "{} ({})".format(port.name_or_id,
                                ",".join([ip['ip_address']
                                          for ip in port['fixed_ips']]))
    ports = []
    if api.base.is_service_enabled(request, 'network'):
        network_list = api.neutron.network_list_for_tenant(
            request, request.user.tenant_id)
        for network in network_list:
            ports.extend(
                [(port.id, add_more_info_port_name(port))
                 for port in api.neutron.port_list(request,
                                                   network_id=network.id)
                 if port.device_owner == ''])
    ports.sort(key=lambda obj: obj[1])
    return ports
 
[docs]def server_group_field_data(request):
    """Returns a list of tuples of all server groups.
    Generates a list of server groups available. And returns a list of
    (id, name) tuples.
    :param request: django http request object
    :return: list of (id, name) tuples
    """
    server_groups = server_group_list(request)
    if server_groups:
        server_groups_list = [(sg.id, sg.name) for sg in server_groups]
        server_groups_list.sort(key=lambda obj: obj[1])
        return [("", _("Select Server Group")), ] + server_groups_list
    return [("", _("No server groups available")), ]