auto_copy.py

# -*- coding:utf-8 -*-
# __author__ = "DT"
# __date__ = "20190118"

import os
import os.path
import shutil
import time, datetime


def del_file(path):
    for f in os.listdir(path):
        path_file = os.path.join(path, f)
        if os.path.isfile(path_file):
            print("文件" + "\033[1;35m %s \033[0m" %f + "正在删除，请等待2秒钟……")
            os.remove(path_file)
            time.sleep(2)
        else:
            print("文件夹" + "\033[1;32m %s \033[0m" %f + "正在删除，请等待2秒钟……")
            shutil.rmtree(path_file)
            time.sleep(2)


def get_nul_file(fname):
    fpath = "/home/qa/" + fname + "/"
    result = 0
    if not os.path.exists(fpath):
        print("\n目标文件不存在，即将为您创建该目录")
        while True:
            inputExit = raw_input('\n输入回车继续')
            if not len(inputExit):
                os.mkdir(fpath)
                result = 1
                break
            else:
                print("\n\033[1;35m请按回车继续，退出按 ctrl+c \033[0m")
                continue

    else:
        if not os.listdir(fpath):
            print("\n目标文件夹存在，为空可以使用")
            result = 1
        else:
            print("\n目标文件夹有以下文件，即将为您清空该目录是否继续……\n")
            for i in os.listdir(fpath):
                print("\033[1;35m %s \033[0m" %i)

            while True:
                inputExit = raw_input('\n请输入0或1 (0:退出，1:继续)')
                if not len(inputExit):
                    print("\n\033[1;35m 请输入0或1 \033[0m")
                    continue
                if str(inputExit) == '0':
                    # print("退出")
                    result = 0
                    break
                if str(inputExit) == '1':
                    del_file(fpath)
                    result = 1
                    break
                else:
                    print("\n\033[1;32m 输入有误，请输入0或1 \033[0m")
    return result


def mv_file(path, fname):
    result1 = 0
    from_path = path + fname + "/"
    commands = "scp qa@192.168.12.158:/home/qa/trunk/Server/Game/dist/MobilePhone-release* %s" %from_path
    print("\n\033[1;35m正在从158服务器拷贝文件，请等待2秒钟 \033[0m")
    os.system(commands)
    result1 = 1
    return result1


def call_install(fname):
    os.chdir("/home/qa/")
    callpath = os.getcwd()
    print("\n准备安装版本目录:\033[1;32m %s \033[0m" %fname)
    while True:
        src = raw_input("\n请输入对比安装的版本文件夹名：")
        allpath = os.path.join(callpath, src)
        if not os.path.exists(allpath):
            print("您输入的文件夹不存在，请重新输入")
            continue
        else:
            if "release" in os.listdir(allpath):
                commands1 = "./installPackage %s %s" %(fname, src)
                os.system(commands1)
                break
            else:
                print("对比文件中没有release目录，请检查后重新输入")
                continue
    

def restart_1(fname):
    print("\n重启新服务器版本，请稍等……")
    os.chdir("/home/qa/")
    commands1 = "./2RESTARTAlpha %s" %fname
    os.system(commands1)


def main():
    path = "/home/qa/"
    os.chdir(path)
    os.system("ls")
    fname = raw_input("\n\033[1;33m请输入安装文件夹名：\033[0m")
    goon = get_nul_file(fname)
    if goon == 1:
       instllnum = mv_file(path, fname)
       if instllnum == 1:
           call_install(fname)
           time.sleep(2)
           restart_1(fname)
    else:
        print("\n\033[1;33m程序结束，欢迎使用 \033[0m")


if __name__ == '__main__':
	main()