autotoy

kid.xiyang

无聊给外包公司写了个邮件接口,外包人员可通过邮件对本公司内网服务器进行一些检查及操作,保留了三个较普遍的需求.
各位不要见笑

1. #!/usr/bin/python
   
2. 
   
3. import sys
   
4. import email
   
5. import pxssh
   
6. import string
   
7. import poplib
   
8. import smtplib
   
9. import threading
   
10. import subprocess
    
11. 
    
12. SERVER = 'demo'
    
13. USER = 'demo'
    
14. PASSWD = 'demo'
    
15. 
    
16. MAX_THREAD = 100
    
17. sem=threading.BoundedSemaphore(MAX_THREAD)
    
18. 
    
19. class GetMail(object):
    
20. 
    
21.         def __init__(self, popServer, user, passwd):
    
22. 
    
23.                 self.popServer = popServer
    
24.                 self.user = user
    
25.                 self.passwd = passwd
    
26. 
    
27.         def doit(self):
    
28.                 result = { 'Msgs':0 }
    
29. 
    
30.                 popServer = poplib.POP3_SSL(self.popServer)
    
31.                 popServer.user(self.user)
    
32.                 popServer.pass_(self.passwd)
    
33. 
    
34.                 result['Msgs'] = popServer.stat()[0]
    
35. 
    
36.                 if result['Msgs'] == 0:
    
37.                         return result
    
38.                         sys.exit()
    
39. 
    
40.                 message = popServer.retr(result['Msgs'])[1]
    
41.                 mail = email.message_from_string(string.join(message,'\n'))
    
42.                 result['Subject'] = email.Header.decode_header(mail['subject'])[0][0]
    
43.                 result['From'] = email.Header.decode_header(mail['from'])[0][0]
    
44.                 result['To'] = email.Header.decode_header(mail['To'])[0][0]
    
45.                 result['Body'] = mail.get_payload(decode=True)
    
46. 
    
47.                 popServer.dele(result['Msgs'])
    
48. 
    
49.                 popServer.quit()
    
50.                 return result
    
51. 
    
52. class SendMail:
    
53. 
    
54.         def __init__(self, smtpServer, user, passwd):
    
55.                 self.smtpServer = smtpServer
    
56.                 self.user = user
    
57.                 self.passwd = passwd
    
58. 
    
59.         def doit(self, fromAddr, toAddr, subject, msg):
    
60.                 smtpServer = smtplib.SMTP(self.smtpServer)
    
61.                 smtpServer.ehlo()
    
62.                 smtpServer.starttls()
    
63.                 smtpServer.ehlo()
    
64.                 smtpServer.login(self.user,self.passwd)
    
65.                 body=string.join((
    
66.                         "FROM: %s" %fromAddr,
    
67.                         "TO: %s" %toAddr,
    
68.                         "Subject: %s" %subject,
    
69.                         "",
    
70.                         msg),"\n")
    
71.                 smtpServer.sendmail(fromAddr,toAddr,body)
    
72.                 smtpServer.quit()
    
73. 
    
74. 
    
75. def iLOCheck(ip,l):
    
76.         '''Check ilo connectivity'''
    
77. 
    
78.         ILOPING = '/home/wangruoyan/ping.sh '
    
79. 
    
80.         ret = subprocess.call("%s %s" % (ILOPING, ip),shell=True,stdout=open('/dev/null', 'w'),stderr=subprocess.STDOUT)
    
81.         if ret == 0:
    
82.                 truth[ip] = 'Alive'
    
83.         elif ret == 1:
    
84.                 truth[ip] = 'Dead'
    
85.         else:
    
86.                 truth[ip] = 'Error'
    
87. 
    
88.         sem.release()
    
89. 
    
90. 
    
91. def iLOConfig(ip ,l):
    
92.         '''Config server ilo'''
    
93. 
    
94.         result = 'Configuration done.'
    
95. 
    
96.         REMOTEUSER = 'user'
    
97.         REMOTPASSWD = 'password'
    
98. 
    
99. 
    
100.         GETSCRIPT = 'wget -O /tmp/ilodrac-all.sh ftp://10.23.247.252/home/bakup/auto-scripts/ilodrac-all.sh > /dev/null'
     
