/usr/lib/python2.7/dist-packages/cs/client.py is in python-cs 2.0.0-1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
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 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 | #! /usr/bin/env python
import base64
import hashlib
import hmac
import os
import sys
try:
from configparser import ConfigParser
except ImportError: # python 2
from ConfigParser import ConfigParser
try:
from urllib.parse import quote
except ImportError: # python 2
from urllib import quote
import requests
PY2 = sys.version_info < (3, 0)
if PY2:
text_type = unicode # noqa
string_type = basestring # noqa
integer_types = int, long # noqa
else:
text_type = str
string_type = str
integer_types = int
if sys.version_info >= (3, 5):
try:
from cs.async import AIOCloudStack # noqa
except ImportError:
pass
def cs_encode(value):
"""
Try to behave like cloudstack, which uses
java.net.URLEncoder.encode(stuff).replace('+', '%20').
"""
if isinstance(value, int):
value = str(value)
elif PY2 and isinstance(value, text_type):
value = value.encode('utf-8')
return quote(value, safe=".-*_")
def transform(params):
for key, value in list(params.items()):
if value is None or value == "":
params.pop(key)
continue
if isinstance(value, string_type):
continue
elif isinstance(value, integer_types):
params[key] = text_type(value)
elif isinstance(value, (list, tuple, set)):
if not value:
params.pop(key)
else:
if isinstance(value, set):
value = list(value)
if not isinstance(value[0], dict):
params[key] = ",".join(value)
else:
params.pop(key)
for index, val in enumerate(value):
for k, v in val.items():
params["%s[%d].%s" % (key, index, k)] = v
else:
raise ValueError(type(value))
return params
class CloudStackException(Exception):
pass
class Unauthorized(CloudStackException):
pass
class CloudStack(object):
def __init__(self, endpoint, key, secret, timeout=10, method='get',
verify=True, cert=None, name=None):
self.endpoint = endpoint
self.key = key
self.secret = secret
self.timeout = int(timeout)
self.method = method.lower()
self.verify = verify
self.cert = cert
self.name = name
def __repr__(self):
return '<CloudStack: {0}>'.format(self.name or self.endpoint)
def __getattr__(self, command):
def handler(**kwargs):
return self._request(command, **kwargs)
return handler
def _prepare_request(self, command, json, opcode_name, fetch_list,
**kwargs):
kwargs.update({
'apiKey': self.key,
opcode_name: command,
})
if json:
kwargs['response'] = 'json'
if 'page' in kwargs or fetch_list:
kwargs.setdefault('pagesize', 500)
kwarg = 'params' if self.method == 'get' else 'data'
return kwarg, kwargs
def _request(self, command, json=True, opcode_name='command',
fetch_list=False, **kwargs):
kwarg, kwargs = self._prepare_request(command, json, opcode_name,
fetch_list, **kwargs)
done = False
final_data = []
page = 1
while not done:
if fetch_list:
kwargs['page'] = page
kwargs = transform(kwargs)
kwargs.pop('signature', None)
kwargs['signature'] = self._sign(kwargs)
response = getattr(requests, self.method)(self.endpoint,
timeout=self.timeout,
verify=self.verify,
cert=self.cert,
**{kwarg: kwargs})
try:
data = response.json()
except ValueError as e:
msg = "Make sure endpoint URL '%s' is correct." % self.endpoint
raise CloudStackException(
"HTTP {0} response from CloudStack".format(
response.status_code), response, "%s. " % str(e) + msg)
[key] = data.keys()
data = data[key]
if response.status_code != 200:
raise CloudStackException(
"HTTP {0} response from CloudStack".format(
response.status_code), response, data)
if fetch_list:
try:
[key] = [k for k in data.keys() if k != 'count']
except ValueError:
done = True
else:
final_data.extend(data[key])
page += 1
else:
final_data = data
done = True
return final_data
def _sign(self, data):
"""
Computes a signature string according to the CloudStack
signature method (hmac/sha1).
"""
params = "&".join(sorted([
"=".join((key, cs_encode(value)))
for key, value in data.items()
])).lower()
digest = hmac.new(
self.secret.encode('utf-8'),
msg=params.encode('utf-8'),
digestmod=hashlib.sha1).digest()
return base64.b64encode(digest).decode('utf-8').strip()
def read_config(ini_group=None):
if not ini_group:
ini_group = os.environ.get('CLOUDSTACK_REGION', 'cloudstack')
# Try env vars first
os.environ.setdefault('CLOUDSTACK_METHOD', 'get')
os.environ.setdefault('CLOUDSTACK_TIMEOUT', '10')
keys = ['endpoint', 'key', 'secret', 'method', 'timeout']
env_conf = {}
for key in keys:
if 'CLOUDSTACK_{0}'.format(key.upper()) not in os.environ:
break
else:
env_conf[key] = os.environ['CLOUDSTACK_{0}'.format(key.upper())]
else:
env_conf['verify'] = os.environ.get('CLOUDSTACK_VERIFY', True)
env_conf['cert'] = os.environ.get('CLOUDSTACK_CERT', None)
env_conf['name'] = None
return env_conf
# Config file: $PWD/cloudstack.ini or $HOME/.cloudstack.ini
# Last read wins in configparser
paths = (
os.path.join(os.path.expanduser('~'), '.cloudstack.ini'),
os.path.join(os.getcwd(), 'cloudstack.ini'),
)
# Look at CLOUDSTACK_CONFIG first if present
if 'CLOUDSTACK_CONFIG' in os.environ:
paths += (os.path.expanduser(os.environ['CLOUDSTACK_CONFIG']),)
if not any([os.path.exists(c) for c in paths]):
raise SystemExit("Config file not found. Tried {0}".format(
", ".join(paths)))
conf = ConfigParser()
conf.read(paths)
try:
cs_conf = conf[ini_group]
except AttributeError: # python 2
cs_conf = dict(conf.items(ini_group))
cs_conf['name'] = ini_group
return cs_conf
|