generateScriptFile.py脚本使用过程中遇到的问题及解决

generateScriptFile.py脚本

#!/usr/bin/env python
# -*- coding: utf-8 -*-

"""
use case:
python ./generateScriptFile.py
"""

import MySQLdb
import sys
import os
from lxml import etree

#IP=sys.argv[1]
IP="10.0.20.252"
DIR="ScriptsDir"
try:
    print "connect to database..."
    acheron_db=MySQLdb.connect(IP,"root","Free-Wi11","acheron_db")
except Exception as e:
    print e.message
    print "Can't connect to "+IP
    exit(1)
print "Done..."
print "Ready to read data from database..."
acheron_db_cursor=acheron_db.cursor()

sqlCMD="select Description,TestScript from Script where Class='System'"
acheron_db_cursor.execute(sqlCMD)
content=acheron_db_cursor.fetchall()
if len(content)==0:
    print "no data obtained"
    exit(1)
if os.path.exists(DIR):
    os.system("RD /S /Q  " + DIR)
os.system("mkdir "+DIR)
for item in content:
    file_name=item[0]
    if len(file_name)==0:
        print item
        continue
        
    directory=item[0].split(".")[0]
    if len(directory)==0:
        print item[0]
        continue
    if not os.path.exists(DIR+"\\"+directory):
        os.system("mkdir "+DIR+"\\"+directory)
    try:
        f=open(DIR+"\\"+directory+"\\"+file_name+".xml","w")
        root=etree.fromstring(item[1])
    f.write("<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n");
        f.write(etree.tostring(root))
        #f.write(etree.tostring(root,xml_declaration=True,encoding='UTF-8'))
        f.close()
    except Exception as e:
    print file_name
        print e.message
        print "Failed, please retry..."
        break
print "See scripts data file in '"+DIR+"'"
print "Done"

装Python的依赖库

安装MySQLdb

http://www.codegood.com/download/11/

下载：MySQL-python-1.2.3.win-amd64-py2.7.exe

点击安装

Python 2.7.13 (v2.7.13:a06454b1afa1, Dec 17 2016, 20:53:40) [MSC v.1500 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import MySQLdb
>>>