# -*- coding: utf-8 -*-
from XmlToExcel import XmlToExcel
import sys
import os
import time

# 定义后缀过滤列表
filter_list = ['.xml']

# 说明：检查output文件夹是否存在，不存在则创建
# 参数：无
# 返回值：输出路径
def get_output_dir():
    output = os.path.join(os.getcwd(), 'output')
    if not os.path.exists(output):
        os.mkdir(output)
    return output


# 说明：遍历指定文件夹扫描xml文件
# 参数：无
# 返回值：输出文件夹下所有xml文件路径列表
def get_input_xml(input_dir):
    files = []
    # input文件夹不存在
    if not os.path.exists(input_dir):
        return files
    else: # 遍历该目录下所有文件
        for _, _, file_list in os.walk(input_dir):
            for filename in file_list:
                file_path = os.path.join(input_dir, filename)
                # 获取文件后缀
                ext = os.path.splitext(file_path)[1]
                # 满足要求的则放入
                if ext in filter_list:
                    files.append(file_path)
    return files

# 将命令行参数转换为对应的文件输入
def get_command():
    xmls = []
    input_dir = ''
    # 未传入参数，使用默认设置
    if len(sys.argv) == 1:
        input_dir = os.path.join(os.getcwd(), 'input')
        xmls = get_input_xml(input_dir)
    elif len(sys.argv) == 2:  # 传入了参数
        input_dir = sys.argv[1]
        # 如果传入的是单个文件
        if os.path.isfile(input_dir):
            ext = os.path.splitext(input_dir)[1]
            # 判断是否是xml
            if ext not in filter_list:
                print(u'ERROR：请传入正确的xml文件')
                return False
            xmls.append(input_dir)
        elif os.path.exists(input_dir): # 如果是文件夹
            xmls = get_input_xml(input_dir)
    return xmls, input_dir


# 说明：将指定xml转换为Excel文件
# 参数：无
# 返回值：成功返回True
def converter():
    # 获取命令行参数
    xmls, input_dir = get_command()
    # 校验文件输入
    if len(xmls) == 0:
        print('ERROR: 未发现输入的xml文件')
        return False
    print('--------------------------------')
    print('INFO: 当前输入文件为：', input_dir)
    output = get_output_dir()
    # 遍历文件输出
    print('INFO: 开始执行xml转Excel...')
    print('--------------------------------')
    # 实例化
    xm = XmlToExcel()
    # 返回转换结果
    result = xm.read_xml(xmls)
    file_name = time.strftime(r'%Y%m%d%H%M%S', time.localtime()) + '_Example_Testsuite_Default.xls'
    excel = os.path.join(output, file_name)
    # 写入Excel文件
    res = xm.write_excel(excel)
    
    print('成功转换文件数：\t', result['success'])
    print('失败转换文件数：\t', result['fail'])
    print('模块总数为：\t', res['module'])
    print('案例总数为：\t', res['case'])
    print('------------------------')
    print('INFO: 全部转换成功，已输出到文件：', excel)
    return True


if __name__ == "__main__":
    # 开始转换
    converter()