Zhenguo Niu 8b4e3b1f51 Add support to specify userdata when creating a server
Change-Id: I283253a778958de5bfc4417fe8cd8b79927d63cb
2017-04-12 20:23:31 +08:00

105 lines
3.6 KiB
Python

# Copyright 2016 Huawei, Inc. All rights reserved.
#
# 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 base64
from oslo_utils import encodeutils
import six
from moganclient.common import base
class Server(base.Resource):
pass
class ServerManager(base.ManagerWithFind):
resource_class = Server
def create(self, name, image_uuid, flavor_uuid, networks,
description=None, availability_zone=None, extra=None,
userdata=None, min_count=None, max_count=None):
url = '/instances'
data = {
'name': name,
'image_uuid': image_uuid,
'instance_type_uuid': flavor_uuid,
'networks': networks
}
if userdata is not None:
if hasattr(userdata, 'read'):
userdata = userdata.read()
# NOTE(melwitt): Text file data is converted to bytes prior to
# base64 encoding. The utf-8 encoding will fail for binary files.
if six.PY3:
try:
userdata = userdata.encode("utf-8")
except AttributeError:
# In python 3, 'bytes' object has no attribute 'encode'
pass
else:
try:
userdata = encodeutils.safe_encode(userdata)
except UnicodeDecodeError:
pass
userdata_b64 = base64.b64encode(userdata).decode('utf-8')
data["user_data"] = userdata_b64
if availability_zone is not None:
data['availability_zone'] = availability_zone
if description is not None:
data['description'] = description
if extra is not None:
data['extra'] = extra
if min_count is not None:
data['min_count'] = min_count
if max_count is not None:
data['max_count'] = max_count
return self._create(url, data=data)
def delete(self, server_id):
url = '/instances/%s' % base.getid(server_id)
return self._delete(url)
def get(self, server_id):
url = '/instances/%s' % base.getid(server_id)
return self._get(url)
def list(self, detailed=False, all_projects=False):
url = '/instances/detail' if detailed else '/instances'
if all_projects:
url = '%s?%s' % (url, 'all_tenants=True')
return self._list(url, response_key='instances')
def update(self, server_id, updates):
url = '/instances/%s' % base.getid(server_id)
return self._update(url, data=updates)
def set_power_state(self, server_id, power_state):
url = '/instances/%s/states/power' % base.getid(server_id)
return self._update_all(url, data={'target': power_state})
def set_lock_state(self, server_id, lock_state):
url = '/instances/%s/states/lock' % base.getid(server_id)
return self._update_all(url, data={'target': lock_state})
def get_network_info(self, server_id):
url = '/instances/%s/networks' % base.getid(server_id)
return self._get(url, response_key='ports')