JimV-C/jimvc/api/vlan.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-


import requests
from flask import Blueprint, request, url_for
import json
import jimit as ji

from jimvc.models import Utils, Guest
from jimvc.models import Rules
from jimvc.models import VLAN
from jimvc.models import Host

from base import Base


__author__ = 'James Iter'
__date__ = '2019-03-12'
__contact__ = 'james.iter.cn@gmail.com'
__copyright__ = '(c) 2019 by James Iter.'


blueprint = Blueprint(
    'api_vlan',
    __name__,
    url_prefix='/api/vlan'
)

blueprints = Blueprint(
    'api_vlans',
    __name__,
    url_prefix='/api/vlans'
)


vlan_base = Base(the_class=VLAN, the_blueprint=blueprint, the_blueprints=blueprints)


@Utils.dumps2response
def r_create():

    args_rules = [
        Rules.VLAN_ID.value,
        Rules.LABEL.value,
        Rules.DESCRIPTION.value
    ]

    vlan = VLAN()
    vlan.vlan_id = request.json.get('vlan_id')
    vlan.label = request.json.get('label')
    vlan.description = request.json.get('description', '')

    try:
        ji.Check.previewing(args_rules, vlan.__dict__)

        ret = dict()
        ret['state'] = ji.Common.exchange_state(20000)

        if vlan.exist_by('vlan_id'):
            ret['state'] = ji.Common.exchange_state(40901)
            ret['state']['sub']['zh-cn'] = ''.join([ret['state']['sub']['zh-cn'], u': Vlan ID: ', str(vlan.vlan_id)])
            return ret

        vlan.create()
        vlan.get()

        # 取全部活着的 hosts
        available_hosts = Host.get_available_hosts(nonrandom=None)
        for host in available_hosts:
            message = {
                '_object': 'vlan',
                'action': 'create',
                'uuid': None,
                'node_id': host['node_id'],
                'vlan_id': vlan.vlan_id,
                'passback_parameters': {'vlan_id': vlan.vlan_id}
            }

            Utils.emit_instruction(message=json.dumps(message, ensure_ascii=False))

        ret['data'] = vlan.__dict__
        return ret
    except ji.PreviewingError, e:
        return json.loads(e.message)


@Utils.dumps2response
def r_update(ids):

    ret = dict()
    ret['state'] = ji.Common.exchange_state(20000)
    ret['data'] = list()

    args_rules = [
        Rules.IDS.value
    ]

    if 'label' in request.json:
        args_rules.append(
            Rules.LABEL.value,
        )

    if 'description' in request.json:
        args_rules.append(
            Rules.DESCRIPTION.value,
        )

    if args_rules.__len__() < 2:
        return ret

    request.json['ids'] = ids

    try:
        ji.Check.previewing(args_rules, request.json)

        vlan = VLAN()

        # 检测所指定的 vlan 都存在
        for _id in ids.split(','):
            vlan.id = _id
            vlan.get()

        for _id in ids.split(','):
            vlan.id = _id
            vlan.get()
            vlan.label = request.json.get('label', vlan.label)
            vlan.description = request.json.get('description', vlan.description)

            vlan.update()
            vlan.get()

            ret['data'].append(vlan.__dict__)

        return ret
    except ji.PreviewingError, e:
        return json.loads(e.message)


@Utils.dumps2response
def r_get(ids):
    ret = vlan_base.get(ids=ids, ids_rule=Rules.IDS.value, by_field='id')

    if '200' != ret['state']['code']:
        return ret

    hosts_url = url_for('api_hosts.r_get_by_filter', _external=True)
    hosts_ret = requests.get(url=hosts_url, cookies=request.cookies)
    hosts_ret = json.loads(hosts_ret.content)

    hosts_mapping_by_vlan_bridge = dict()
    for vlan in ret['data']:
        vlan_bridge_name = 'vlan' + vlan['vlan_id'].__str__()
        vlan_bond_name = 'bond0.' + vlan['vlan_id'].__str__()

        if vlan_bridge_name not in hosts_mapping_by_vlan_bridge:
            hosts_mapping_by_vlan_bridge[vlan_bridge_name] = list()

        for host in hosts_ret['data']:
            if vlan_bridge_name in host['vlans'] and vlan_bond_name in host['vlans']:
                hosts_mapping_by_vlan_bridge[vlan_bridge_name].append(host)

    hosts = hosts_ret['data']
    posts = hosts.__len__()

    for i, vlan in enumerate(ret['data']):
        vlan_bridge_name = 'vlan' + vlan['vlan_id'].__str__()
        ret['data'][i]['posts'] = posts
        ret['data'][i]['arrival'] = hosts_mapping_by_vlan_bridge[vlan_bridge_name].__len__()

    return ret


