File: plugin.py

package info (click to toggle)
python-vmmsclient 0.0.3
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 136 kB
  • sloc: python: 559; makefile: 18; sh: 10
file content (66 lines) | stat: -rw-r--r-- 2,041 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
# Copyright (c) 2025 Thomas Goirand
#
# 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.

try:
    from unittest import mock
except ImportError:
    import mock
from osc_lib.cli import parseractions
from osc_lib import utils

from vmmsclient.v2.client import AddVMCommand, ListVMsCommand, RemoveVMCommand, ShowVMCommand, SetVMCommand, UnsetVMCommand

DEFAULT_API_VERSION = '2'
API_VERSION_OPTION = 'os_vmms_api_version'
API_NAME = 'vmms'
API_VERSIONS = {
    '2': 'vmmsclient.v2.client.Client',
}

__all__ = [
    'AddVMCommand',
    'ListVMsCommand', 
    'RemoveVMCommand',
    'ShowVMCommand',
    'SetVMCommand',
    'UnsetVMCommand'
]

def make_client(instance):
    """Returns a client to the ClientManager."""
    # Safely get the configuration value
    config = instance.get_configuration()
    requested_api_version = config.get(API_VERSION_OPTION, DEFAULT_API_VERSION)
    
    client_class = utils.get_client_class(API_NAME, requested_api_version, API_VERSIONS)
    
    kwargs = {
        'session': instance.session,
        'service_type': 'vmms',
        'region_name': instance.region_name,
        'interface': instance.interface,
    }
    
    client = client_class(**kwargs)
    return client

def build_option_parser(parser):
    """Hook to add global options."""
    parser.add_argument(
        '--os-vmms-api-version',
        metavar='<vmms-api-version>',
        default=utils.env('OS_VMMS_API_VERSION') or DEFAULT_API_VERSION,
        help='VM Migration Scheduler API version (default: %(default)s)')
    return parser