Python使用总结二

近来因为工作需要，用Python比较多，写得多了，收获也多。借此记录总结一下，方便以后反思。

一、IDE的选择

1、notepad++加上cmd窗口

前些时候写python脚本都用notepad++编辑代码，然后用控制台运行。当时觉得这样挺方便，不用去折腾IDE。这种组合对编写一些代码量不多，且使用的模块是自己熟悉的脚本是可以的，极推荐使用。但由于没有工程管理，源代码文件的存放可能会比较乱，而且没有代码提示功能，代码产出的效率比较低。

2、eric5 与eclipse 加 pydev

发现notepad++满足不了我的需求后，我便开始为python配一个IDE开发环境。

首先尝试的是eric5，因为网上评分很高。而且它还是用QT写的，用python开发GUI程序应该很方便。不过让人纠结的是eric5的代码提示功能很弱，只能提示自带的模块，对第三方模块，如我装的win32com模块不能提示。唉~~，看来只好另寻IDE了。

最后改用了eclipse + pydev的环境，eclipse就一神器，代码提示能力甩eric5几条街。不过它很吃内存，但做开发的，你的机器没个4G以上，说不过去吧，所以内存这方面不是大问题，就选它了。配置也比较简单，网上很多前辈给出了方法。

二、xml操作

Python自带有xml相关模块xml.etree.ElementTree与xml.dom.minidom。在使用的过程中，我用xml.etree.ElementTree来对xml文件进行读，用xml.dom.minidom对xml文件进行写操作。相关模块使用可以查看python手册，地址如下：

http://docs.python.org/2/library/xml.etree.elementtree.html

http://docs.python.org/2/library/xml.dom.minidom.html

以下给出的代码是实现对两个xml文件进行节点比较，并把不同的结点信息生成一个新的xml文件。代码覆盖了xml文件的读与写操作。

 

import xml.etree.ElementTree as ET
import sys, os

# 属性比较
def compare_attr(dictattr1,  dictattr2):
    dictvalue = {}
    # dictattr1 - dictattr2
    diffLis = dictattr1.keys() - dictattr2.keys()
    print (diffLis)
    for item in diffLis:
        dictvalue[item] = "only ({0}) exist attrib".format(os.path.split(sys.argv[1])[1])
    # dictattr2 - dictattr1    
    diffLis = dictattr2.keys() - dictattr1.keys()
    print (diffLis)
    for item in diffLis:
        dictvalue[item] = "only ({0}) exist attrib".format(os.path.split(sys.argv[2])[1])
    
    # dictattr2 & dictattr1 
    andList = dictattr1.keys() & dictattr2.keys()
    #print (andList)
    for key in andList:
        if dictattr1[key] != dictattr2[key]:
            dictvalue[key] = "different_value({0})and({1})".format(dictattr1[key],  dictattr2[key])
    return dictvalue        
   
def getAttribute(node):
    if node == None:
        return dict()
    else:
        return node.attrib

def getChildren(node):
    if node == None:
        return list()
    else:
        return node.getchildren()
        
#节点比较
def compare_xmltree(root1, root2, domTree,  xmlNode):    
    if root1 == None or root2 == None:
        strValue = ""
        if root2 == None:
            strValue = "only ({0}) has the Node and ChildrenNode".format(os.path.split(sys.argv[1])[1])
        else:
            strValue = "only ({0}) has the Node and ChildrenNode".format(os.path.split(sys.argv[2])[1])
            
        xmlNode.setAttribute("differentNode",  strValue)
        return True

    dictvalue = compare_attr(getAttribute(root1),  getAttribute(root2))
    if len(dictvalue) > 0:
        for k,  v in dictvalue.items():
            xmlNode.setAttribute(k,  v)   
  
    #取孩子结点
    childrenlist1 = getChildren(root1)
    childrenlist2 = getChildren(root2)
    print (childrenlist1)
    print (childrenlist2)    
    list1 = [child.tag for child in childrenlist1]
    list2 = [child.tag for child in childrenlist2]
    set1 = set(list1)
    set2 = set(list2)
    
    diffSet = set2 & set2
    print (diffSet)
    for item in diffSet:
        index1 = list1.index(item)
        index2 = list2.index(item)
        childnode = domTree.createElement(item)
        compare_xmltree(childrenlist1[index1],  childrenlist2[index2],  domTree,  childnode)
        if childnode.hasAttributes() or childnode.hasChildNodes():
            xmlNode.appendChild(childnode)
        
    diffSet = set1 - set2
    print (diffSet)
    for item in diffSet:
        index = list1.index(item)
        childnode = domTree.createElement(item)
        compare_xmltree(childrenlist1[index],  None,  domTree,  childnode)
        if childnode.hasAttributes() or childnode.hasChildNodes():
            xmlNode.appendChild(childnode)
            
    diffSet = set2 - set1
    print (diffSet)
    for item in diffSet:
        index = list2.index(item)
        childnode = domTree.createElement(item)
        compare_xmltree(None,  childrenlist2[index],  domTree,  childnode)
        if childnode.hasAttributes() or childnode.hasChildNodes():
            xmlNode.appendChild(childnode)
            
    return True