@Utils.dumps2response
def r_get_by_filter():
    ret = vlan_base.get_by_filter()
    if '200' != ret['state']['code']:
        return ret

    hosts_url = url_for('api_hosts.r_get_by_filter', _external=True)
    hosts_ret = requests.get(url=hosts_url, cookies=request.cookies)
    hosts_ret = json.loads(hosts_ret.content)

    hosts_mapping_by_vlan_bridge = dict()
    for vlan in ret['data']:
        vlan_bridge_name = 'vlan' + vlan['vlan_id'].__str__()
        vlan_bond_name = 'bond0.' + vlan['vlan_id'].__str__()

        if vlan_bridge_name not in hosts_mapping_by_vlan_bridge:
            hosts_mapping_by_vlan_bridge[vlan_bridge_name] = list()

        for host in hosts_ret['data']:
            if vlan_bridge_name in host['vlans'] and vlan_bond_name in host['vlans']:
                hosts_mapping_by_vlan_bridge[vlan_bridge_name].append(host)

    hosts = hosts_ret['data']
    posts = hosts.__len__()

    for i, vlan in enumerate(ret['data']):
        vlan_bridge_name = 'vlan' + vlan['vlan_id'].__str__()
        ret['data'][i]['posts'] = posts
        ret['data'][i]['arrival'] = hosts_mapping_by_vlan_bridge[vlan_bridge_name].__len__()

    return ret


@Utils.dumps2response
def r_content_search():
    ret = vlan_base.content_search()

    if '200' != ret['state']['code']:
        return ret

    hosts_url = url_for('api_hosts.r_get_by_filter', _external=True)
    hosts_ret = requests.get(url=hosts_url, cookies=request.cookies)
    hosts_ret = json.loads(hosts_ret.content)

    hosts_mapping_by_vlan_bridge = dict()
    for vlan in ret['data']:
        vlan_bridge_name = 'vlan' + vlan['vlan_id'].__str__()
        vlan_bond_name = 'bond0.' + vlan['vlan_id'].__str__()

        if vlan_bridge_name not in hosts_mapping_by_vlan_bridge:
            hosts_mapping_by_vlan_bridge[vlan_bridge_name] = list()

        for host in hosts_ret['data']:
            if vlan_bridge_name in host['vlans'] and vlan_bond_name in host['vlans']:
                hosts_mapping_by_vlan_bridge[vlan_bridge_name].append(host)

    hosts = hosts_ret['data']
    posts = hosts.__len__()

    for i, vlan in enumerate(ret['data']):
        vlan_bridge_name = 'vlan' + vlan['vlan_id'].__str__()
        ret['data'][i]['posts'] = posts
        ret['data'][i]['arrival'] = hosts_mapping_by_vlan_bridge[vlan_bridge_name].__len__()

    return ret


@Utils.dumps2response
def r_delete(ids):
    ret = dict()
    ret['state'] = ji.Common.exchange_state(20000)

    args_rules = [
        Rules.IDS.value
    ]

    try:

        ji.Check.previewing(args_rules, {'ids': ids})

        vlan = VLAN()

        # 检测所指定的 vlan 都存在
        for _id in ids.split(','):
            vlan.id = _id
            vlan.get()

            guests, _ = Guest.get_by_filter(filter_str=':'.join(['vlan_id', 'eq', vlan.vlan_id.__str__()]))

            # 检测 vlan 中是否存在 虚拟机，若存在，则拒绝删除操作。
            if guests.__len__() > 0:

                guests_info = ''
                for guest in guests:
                    guests_info += guest['label'] + ',' + guest['uuid'] + '.'

                ret['state'] = ji.Common.exchange_state(41262)
                ret['state']['sub']['zh-cn'] = ''.join([
                    ret['state']['sub']['zh-cn'], u': Vlan ID: ', vlan.vlan_id, u': Guests: ', guests_info])

                return ret

        for _id in ids.split(','):
            vlan.id = _id
            vlan.get()

            # 取全部活着的 hosts
            available_hosts = Host.get_available_hosts(nonrandom=None)
            for host in available_hosts:
                message = {
                    '_object': 'vlan',
                    'action': 'delete',
                    'uuid': None,
                    'node_id': host['node_id'],
                    'vlan_id': vlan.vlan_id,
                    'passback_parameters': {'id': vlan.id, 'vlan_id': vlan.vlan_id}
                }

                Utils.emit_instruction(message=json.dumps(message, ensure_ascii=False))

        return ret

    except ji.PreviewingError, e:
        return json.loads(e.message)

    # return vlan_base.delete(ids=ids, ids_rule=Rules.IDS.value, by_field='id')