#!/usr/bin/env python # -*- coding: utf-8 -*- import json from flask import Blueprint, render_template, url_for, request import requests from math import ceil from models.status import StorageMode __author__ = 'James Iter' __date__ = '2017/6/25' __contact__ = 'james.iter.cn@gmail.com' __copyright__ = '(c) 2017 by James Iter.' blueprint = Blueprint( 'v_disk', __name__, url_prefix='/disk' ) blueprints = Blueprint( 'v_disks', __name__, url_prefix='/disks' ) def show(): args = list() page = request.args.get('page', 1) if page == '': page = 1 page = int(page) page_size = int(request.args.get('page_size', 10)) keyword = request.args.get('keyword', None) show_area = request.args.get('show_area', 'unmount') guest_uuid = request.args.get('guest_uuid', None) sequence = request.args.get('sequence', None) order_by = request.args.get('order_by', None) order = request.args.get('order', None) filters = list() resource_path = request.path if page is not None: args.append('page=' + page.__str__()) if page_size is not None: args.append('page_size=' + page_size.__str__()) if keyword is not None: args.append('keyword=' + keyword.__str__()) if guest_uuid is not None: filters.append('guest_uuid:in:' + guest_uuid.__str__()) show_area = 'all' if sequence is not None: filters.append('sequence:in:' + sequence.__str__()) show_area = 'all' if show_area in ['unmount', 'data_disk', 'all']: if show_area == 'unmount': filters.append('sequence:eq:-1') elif show_area == 'data_disk': filters.append('sequence:gt:0') else: pass else: # 与前端页面相照应，首次打开时，默认只显示未挂载的磁盘 filters.append('sequence:eq:-1') if order_by is not None: args.append('order_by=' + order_by) if order is not None: args.append('order=' + order) if filters.__len__() > 0: args.append('filter=' + ','.join(filters)) host_url = request.host_url.rstrip('/') hosts_url = host_url + url_for('api_hosts.r_get_by_filter') disks_url = host_url + url_for('api_disks.r_get_by_filter') config_url = host_url + url_for('api_config.r_get') if keyword is not None: disks_url = host_url + url_for('api_disks.r_content_search') # 关键字检索，不支持显示域过滤 show_area = 'all' hosts_ret = requests.get(url=hosts_url, cookies=request.cookies) hosts_ret = json.loads(hosts_ret.content) hosts_mapping_by_node_id = dict() for host in hosts_ret['data']: hosts_mapping_by_node_id[int(host['node_id'])] = host if args.__len__() > 0: disks_url = disks_url + '?' + '&'.join(args) disks_ret = requests.get(url=disks_url, cookies=request.cookies) disks_ret = json.loads(disks_ret.content) guest_uuids = list() for disk in disks_ret['data']: if disk['guest_uuid'].__len__() == 36: guest_uuids.append(disk['guest_uuid']) if guest_uuids.__len__() > 0: guests_url = host_url + url_for('api_guests.r_get_by_filter', filter='uuid:in:' + ','.join(guest_uuids)) guests_ret = requests.get(url=guests_url, cookies=request.cookies) guests_ret = json.loads(guests_ret.content) guests_uuid_mapping = dict() for guest in guests_ret['data']: guests_uuid_mapping[guest['uuid']] = guest for i, disk in enumerate(disks_ret['data']): if disk['guest_uuid'].__len__() == 36: disks_ret['data'][i]['guest'] = guests_uuid_mapping[disk['guest_uuid']] config_ret = requests.get(url=config_url, cookies=request.cookies) config_ret = json.loads(config_ret.content) show_on_host = False if config_ret['data']['storage_mode'] == StorageMode.local.value: show_on_host = True last_page = int(ceil(disks_ret['paging']['total'] / float(page_size))) page_length = 5 pages = list() if page < int(ceil(page_length / 2.0)): for i in range(1, page_length + 1): pages.append(i) if i == last_page or last_page == 0: break elif last_page - page < page_length / 2: for i in range(last_page - page_length + 1, last_page + 1): if i < 1: continue pages.append(i) else: for i in range(page - page_length / 2, page + int(ceil(page_length / 2.0))): pages.append(i) if i == last_page or last_page == 0: break return render_template('disks_show.html', disks_ret=disks_ret, resource_path=resource_path, hosts_mapping_by_node_id=hosts_mapping_by_node_id, page=page, page_size=page_size, keyword=keyword, pages=pages, order_by=order_by, order=order, last_page=last_page, show_area=show_area, config_ret=config_ret, show_on_host=show_on_host) def create(): host_url = request.host_url.rstrip('/') if request.method == 'POST': size = request.form.get('size') quantity = request.form.get('quantity') remark = request.form.get('remark') node_id = request.form.get('node_id') payload = { "size": int(size), "quantity": int(quantity), "remark": remark, "node_id": node_id } url = host_url + '/api/disk' headers = {'content-type': 'application/json'} r = requests.post(url, data=json.dumps(payload), headers=headers, cookies=request.cookies) j_r = json.loads(r.content) return render_template('success.html', go_back_url='/disks', timeout=10000, title='提交成功', message_title='创建实例的请求已被接受', message='您所提交的资源正在创建中。根据所提交资源的数量，需要等待几到十几秒钟。页面将在10秒钟后自动跳转到实例列表页面！') else: config_url = host_url + url_for('api_config.r_get') config_ret = requests.get(url=config_url, cookies=request.cookies) config_ret = json.loads(config_ret.content) show_on_host = False if config_ret['data']['storage_mode'] == StorageMode.local.value: show_on_host = True return render_template('disk_create.html', show_on_host=show_on_host) def detail(uuid): host_url = request.host_url.rstrip('/') disk_url = host_url + url_for('api_disks.r_get', uuids=uuid) disk_ret = requests.get(url=disk_url, cookies=request.cookies) disk_ret = json.loads(disk_ret.content) guest_ret = None os_template_ret = None config_url = host_url + url_for('api_config.r_get') config_ret = requests.get(url=config_url, cookies=request.cookies) config_ret = json.loads(config_ret.content) if disk_ret['data']['sequence'] != -1: guest_url = host_url + url_for('api_guests.r_get', uuids=disk_ret['data']['guest_uuid']) guest_ret = requests.get(url=guest_url, cookies=request.cookies) guest_ret = json.loads(guest_ret.content) os_template_url = host_url + url_for('api_os_templates.r_get', ids=guest_ret['data']['os_template_id'].__str__()) os_template_ret = requests.get(url=os_template_url, cookies=request.cookies) os_template_ret = json.loads(os_template_ret.content) return render_template('disk_detail.html', uuid=uuid, guest_ret=guest_ret, os_template_ret=os_template_ret, disk_ret=disk_ret, config_ret=config_ret)