mdserver-web/plugins/rsyncd/index.py

# coding: utf-8

import time
import random
import os
import json
import re
import sys

sys.path.append(os.getcwd() + "/class/core")
import public

app_debug = False
if public.isAppleSystem():
    app_debug = True


def getPluginName():
    return 'rsyncd'


def getPluginDir():
    return public.getPluginDir() + '/' + getPluginName()


def getServerDir():
    return public.getServerDir() + '/' + getPluginName()


def getInitDFile():
    if app_debug:
        return '/tmp/' + getPluginName()
    return '/etc/init.d/' + getPluginName()


def getArgs():
    args = sys.argv[2:]
    tmp = {}
    args_len = len(args)

    if args_len == 1:
        t = args[0].strip('{').strip('}')
        t = t.split(':')
        tmp[t[0]] = t[1]
    elif args_len > 1:
        for i in range(len(args)):
            t = args[i].split(':')
            tmp[t[0]] = t[1]

    return tmp


def checkArgs(data, ck=[]):
    for i in range(len(ck)):
        if not ck[i] in data:
            return (False, public.returnJson(False, '参数:(' + ck[i] + ')没有!'))
    return (True, public.returnJson(True, 'ok'))


def status():
    data = public.execShell(
        "ps -ef|grep rsync |grep -v grep | grep -v python | awk '{print $2}'")
    if data[0] == '':
        return 'stop'
    return 'start'


def appConf():
    if public.isAppleSystem():
        return getServerDir() + '/rsyncd.conf'
    return '/etc/rsyncd.conf'


def appConfPwd():
    if public.isAppleSystem():
        return getServerDir() + '/rsyncd.passwd'
    return '/etc/rsyncd.passwd'


def getLog():
    conf_path = appConf()
    conf = public.readFile(conf_path)
    rep = 'log file\s*=\s*(.*)'
    tmp = re.search(rep, conf)
    if not tmp:
        return ''
    return tmp.groups()[0]


def initConf():
    conf_path = appConf()
    conf = public.readFile(conf_path)
    conf = re.sub('^#(.*)', '', conf)
    conf_tpl_path = getPluginDir() + '/conf/rsyncd.conf'
    if conf.strip() == '':
        content = public.readFile(conf_tpl_path)
        public.writeFile(conf_path, content)
    confpwd_path = appConfPwd()
    if not os.path.exists(confpwd_path):
        public.writeFile(confpwd_path, '')
        public.execShell('chmod 0600 ' + confpwd_path)


def start():
    initConf()

    if public.isAppleSystem():
        return "Apple Computer does not support"

    data = public.execShell('systemctl start rsyncd.service')
    if data[1] == '':
        return 'ok'
    return 'fail'


def stop():
    if public.isAppleSystem():
        return "Apple Computer does not support"
    data = public.execShell('systemctl stop rsyncd.service')
    if data[1] == '':
        return 'ok'
    return 'fail'


def restart():
    if public.isAppleSystem():
        return "Apple Computer does not support"
    data = public.execShell('systemctl restart rsyncd.service')
    if data[1] == '':
        return 'ok'
    return 'fail'


def reload():
    if public.isAppleSystem():
        return "Apple Computer does not support"

    data = public.execShell('systemctl reload rsyncd.service')
    if data[1] == '':
        return 'ok'
    return 'fail'


def initdStatus():
    if public.isAppleSystem():
        return "Apple Computer does not support"

    data = public.execShell('systemctl status rsyncd.service | grep enabled')
    if data[1] == '':
        return 'ok'
    return 'fail'


def initdInstall():
    if public.isAppleSystem():
        return "Apple Computer does not support"
    data = public.execShell('systemctl enable rsyncd.service')
    if data[1] == '':
        return 'ok'
    return 'fail'


def initdUinstall():
    if public.isAppleSystem():
        return "Apple Computer does not support"
    data = public.execShell('systemctl disable rsyncd.service')
    if data[1] == '':
        return 'ok'
    return 'fail'


