Initial YakPanel commit

class/tomcat.py

@@ -0,0 +1,117 @@
+#!/usr/bin/env python
+#coding:utf-8
+# +-------------------------------------------------------------------
+# | YakPanel
+# +-------------------------------------------------------------------
+# | Copyright (c) 2015-2017 YakPanel(www.yakpanel.com) All rights reserved.
+# +-------------------------------------------------------------------
+# | Author: hwliang <hwl@yakpanel.com>
+# +-------------------------------------------------------------------
+
+#------------------------------
+# Tomcat管理类
+#------------------------------
+from xml.etree.ElementTree import ElementTree,Element
+import os
+class tomcat:
+    __TREE = None
+    __ENGINE = None
+    __ROOT = None
+    __CONF_FILE = '/www/server/tomcat/conf/server.xml';
+    #打开配置文件
+    def __init__(self):
+        if self.__TREE: return;
+        self.__TREE = ElementTree();
+        self.__TREE.parse(self.__CONF_FILE);
+        self.__ROOT = self.__TREE.getroot();
+        self.__ENGINE = self.__TREE.findall('Service/Engine')[0];
+        
+    #获取虚拟主机列表
+    def GetVhosts(self):
+        try:
+            Hosts = self.__ENGINE.getchildren()
+        except:
+            Hosts = list(self.__ENGINE)
+        data = []
+        for host in Hosts:
+            if host.tag != 'Host': continue;
+            tmp = host.attrib
+            ch = host.getchildren();
+            tmp['item'] = {}
+            for c in ch:
+                tmp['item'][c.tag] = c.attrib;
+            data.append(tmp);
+        return data;
+
+    #添加虚拟主机
+    def AddVhost(self,path,domain):
+        if self.GetVhost(domain): return False;
+        if not os.path.exists(path): return False;
+        attr = {"autoDeploy":"true","name":domain,"unpackWARs":"true","xmlNamespaceAware":"false","xmlValidation":"false"}
+        Host = Element("Host", attr);
+        attr = {"docBase":path,"path":"","reloadable":"true","crossContext":"true",}
+        Context = Element("Context", attr);
+        Host.append(Context);
+        self.__ENGINE.append(Host);
+        self.Save();
+        return True;
+    
+    #删除虚拟主机
+    def DelVhost(self,name):
+        host = self.GetVhost(name);
+        if not host: return False
+        self.__ENGINE.remove(host);
+        self.Save();
+        return True;
+    
+    #获取指定虚拟主机
+    def GetVhost(self,name):
+        try:
+            Hosts = self.__ENGINE.getchildren()
+        except:
+            Hosts = list(self.__ENGINE)
+        for host in Hosts:
+            if host.tag != 'Host': continue;
+            if host.attrib['name'] == name:
+                return host
+        return None;
+    
+    #修改根目录
+    def SetPath(self,name,path):
+        if not os.path.exists(path): return False;
+        host = self.GetVhost(name);
+        if not host: return False
+        #host.attrib['appBase'] = path;
+        host.getchildren()[0].attrib['docBase'] = path;
+        self.Save();
+        return True;
+    
+    #修改虚拟主机属性
+    def SetVhost(self,name,key,value):
+        host = self.GetVhost(name);
+        if not host: return False
+        host.attrib[key] = value;
+        self.Save();
+        return True
+    
+    #保存配置
+    def Save(self):
+        self.format(self.__ROOT);
+        self.__TREE.write(self.__CONF_FILE,'utf-8');
+    
+    #整理配置文件格式
+    def format(self,em,level = 0):
+        i = "\n" + level*"  "
+        if len(em):
+            if not em.text or not em.text.strip():
+                em.text = i + "  "
+            for e in em:
+                self.format(e, level+1)
+            if not e.tail or not e.tail.strip():
+                e.tail = i
+        if level and (not em.tail or not em.tail.strip()):
+            em.tail = i
+if __name__ == '__main__':
+    tom = tomcat();
+    print(tom.DelVhost('w1.hao.com'));
+