skyline-apiserver/libs/skyline-apiserver/skyline_apiserver/api/v1/utils.py
Gao Hanxiang 4935be6f47 refactor: Move skyline-apiserver to libs
1. Move skyline-apiserver to libs
2. Modify Makefile
3. Rename the skyline-apiserver.yaml.sample file to skyline.yaml.sample

At the top level, we use "skyline" as the namespace, using the idea of
monorepo to manage the project. At the top level, only some common
configuration files (mypy.ini, black.conf, isort.conf, etc.) and
common tools are included.

In the `libs` directory, there are three core libraries,
skyline-console, skyline-apiserver, skyline-nginx, and some dependent
libraries.
There are cross-imports between these libraries,
for example: skyline-nginx requires skyline-apiserver;
Both skyline-nginx and skyline-apiserver require skyline-config.

Therefore, skyline-apiserver should not be placed
at the top level, it is also suitable for management as a library.


Change-Id: Ie2f1f4bdfbc2e985ec4327705eecaae3181f5b50
2021-07-26 03:48:54 +00:00

264 lines
7.8 KiB
Python

# Copyright 2021 99cloud
#
# 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.
from __future__ import annotations
from typing import Any, Dict, List
class APIResourceWrapper(object):
"""Simple wrapper for api objects."""
_attrs: List[str] = []
_attrs_mapping: Dict[str, Any] = {}
_apiresource: Any = None
def __init__(self, apiresource: Any) -> None:
self._apiresource = apiresource
def __getattribute__(self, attr: str) -> Any:
try:
return object.__getattribute__(self, attr)
except AttributeError:
raise
def __repr__(self) -> str:
return "<%s: %s>" % (
self.__class__.__name__,
self.to_dict(),
)
def _get_value(self, key: str) -> Any:
if isinstance(self._apiresource, dict):
value = self._apiresource.get(key, None)
else:
value = getattr(self._apiresource, key, None)
return value
def to_dict(self) -> Dict[str, Any]:
obj: Dict[str, Any] = {}
for key, value in self._attrs_mapping.items():
obj[key] = self._get_value(value)
return obj
class Server(APIResourceWrapper):
_attrs_mapping = {
"id": "id",
"name": "name",
"project_id": "tenant_id",
"project_name": "project_name",
"host": "OS-EXT-SRV-ATTR:host",
"hostname": "OS-EXT-SRV-ATTR:hostname",
"image": "image",
"image_name": "image_name",
"image_os_distro": "image_os_distro",
"fixed_addresses": "fixed_addresses",
"floating_addresses": "floating_addresses",
"flavor": "flavor",
"flavor_info": "flavor",
"status": "status",
"locked": "locked",
"created_at": "created",
"updated_at": "updated",
"task_state": "OS-EXT-STS:task_state",
"vm_state": "OS-EXT-STS:vm_state",
"power_state": "OS-EXT-STS:power_state",
"volumes_attached": "os-extended-volumes:volumes_attached",
"root_device_name": "OS-EXT-SRV-ATTR:root_device_name",
"metadata": "metadata",
}
def _format_flavor(self, obj: Dict[str, Any], key: str, return_key: str) -> None:
flavor = self._get_value(key)
obj[return_key] = flavor["original_name"] if flavor else None
def _format_addr(
self,
obj: Dict[str, Any],
key: str,
address_type: str,
return_key: str,
) -> None:
addresses = []
for _, v in self._get_value(key).items():
addresses.extend(v)
_addresses = []
for address in addresses:
if address.get("OS-EXT-IPS:type") == address_type:
_addresses.append(address.get("addr"))
obj[return_key] = _addresses
def _format_image(self, obj: Dict[str, Any], key: str, return_key: str) -> None:
image = self._get_value(key)
obj[return_key] = image["id"] if image else None
def to_dict(self) -> Dict[str, Any]:
obj: Dict[str, Any] = {}
for key, value in self._attrs_mapping.items():
if key == "flavor":
self._format_flavor(obj, "flavor", key)
continue
if value == "fixed_addresses":
self._format_addr(obj, "addresses", "fixed", key)
continue
if value == "floating_addresses":
self._format_addr(obj, "addresses", "floating", key)
continue
if value == "image":
self._format_image(obj, "image", key)
continue
obj[key] = self._get_value(value)
return obj
class Volume(APIResourceWrapper):
_attrs_mapping = {
"id": "id",
"name": "name",
"project_id": "os-vol-tenant-attr:tenant_id",
"project_name": "project_name",
"host": "os-vol-host-attr:host",
"snapshot_id": "snapshot_id",
"source_volid": "source_volid",
"size": "size",
"status": "status",
"volume_type": "volume_type",
"attachments": "attachments",
"encrypted": "encrypted",
"bootable": "bootable",
"multiattach": "multiattach",
"availability_zone": "availability_zone",
"created_at": "created_at",
"volume_image_metadata": "volume_image_metadata",
}
class VolumeSnapshot(APIResourceWrapper):
_attrs_mapping = {
"id": "id",
"name": "name",
"project_id": "os-extended-snapshot-attributes:project_id",
"project_name": "project_name",
"size": "size",
"status": "status",
"volume_id": "volume_id",
"volume_name": "volume_name",
"created_at": "created_at",
"metadata": "metadata",
}
class Flavor(APIResourceWrapper):
_attrs_mapping = {
"id": "id",
"name": "name",
"vcpus": "vcpus",
"ram": "ram",
"disk": "disk",
"is_public": "os-flavor-access:is_public",
"is_in_use": "is_in_use",
}
class Service(APIResourceWrapper):
_attrs_mapping = {
"id": "id",
"binary": "binary",
"disabled_reason": "disabled_reason",
"host": "host",
"state": "state",
"status": "status",
"updated_at": "updated_at",
"forced_down": "forced_down",
"zone": "zone",
}
class Image(APIResourceWrapper):
_attrs_mapping = {
"id": "id",
"name": "name",
"os_distro": "os_distro",
"block_device_mapping": "block_device_mapping",
"image_type": "image_type",
"status": "status",
"visibility": "visibility",
"disk_format": "disk_format",
"size": "size",
"created_at": "created_at",
}
class Project(APIResourceWrapper):
_attrs_mapping = {
"id": "id",
"name": "name",
}
class Port(APIResourceWrapper):
_attrs_mapping = {
"id": "id",
"name": "name",
"ipv4": "fixed_ips",
"ipv6": "fixed_ips",
"mac_address": "mac_address",
"device_owner": "device_owner",
"device_id": "device_id",
"server_name": "server_name",
"status": "status",
"created_at": "created_at",
"project_id": "project_id",
"network_id": "network_id",
"binding_vnic_type": "binding:vnic_type",
"description": "description",
"port_security_enabled": "port_security_enabled",
"qos_policy_id": "qos_policy_id",
"fixed_ips": "fixed_ips",
}
def _format_ip(
self,
obj: Dict[str, Any],
key: str,
return_key: str,
) -> None:
ips = []
fixed_ips = self._get_value(key)
if fixed_ips:
for ip in fixed_ips:
if return_key == "ipv4" and ":" not in ip["ip_address"]:
ips.append(ip["ip_address"])
elif return_key == "ipv6" and "." not in ip["ip_address"]:
ips.append(ip["ip_address"])
obj[return_key] = ips
def to_dict(self) -> Dict[str, Any]:
obj: Dict[str, Any] = {}
for key, value in self._attrs_mapping.items():
if key == "ipv4":
self._format_ip(obj, "fixed_ips", key)
continue
if key == "ipv6":
self._format_ip(obj, "fixed_ips", key)
continue
obj[key] = self._get_value(value)
return obj