#! encoding:UTF-8
import cairosvg
import os
 
loop = True
while loop:
    svgDir = raw_input("請(qǐng)輸入SVG文件目錄")
    if os.path.exists(svgDir) and os.path.isdir(svgDir):
        loop = False
    else:
        print "錯(cuò)誤：您輸入的SVG文件目錄不存在或者不是一個(gè)有效的目錄，請(qǐng)重新輸入"
 
loop = True
while loop:
    exportDir = raw_input("請(qǐng)輸入導(dǎo)出目錄")
    if os.path.exists(exportDir):
        loop = False
    else:
        print "錯(cuò)誤：您輸入的導(dǎo)出目錄[" , exportDir  , "] 不存在，是否要?jiǎng)?chuàng)建這個(gè)目錄？"
        loops = True     
        while loops:
            msg = ""
            cmd = raw_input("創(chuàng)建 (Y) 重新 (R)")
            if cmd.upper() == "R":
                loops = False
            elif cmd.upper() == "Y":
                os.makedirs(exportDir, True)
                if os.path.exists(exportDir):
                    loop = False
                    loops = False
                else:
                    print "創(chuàng)建目錄失敗[",exportDir,"]， 請(qǐng)重新輸入"
            else:
                print "找不到您輸入的命令，請(qǐng)重新輸入"
             
 
cate = ("png", "pdf")
print "導(dǎo)出類型："
for i in cate:
    print i
     
loop = True
while loop:
    exportFormat = raw_input("請(qǐng)輸入導(dǎo)出類型")
    if exportFormat.lower() in cate:
        loop = False
    else:
        print "您輸入的類型不存在，請(qǐng)重新輸入"
 
def export(fromDir, targetDir, exportType):
    print "開始執(zhí)行轉(zhuǎn)換命令..."
    files = os.listdir(fromDir)
    num = 0
    for fileName in files:
        path = os.path.join(fromDir,fileName)
        if os.path.isfile(path) and fileName[-3:] == "svg":
            num += 1
            fileHandle = open(path)
            svg = fileHandle.read()
            fileHandle.close()
            exportPath = os.path.join(targetDir, fileName[:-3] + exportType)
            exportFileHandle = open(exportPath,'w')
             
            if exportType == "png":
                cairosvg.svg2png(bytestring=svg, write_to=exportPath)
            elif exportType == "pdf":
                cairosvg.svg2pdf(bytestring=svg, write_to=exportPath)
                 
            exportFileHandle.close()
            print "Success Export ", exportType, " -> " , exportPath
     
    print "已導(dǎo)出 ", num, "個(gè)文件"
export(svgDir, exportDir, exportFormat)