File: operation.py

package info (click to toggle)
python-pylxd 2.2.10-1
  • links: PTS, VCS
  • area: main
  • in suites: bookworm, bullseye
  • size: 820 kB
  • sloc: python: 7,258; sh: 104; makefile: 21
file content (91 lines) | stat: -rw-r--r-- 3,380 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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
# Copyright (c) 2016 Canonical Ltd
#
#    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 warnings
import os

from pylxd import exceptions
from six.moves.urllib import parse


# Global used to record which warnings have been issues already for unknown
# attributes.
_seen_attribute_warnings = set()


class Operation(object):
    """An LXD operation.

    If the LXD server sends attributes that this version of pylxd is unaware of
    then a warning is printed.  By default the warning is issued ONCE and then
    supressed for every subsequent attempted setting.  The warnings can be
    completely suppressed by setting the environment variable PYLXD_WARNINGS to
    'none', or always displayed by setting the PYLXD_WARNINGS variable to
    'always'.
    """

    __slots__ = [
        '_client',
        'class', 'created_at', 'description', 'err', 'id', 'may_cancel',
        'metadata', 'resources', 'status', 'status_code', 'updated_at']

    @classmethod
    def wait_for_operation(cls, client, operation_id):
        """Get an operation and wait for it to complete."""
        operation = cls.get(client, operation_id)
        operation.wait()
        return cls.get(client, operation.id)

    @classmethod
    def extract_operation_id(cls, s):
        return os.path.split(parse.urlparse(s).path)[-1]

    @classmethod
    def get(cls, client, operation_id):
        """Get an operation."""
        operation_id = cls.extract_operation_id(operation_id)
        response = client.api.operations[operation_id].get()
        return cls(_client=client, **response.json()['metadata'])

    def __init__(self, **kwargs):
        super(Operation, self).__init__()
        for key, value in kwargs.items():
            try:
                setattr(self, key, value)
            except AttributeError:
                # ignore attributes we don't know about -- prevent breakage
                # in the future if new attributes are added.
                global _seen_attribute_warnings
                env = os.environ.get('PYLXD_WARNINGS', '').lower()
                if env != 'always' and key in _seen_attribute_warnings:
                    continue
                _seen_attribute_warnings.add(key)
                if env == 'none':
                    continue
                warnings.warn(
                    'Attempted to set unknown attribute "{}" '
                    'on instance of "{}"'
                    .format(key, self.__class__.__name__))
                pass

    def wait(self):
        """Wait for the operation to complete and return."""
        response = self._client.api.operations[self.id].wait.get()

        try:
            if response.json()['metadata']['status'] == 'Failure':
                raise exceptions.LXDAPIException(response)
        except KeyError:
            # Support for legacy LXD
            pass