python_project/changexml.py at master · silenceGuo/python_project · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
#!/usr/bin/env python
# -*-coding:utf-8-*-
# @Author : gzq
# @date   : 2018/5/14 0014 10:33
# @file   : changexml.py

import xml.dom.minidom
#import dom
import os
import sys
import codecs
# 读取xml 配置文件
path = "D:\logs\\"
changfile = "ehcache.xml"
#fileType = "ehcache.xml"
changxmlname = "accessTokenCache"
list = []
changDict = {
    "eternal": "false",
    "timeToIdleSeconds": "3800",
    "timeToLiveSeconds": "3800",
    "timeToLiveSeconds12": "4800",
    "overflowToDisk": "false"}

def listDir(path):
    #list = []
    lsdir = os.listdir(path)
    dirs = [i for i in lsdir if os.path.isdir(os.path.join(path, i))]
    files = [i for i in lsdir if os.path.isfile(os.path.join(path, i))]
    if files:
        for f in files:
            if f == changfile:
                list.append(os.path.join(path, f))
    if dirs:
        for d in dirs:
            listDir(os.path.join(path, d))
    return list

def readXml(serverName):
    xmlPath = os.path.join(deploymentTomcatName(serverName), "conf/server.xml")
    domtree = xml.dom.minidom.parse(xmlPath)
    collection = domtree.documentElement
    service = collection.getElementsByTagName("Service")
    shutdown_port = str(collection.getAttribute("port"))  # shutdown port
    for i in service:
        connector = i.getElementsByTagName("Connector")
        http_port = str(connector[0].getAttribute("port"))  # http port
        ajp_port = str(connector[1].getAttribute("port"))  # ajp port
        # URIEconding= str(connector[0].getAttribute("URIEconding"))  # http port
    return {serverName: {"shutdown_port": shutdown_port, "http_port": http_port, "ajp_port": ajp_port}}

# 修改xml 配置文件
def changeXml(xmlpath,changxmlname,):
    print xmlpath
    domtree = xml.dom.minidom.parse(xmlpath)
    #print domtree
    collection = domtree.documentElement
    #print collection
    caches = collection.getElementsByTagName("cache")
    for cache in caches:
        #if cache.hasAttribute("name") :
            if cache.getAttribute("name") == changxmlname:
                print"修改的块名:", cache.getAttribute("name")
                for k, v in changDict.iteritems():
                    if cache.hasAttribute(k):
                        print"原属性值：", cache.getAttribute(k)
                        print "修改属性:%s为值:%s" % (k,v)
                        cache.setAttribute(k,v)
                    else:
                        print "not attribute: %s" % k
    #collection.setAttribute("port", shutdown_port)  # shutdown port
    # URIEconding = "UTF-8"
    # useBodyEncodingForURI = "true"
    # for i in service:
    #     #print i
    #     connector = i.getElementsByTagName("Connector")
    #     appdeploy = i.getElementsByTagName("Host")
    #     appdeploy[0].setAttribute("appBase", deploydir) #部署目录 默认为webapps
    #     connector[0].setAttribute("port", http_port)  # http port
    #     #connector[0].setAttribute("URIEnconding", "UTF-8")  # http port
    #     #connector[0].setAttribute("proxyPort", "443")  #proxyPort
    #     # connector[0].setAttribute("useBodyEncodingForURI", "true")  # http port
    #     #connector[1].setAttribute("port", ajp_port)  # ajp port

    outfile = file(xmlpath, 'w')
    write = codecs.lookup("utf-8")[3](outfile)
    # domtree.writexml(write, addindent=" ",newl=' ', encoding='utf-8')
    domtree.writexml(write, addindent=" ", encoding='utf-8')
    write.close()

def Indent(dom, node, indent = 0):
    # Copy child list because it will change soon
    children = node.childNodes[:]
    # Main node doesn’t need to be indented
    if indent:
        text = dom.createTextNode("/n" + "/t" * indent)
        node.parentNode.insertBefore(text, node)
    if children:
        # Append newline after last child, except for text nodes
        if children[-1].nodeType == node.ELEMENT_NODE:
            text = dom.createTextNode("/n" + "/t" * indent)
            node.appendChild(text)
        # Indent children which are elements
        for n in children:
            if n.nodeType == node.ELEMENT_NODE:
                Indent(dom, n, indent + 1)


for xmlfile in listDir(path):
    changeXml(xmlfile,changxmlname)

    # domtree = xml.dom.minidom.parse(xmlpath)
    # domcopy=domtree.cloneNode(True )
    # Indent(domcopy, domcopy.documentElement)
    # f = file(xmlfile, "wb")
    # writer = codecs.lookup("utf-8")[3](f)
    # domcopy.writexml(writer, encoding="utf-8")
    # domcopy.unlink()