def update_ssh_key(uuid):
guest = Guest()
guest.uuid = uuid
guest.get_by('uuid')
# 不支持更新离线虚拟机的 SSH-KEY
if guest.status != GuestState.running.value:
return
os_template_image = OSTemplateImage()
os_template_profile = OSTemplateProfile()
os_template_image.id = guest.os_template_image_id
os_template_image.get()
os_template_profile.id = os_template_image.os_template_profile_id
os_template_profile.get()
# 不支持更新 Windows 虚拟机的 SSH-KEY
if os_template_profile.os_type == 'windows':
return
rows, _ = SSHKeyGuestMapping.get_by_filter(
filter_str=':'.join(['guest_uuid', 'eq', uuid]))
ssh_keys_id = list()
for row in rows:
ssh_keys_id.append(row['ssh_key_id'].__str__())
ssh_keys = list()
if ssh_keys_id.__len__() > 0:
rows, _ = SSHKey.get_by_filter(
filter_str=':'.join(['id', 'in', ','.join(ssh_keys_id)]))
for row in rows:
ssh_keys.append(row['public_key'])
else:
ssh_keys.append('')
message = {
'_object': 'guest',
'uuid': uuid,
'node_id': guest.node_id,
'action': 'update_ssh_key',
'ssh_keys': ssh_keys,
'os_type': os_template_profile.os_type,
'passback_parameters': {
'uuid': uuid,
'ssh_keys': ssh_keys,
'os_type': os_template_profile.os_type
}
}
Utils.emit_instruction(message=json.dumps(message, ensure_ascii=False))
def r_delete(ids):
ret = dict()
ret['state'] = ji.Common.exchange_state(20000)
config = Config()
config.id = 1
config.get()
# 取全部活着的 hosts
available_hosts = Host.get_available_hosts(nonrandom=None)
if available_hosts.__len__() == 0:
ret['state'] = ji.Common.exchange_state(50351)
return ret
chosen_host = available_hosts[0]
node_id = chosen_host['node_id']
os_template_image = OSTemplateImage()
# TODO: 加入对，是否有被 Guest 引用的判断
for _id in ids.split(','):
os_template_image.id = _id
os_template_image.get()
for _id in ids.split(','):
os_template_image.id = _id
os_template_image.get()
# 暂时不支持从计算节点上，删除公共镜像
if os_template_image.kind == OSTemplateImageKind.public.value:
os_template_image.delete()
continue
elif os_template_image.kind == OSTemplateImageKind.custom.value:
os_template_image.progress = 254
message = {
'_object': 'os_template_image',
'action': 'delete',
'storage_mode': config.storage_mode,
'dfs_volume': config.dfs_volume,
'template_path': os_template_image.path,
# uuid 这里没有实际意义，仅仅是为了迁就 JimV-C 的个命令格式
'uuid': None,
'node_id': node_id,
'os_template_image_id': os_template_image.id,
'passback_parameters': {'id': os_template_image.id}
}
Utils.emit_instruction(message=json.dumps(message))
os_template_image.update()
return ret
def r_detail(uuid):
disk = Disk()
disk.uuid = uuid
disk.get_by(field='uuid')
disk.wrap_device(dev_table=dev_table)
guest = None
os_template_image = None
config = Config()
config.id = 1
config.get()
if disk.sequence != -1:
guest = Guest()
guest.uuid = disk.guest_uuid
guest.get_by('uuid')
os_template_image = OSTemplateImage()
os_template_image.id = guest.os_template_image_id
os_template_image.get()
guest = guest.__dict__
os_template_image = os_template_image.__dict__
ret = dict()
ret['state'] = ji.Common.exchange_state(20000)
ret['data'] = {
'guest': guest,
'os_template_image': os_template_image,
'disk': disk.__dict__,
'config': config.__dict__
}
return ret
def r_update(_id):
os_template_image = OSTemplateImage()
args_rules = [
Rules.ID.value
]
if 'label' in request.json:
args_rules.append(
Rules.LABEL.value,
)
if 'description' in request.json:
args_rules.append(
Rules.DESCRIPTION.value,
)
if 'path' in request.json:
args_rules.append(
Rules.PATH.value,
)
if 'active' in request.json:
args_rules.append(
Rules.ACTIVE.value,
)
if 'logo' in request.json:
args_rules.append(
Rules.LOGO.value,
)
if 'os_template_profile_id' in request.json:
args_rules.append(
Rules.OS_TEMPLATE_PROFILE_ID_EXT.value,
)
if 'kind' in request.json:
args_rules.append(
Rules.OS_TEMPLATE_IMAGE_KIND.value,
)
if args_rules.__len__() < 2:
ret = dict()
ret['state'] = ji.Common.exchange_state(20000)
return ret
request.json['id'] = _id
try:
ji.Check.previewing(args_rules, request.json)
os_template_image.id = request.json.get('id')
os_template_image.get()
os_template_image.label = request.json.get('label', os_template_image.label)
os_template_image.description = request.json.get('description', os_template_image.description)
os_template_image.path = request.json.get('path', os_template_image.path)
os_template_image.active = request.json.get('active', os_template_image.active)
os_template_image.logo = request.json.get('logo', os_template_image.logo)
os_template_image.os_template_profile_id = \
request.json.get('os_template_profile_id', os_template_image.os_template_profile_id)
os_template_image.kind = request.json.get('kind', os_template_image.kind)
os_template_image.update()
os_template_image.get()
ret = dict()
ret['state'] = ji.Common.exchange_state(20000)
ret['data'] = os_template_image.__dict__
return ret
except ji.PreviewingError, e:
return json.loads(e.message)
def r_convert_to_os_template_image(snapshot_id, disk_uuid):
args_rules = [
Rules.SNAPSHOT_ID.value, Rules.DISK_UUID.value, Rules.LABEL.value
]
try:
ret = dict()
ret['state'] = ji.Common.exchange_state(20000)
ji.Check.previewing(
args_rules, {
'snapshot_id': snapshot_id,
'disk_uuid': disk_uuid,
'label': request.json.get('label')
})
rows, _ = SnapshotDiskMapping.get_by_filter(
filter_str=':'.join(['snapshot_id', 'eq', snapshot_id]))
disks_uuid = list()
for row in rows:
disks_uuid.append(row['disk_uuid'])
if disk_uuid not in disks_uuid:
ret['state'] = ji.Common.exchange_state(40401)
ret['state']['sub']['zh-cn'] = ''.join([
ret['state']['sub']['zh-cn'], u': 未在快照: ', snapshot_id,
u' 中找到磁盘：', disk_uuid
])
return ret
config = Config()
config.id = 1
config.get()
snapshot = Snapshot()
os_template_image = OSTemplateImage()
guest = Guest()
disk = Disk()
snapshot.snapshot_id = snapshot_id
snapshot.get_by('snapshot_id')
snapshot.progress = 252
guest.uuid = snapshot.guest_uuid
guest.get_by('uuid')
disk.uuid = disk_uuid
disk.get_by('uuid')
os_template_image.id = guest.os_template_image_id
os_template_image.get()
image_name = '_'.join([snapshot.snapshot_id, disk.uuid
]) + '.' + disk.format
os_template_image.id = 0
os_template_image.label = request.json.get('label')
os_template_image.path = '/'.join(
[os.path.dirname(os_template_image.path), image_name])
os_template_image.kind = OSTemplateImageKind.custom.value
os_template_image.progress = 0
os_template_image.create_time = ji.Common.tus()
if os_template_image.exist_by('path'):
ret['state'] = ji.Common.exchange_state(40901)
ret['state']['sub']['zh-cn'] = ''.join(
[ret['state']['sub']['zh-cn'], ': ', os_template_image.path])
return ret
os_template_image.create()
os_template_image.get_by('path')
message = {
'_object': 'snapshot',
'action': 'convert',
'uuid': disk.guest_uuid,
'snapshot_id': snapshot.snapshot_id,
'storage_mode': config.storage_mode,
'dfs_volume': config.dfs_volume,
'node_id': disk.node_id,
'snapshot_path': disk.path,
'template_path': os_template_image.path,
'os_template_image_id': os_template_image.id,
'passback_parameters': {
'id': snapshot.snapshot_id,
'os_template_image_id': os_template_image.id
}
}
Utils.emit_instruction(message=json.dumps(message, ensure_ascii=False))
snapshot.update()
return ret
except ji.PreviewingError, e:
return json.loads(e.message)