if __name__ == "__main__":
    from xml.dom import  minidom
    tree1 = ET.parse(sys.argv[1])
    tree2 = ET.parse(sys.argv[2])
    root1 = tree1.getroot()
    root2 = tree2.getroot()
    if root1.tag != root2.tag:
        print ("两个xml文件的根结点不同，不能比较")
        sys.exit(0)
    #生成domTree
    impl = minidom.getDOMImplementation()
    dom = impl.createDocument(None, None, None)
    
    #i添加节点
    root = dom.createElement(root1.tag)    
    compare_xmltree(root1, root2, dom, root)
    if root.hasAttributes() or root.hasChildNodes():
        dom.appendChild(root)
    
    save2file = ""
    if sys.argv[3][1] == ":":
        save2file = sys.argv[3]
    else:
        save2file = os.getcwd() + "\\" + sys.argv[3]
        
    file = open(save2file,  "w",  encoding = 'UTF8')
    dom.writexml(file,  "","\t",  "\n",  "utf-8")
    file.close()

 

 

 

 

三、win32com模块

win32com模块里面能做的事情很多，但我只调用这个模块对com对象的操作功能，通过dispatch调用WPP(金山办公软件演示)，并通过类似vba的语句来对它进行操作。给出的代码示例是查找一个文档里面是否含有flash对象。

 

import win32com.client
import os, sys


def findFlashFile(fileName):
    wpp = win32com.client.Dispatch('wpp.application')
    wpp.visible = True

    if wpp.Presentations.count > 1:
        print (wpp.Presentations.count)
        for i in range(wpp.Presentations.count):
            wpp.Presentations[0].close
            
    isHasFlash = False        
    pres = wpp.Presentations.Open(fileName, ReadOnly = -1)

    for slide_i in range(1, pres.slides.count + 1):
        slide = pres.slides(slide_i)
        for shape_i in range(1, slide.shapes.count + 1):
            shape = pres.slides(slide_i).shapes(shape_i)
            if shape.Type == 12:
                print (shape.OLEFormat.ClassType)
                shapeType = shape.OLEFormat.ClassType
                print (shapeType.find("ShockwaveFlash.ShockwaveFlash"))
                if shapeType.find("ShockwaveFlash.ShockwaveFlash") > -1:
                    print ("slid{0}, shape{1} has flash".format(slide_i, shape_i))
                    pres.saved = True
                    pres.close
                    return True
    pres.saved = True
    pres.close    
    return isHasFlash
    

 

 

 

四、发送邮件

一直想写一个自动发送邮件的程序，弄了好久，终于搞出来了。使用的是smtp协议。代码好下，从配置文件里面读取邮件服务器及相关信息。

 

import smtplib
from email.mime.multipart import MIMEMultipart
from email.mime.text import MIMEText
import os

def smtpInit(dict_value):
    SMTPserver = dict_value["SMTPserver"]
    s = smtplib.SMTP(SMTPserver)
    s.set_debuglevel(1)
    s.login(dict_value["Account"], dict_value["PassWord"])
    return s
    
def loadFile(fileName, msgRoot):
    att = MIMEText(open(fileName, "rb").read(), 'base64', 'gb2312')
    att['Content-Type'] = 'application/octet-stream'
    att["Content-Disposition"] = 'attachment; filename=%s'%os.path.split(fileName)[1]
    msgRoot.attach(att)

def checkFile(fileName):
    return os.path.exists(fileName)

    
def sendMail(dict_value, func_smtp_init):

    msgRoot = MIMEMultipart('related')
    msgRoot['Subject'] = dict_value["Subject"]
    msgRoot['to'] = dict_value["toaddr"]
    msgRoot['from'] = dict_value["fromaddr"]
    
    file = open(dict_value["Text"])
    str = file.read()
    file.close()
    msgText = MIMEText(str,'html','utf-8')
    msgRoot.attach(msgText)
    os.remove(dict_value["Text"])
    
    list = dict_value["Attachment"].split(";")
    for filename in list:
        filename = filename.strip()
        print (filename)
        if (checkFile(filename) == True):
            loadFile(filename, msgRoot)
    
    s = func_smtp_init(dict_value)
    s.sendmail(dict_value["fromaddr"], dict_value["toaddr"], msgRoot.as_string().encode('utf-8'))
    s.quit()

if __name__ == "__main__":
    import sys
    #打开配置文件
    file = open(sys.argv[1])
    dict_value = dict()
    for line in file.readlines():
        list = line.strip().split(" ")
        dict_value[list[0]] = list[2]
    file.close()
    print (dict_value)
    sendMail(dict_value, smtpInit)

 

SMTPserver : ***.***.com
Account : ******
PassWord : ******
fromaddr : ******
toaddr : ******
Subject : 能过python程序发送邮件
Text : message.txt（将要发送的邮件的内容，存到这个文件里，这里指定路径）
Attachment : deal_result\result_success.xml;deal_result\result_failed.xml（发送的附件的路径）