#!/usr/bin/env python
# -*- coding: UTF-8 -*-
import os
import sys
current_file_path = os.path.split(os.path.realpath(__file__))[0]
site_packages_path = os.path.abspath(os.path.join(current_file_path, "../site-packages"))
sys.path.append(site_packages_path)
import total
reload(sys)
sys.setdefaultencoding('utf-8')
from util import SPLIT_STR

# use as python tran.py "srcFolder" "dstFolder"


def add_spliter(ss):
    if not ss.endswith(SPLIT_STR):
        return ss + SPLIT_STR
    return ss


if __name__ == '__main__':
    av = sys.argv
    srcFolder = add_spliter(av[1])
    dstFolder = add_spliter(av[2])
    kind = "export"
    if len(av) == 4 and av[3] in ("version", "transform", "oldExport"):
        kind = av[3]
        if kind == "version":
            kind = "transform"
        print("start tran version:\nsrcFolder --> " + srcFolder)
    else:
        print(u"开始导表:配置路径 --> " + srcFolder + " | 导出路径 --> " + dstFolder)
    try:
        result = total.convert_all_excel(srcFolder, dstFolder, kind)
        if result == 0:
            print(u"转换成功")
        elif result == 1:
            print(u"转换失败")
        elif result == 2:
            print(u"转换成功但是check rule失败")
        else:
            print(u"未知错误")
    except BaseException as info:
        print(unicode(info))
        sys.exit(1)