def getRecListData():
    path = appConf()
    content = public.readFile(path)

    flist = re.findall("\[(.*)\]", content)
    flist_len = len(flist)
    ret_list = []
    for i in range(flist_len):
        tmp = {}
        tmp['name'] = flist[i]
        n = i + 1
        reg = ''
        if n == flist_len:
            reg = '\[' + flist[i] + '\](.*)\[?'
        else:
            reg = '\[' + flist[i] + '\](.*)\[' + flist[n] + '\]'

        t1 = re.search(reg, content, re.S)
        if t1:
            args = t1.groups()[0]
            # print 'args start', args, 'args_end'
            t2 = re.findall('\s*(.*)\s*=\s*(.*)', args, re.M)
            for i in range(len(t2)):
                tmp[t2[i][0].strip()] = t2[i][1]
        ret_list.append(tmp)
    return ret_list


def getRecList():
    ret_list = getRecListData()
    return public.returnJson(True, 'ok', ret_list)


def getUPwdList():
    pwd_path = appConfPwd()
    pwd_content = public.readFile(pwd_path)
    plist = pwd_content.strip().split('\n')
    plist_len = len(plist)
    data = {}
    for x in range(plist_len):
        tmp = plist[x].split(':')
        data[tmp[0]] = tmp[1]
    return data


def addRec():
    args = getArgs()
    data = checkArgs(args, ['name', 'path', 'pwd', 'ps'])
    if not data[0]:
        return data[1]

    args_name = args['name']
    args_pwd = args['pwd']
    args_path = args['path']
    args_ps = args['ps']

    pwd_path = appConfPwd()
    pwd_content = public.readFile(pwd_path)
    pwd_content += args_name + ':' + args_pwd + "\n"
    public.writeFile(pwd_path, pwd_content)

    path = appConf()
    content = public.readFile(path)

    con = "\n\n" + '[' + args_name + ']' + "\n"
    con += 'path = ' + args_path + "\n"
    con += 'comment = ' + args_ps + "\n"
    con += 'auth users = ' + args_name + "\n"
    con += 'read only = false'

    content = content + con
    public.writeFile(path, content)
    return public.returnJson(True, '添加成功')


def delRec():
    args = getArgs()
    data = checkArgs(args, ['name'])
    if not data[0]:
        return data[1]
    args_name = args['name']

    cmd = "sed -i '_bak' '/" + args_name + "/d' " + appConfPwd()
    public.execShell(cmd)

    try:

        path = appConf()
        content = public.readFile(path)

        ret_list = getRecListData()
        ret_list_len = len(ret_list)
        is_end = False
        next_name = ''
        for x in range(ret_list_len):
            tmp = ret_list[x]
            if tmp['name'] == args_name:
                if x + 1 == ret_list_len:
                    is_end = True
                else:
                    next_name = ret_list[x + 1]['name']
        reg = ''
        if is_end:
            reg = '\[' + args_name + '\]\s*(.*)'
        else:
            reg = '\[' + args_name + '\]\s*(.*)\s*\[' + next_name + '\]'

        conre = re.search(reg,  content, re.S)
        content = content.replace(
            "[" + args_name + "]\n" + conre.groups()[0], '')
        public.writeFile(path, content)
        return public.returnJson(True, '删除成功!')
    except Exception as e:
        return public.returnJson(False, '删除失败!')


def cmdRec():
    args = getArgs()
    data = checkArgs(args, ['name'])
    if not data[0]:
        return data[1]

    an = args['name']
    pwd_list = getUPwdList()
    ip = public.getLocalIp()

    cmd = 'echo "' + pwd_list[an] + '" > /tmp/p.pass' + "<br>"
    cmd += 'rsync -arv --password-file=/tmp/p.pass --progress --delete  /project  ' + \
        an + '@' + ip + '::' + an
    return public.returnJson(True, 'OK!', cmd)

# rsyncdReceive
if __name__ == "__main__":
    func = sys.argv[1]
    if func == 'status':
        print status()
    elif func == 'start':
        print start()
    elif func == 'stop':
        print stop()
    elif func == 'restart':
        print restart()
    elif func == 'reload':
        print reload()
    elif func == 'initd_status':
        print initdStatus()
    elif func == 'initd_install':
        print initdInstall()
    elif func == 'initd_uninstall':
        print initdUinstall()
    elif func == 'conf':
        print appConf()
    elif func == 'conf_pwd':
        print appConfPwd()
    elif func == 'run_log':
        print getLog()
    elif func == 'rec_list':
        print getRecList()
    elif func == 'add_rec':
        print addRec()
    elif func == 'del_rec':
        print delRec()
    elif func == 'cmd_rec':
        print cmdRec()
    else:
        print 'error'