101.         RUNSCRIPT = 'sh /tmp/ilodrac-all.sh'
     
102. 
     
103.         try:
     
104.                 ssh = pxssh.pxssh()
     
105.                 ssh.login(ip, REMOTEUSER, REMOTPASSWD)
     
106.                 ssh.prompt()
     
107.                 ssh.sendline(GETSCRIPT)
     
108.                 ssh.prompt()
     
109.                 ssh.sendline(RUNSCRIPT)
     
110.                 ssh.prompt(timeout=200)
     
111.                 ssh.logout()
     
112.         except pxssh.ExceptionPxssh, err:
     
113.                 result= str(err)
     
114. 
     
115.         truth[ip] = result
     
116.         sem.release()
     
117. 
     
118. 
     
119. def NodeCheck(ip,l):
     
120.         '''Check node connectivity'''
     
121. 
     
122.         ret = subprocess.call("ping -c 4 %s" % (ip),shell=True,stdout=open('/dev/null', 'w'),stderr=subprocess.STDOUT)
     
123. 
     
124.         if ret == 0:
     
125.                 truth[ip] = 'Alive'
     
126.         elif ret == 1:
     
127.                 truth[ip] = 'Dead'
     
128.         else:
     
129.                 truth[ip] = 'Error'
     
130. 
     
131.         sem.release()
     
132. 
     
133. 
     
134. if __name__ == "__main__":
     
135. 
     
136.         while (True):
     
137. 
     
138.                 ErrString = '''Error request types,only provide:
     
139.                                 \tILOCHECK,NODECHECK,ILOCONFIG'''
     
140. 
     
141.                 result = ''
     
142.                 todo = list()
     
143.                 truth = dict()
     
144. 
     
145.                 Get = GetMail(SERVER, USER, PASSWD)
     
146.                 Info = Get.doit()
     
147.                 if Info['Msgs'] == 0:
     
148.                         sys.exit()
     
149. 
     
150.                 if Info['Subject'] == 'ILOCHECK':
     
151.                         for ip in Info['Body'].split('\n'):
     
152.                                 sem.acquire()
     
153.                                 th = threading.Thread(target=iLOCheck, args=(ip,'J'))
     
154.                                 todo.append(th)
     
155.                                 th.start()
     
156. 
     
157.                         for t in todo:
     
158.                                 t.join()
     
159. 
     
160.                         for key in truth.keys():
     
161.                                 result = result + "%s\t\t%s\n" % (key, truth[key])
     
162. 
     
163.                 elif Info['Subject'] == 'NODECHECK':
     
164.                         for ip in Info['Body'].split('\n'):
     
165.                                 sem.acquire()
     
166.                                 th = threading.Thread(target=NodeCheck, args=(ip,'J'))
     
167.                                 todo.append(th)
     
168.                                 th.start()
     
169. 
     
170.                         for t in todo:
     
171.                                 t.join()
     
172. 
     
173.                         for key in truth.keys():
     
174.                                 result = result + "%s\t\t%s\n" % (key, truth[key])
     
175. 
     
176.                 elif Info['Subject'] == 'ILOCONFIG':
     
177. 
     
178.                         for ip in Info['Body'].split('\n'):
     
179.                                 sem.acquire()
     
180.                                 th = threading.Thread(target=iLOConfig, args=(ip,'J'))
     
181.                                 todo.append(th)
     
182.                                 th.start()
     
183. 
     
184.                         for t in todo:
     
185.                                 t.join()
     
186. 
     
187.                         for key in truth.keys():
     
188.                                 result = result + "%s\t\t%s\n" % (key, truth[key])
     
189.                 else:
     
190.                         result = ErrString
     
191. 
     
192.                 truth.clear()
     
193. 
     
194.                 Info['From'], Info['To'] = Info['To'], Info['From']
     
195. 
     
196.                 Send = SendMail(SERVER, USER, PASSWD)
     
197.                 Send.doit(Info['From'], Info['To'], 'Autoreply: '+Info['Subject'], result)
     
198. 
     
199.                 if Info['Msgs'] == 1:
     
200.                         break
     

复制代码