mdserver-web/route/plugins.py

# coding:utf-8

import time
import sys
import os
import json
import psutil

from flask import Blueprint, render_template
from flask import jsonify
from flask import request

sys.path.append("class/core")
import public
import plugin_api


plugins = Blueprint('plugins', __name__, template_folder='templates')
__plugin_name = "plugins"
__row_num = 3


@plugins.route('/file', methods=['GET'])
def file():
    name = request.args.get('name', '')
    if name.strip() == '':
        return ''

    f = request.args.get('f', '')
    if f.strip() == '':
        return ''

    file = __plugin_name + '/' + name + '/' + f
    if not os.path.exists(file):
        return ''

    c = public.readFile(file)
    return c


@plugins.route('/list', methods=['GET'])
def list():
    typeVal = request.args.get('type', '0')
    data = plugin_api.plugin_api().getPluginList(typeVal, 1)
    return public.getJson(data)


@plugins.route('/index_list', methods=['GET'])
def indexList():
    data = plugin_api.plugin_api().getIndexList()
    return public.getJson(data)


@plugins.route('/index_sort', methods=['POST'])
def indexSort():
    sort = request.form.get('ssort', '')
    if sort.strip() == '':
        return public.returnJson(False, '排序数据不能为空!')
    data = plugin_api.plugin_api().setIndexSort(sort)
    if data:
        return public.returnJson(True, '成功!')
    return public.returnJson(False, '失败!')


@plugins.route('/install', methods=['POST'])
def install():

    rundir = public.getRunDir()
    name = request.form.get('name', '')
    version = request.form.get('version', '')

    mmsg = '安装'
    if hasattr(request.form, 'upgrade'):
        mtype = 'update'
        mmsg = 'upgrade'

    if name.strip() == '':
        return public.returnJson(False, '缺少插件名称!', ())

    if version.strip() == '':
        return public.returnJson(False, '缺少版本信息!', ())

    infoJsonPos = __plugin_name + '/' + name + '/' + 'info.json'

    if not os.path.exists(infoJsonPos):
        return public.retJson(False, '配置文件不存在!', ())

    pluginInfo = json.loads(public.readFile(infoJsonPos))

    execstr = "cd " + os.getcwd() + "/plugins/" + \
        name + " && /bin/bash " + pluginInfo["shell"] \
        + " install " + version

    taskAdd = (None, mmsg + '[' + name + '-' + version + ']',
               'execshell', '0', time.strftime('%Y-%m-%d %H:%M:%S'), execstr)

    public.M('tasks').add('id,name,type,status,addtime, execstr', taskAdd)
    return public.returnJson(True, '已将安装任务添加到队列!')


@plugins.route('/uninstall', methods=['POST'])
def uninstall():
    rundir = public.getRunDir()
    name = request.form.get('name', '')
    version = request.form.get('version', '')
    if name.strip() == '':
        return public.returnJson(False, "缺少插件名称!", ())

    if version.strip() == '':
        return public.returnJson(False, "缺少版本信息!", ())

    infoJsonPos = __plugin_name + '/' + name + '/' + 'info.json'

    if not os.path.exists(infoJsonPos):
        return public.retJson(False, "配置文件不存在!", ())

    pluginInfo = json.loads(public.readFile(infoJsonPos))

    execstr = "cd " + os.getcwd() + "/plugins/" + \
        name + " && /bin/bash " + pluginInfo["shell"] \
        + " uninstall " + version

    taskAdd = (None, '卸载[' + name + '-' + version + ']',
               'execshell', '0', time.strftime('%Y-%m-%d %H:%M:%S'), execstr)

    public.M('tasks').add('id,name,type,status,addtime, execstr', taskAdd)
    return public.returnJson(True, '已将卸载任务添加到队列!')


@plugins.route('/installed', methods=['POST'])
def installed():

    rundir = public.getRunDir()
    name = request.form.get('name', '')

    if name.strip() == '':
        return public.retJson(-1, "缺少插件名称!", ())

    infoJsonPos = __plugin_name + '/' + name + '/' + 'info.json'
    if not os.path.exists(infoJsonPos):
        return public.returnJson(-1, "配置文件不存在!", ())

    pluginInfo = json.loads(public.readFile(infoJsonPos))

    sh = __plugin_name + '/' + name + '/' + pluginInfo['shell']
    os.system('/bin/bash ' + sh + ' install')
    print request.args
    return ''


@plugins.route('/check_installed', methods=['POST'])
def checkInstalled():
    checks = ['nginx', 'apache', 'php', 'mysql']
    for name in checks:
        filename = public.getRootDir() + "/server/" + name
        if os.path.exists(filename):
            return "True"
    return "False"


@plugins.route('/set_index', methods=['POST'])
def setIndex():
    name = request.form.get('name', '')
    status = request.form.get('status', '0')
    version = request.form.get('version', '')
    if status == '1':
        return plugin_api.plugin_api().addIndex(name, version)
    return plugin_api.plugin_api().removeIndex(name, version)


@plugins.route('/setting', methods=['GET'])
def setting():
    name = request.args.get('name', '')
    html = __plugin_name + '/' + name + '/index.html'
    return public.readFile(html)


@plugins.route('/run', methods=['POST', 'GET'])
def run():
    name = request.form.get('name', '')
    func = request.form.get('func', '')
    args = request.form.get('args', '')
    script = request.form.get('script', 'index')

    py = 'python ' + public.getRunDir() + '/' + __plugin_name + '/' + name
    if args == '':
        py = py + '/' + script + '.py' + ' ' + func
    else:
        py = py + '/' + script + '.py' + ' ' + func + ' ' + args

    print py
    data = public.execShell(py)

    if data[1].strip() == '':
        return public.returnJson(True, "OK", data[0].strip())
    return public.returnJson(False, data[1].strip())