Python 数据库备份脚本(邮件通知)

手册/FAQ (454) 2015-10-28 10:05:08

今天用Python写了一个数据库备份脚本,备份失败会用邮件的形式通知系统管理员,感觉还行:

  1. #!/usr/bin/python 
  2. #coding:utf-8 
  3.  
  4. import subprocess 
  5. import time 
  6. import os 
  7. import sys 
  8. import sendEmail 
  9. import getip 
  10.  
  11. mail_to_list = ['lihuipeng007@51cto.com',] 
  12.  
  13. def backup(user='root', password='123456', host='localhost', dbname='mysql'): 
  14.     start_time = time.clock() 
  15.     ip = getip.get_ip_address('eth0') 
  16.     today = time.strftime("%Y%m%d", time.localtime()) 
  17.     backup_dir = '/data/dbbackup/%s' % today 
  18.     if not os.path.isdir(backup_dir): 
  19.         os.makedirs(backup_dir) 
  20.     os.chdir(backup_dir) 
  21.     cmd = "/usr/local/mysql/bin/mysqldump --opt -u%s -p%s -h%s %s | gzip > %s-%s-%s.sql.gz" \ 
  22.                 % (user,password,host,dbname,today,ip,dbname) 
  23.     result = subprocess.Popen(cmd, shell=True,stdout=subprocess.PIPE,stderr=subprocess.STDOUT) 
  24.     content = result.stdout.read() 
  25.     if content: 
  26.         print "Dump error at %s" % content 
  27.         subject = "%s - %s backup error" % (ip,dbname) 
  28.         sendEmail.send_mail(mail_to_list,subject,content) 
  29.     end_time = time.clock() 
  30.     use_time = end_time - start_time 
  31.     print "backup use: %s" % use_time 
  32.      
  33. def help(): 
  34.     print '''''Usage: %s dbname'''  % sys.argv[0] 
  35.     sys.exit(1) 
  36.          
  37. if __name__ == "__main__": 
  38.     if len(sys.argv) != 2: 
  39.         help() 
  40.     backup(dbname=sys.argv[1]) 

发送邮件的PY接口:

sendEmail.py

  1. #!/usr/bin/python 
  2. #coding:utf-8 
  3.  
  4. import smtplib 
  5. from email.mime.text import MIMEText 
  6.  
  7. mail_to_list = ['lihuipeng007@51cto.com',] 
  8. mail_host = 'smtp.163.com' 
  9. mail_user = 'lihuipeng007' 
  10. mail_pass = 'xxxxxxx' 
  11. mail_postfix = '163.com' 
  12.  
  13. def send_mail(to_list,subject,content): 
  14.     me = mail_user+"<"+mail_user+"@"+mail_postfix+">" 
  15.     msg = MIMEText(content) 
  16.     msg['Subject'] = subject 
  17.     msg['From'] = me 
  18.     msg['to'] = ";".join(mail_to_list) 
  19.      
  20.     try
  21.         s = smtplib.SMTP() 
  22.         s.connect(mail_host) 
  23.         s.login(mail_user,mail_pass) 
  24.         s.sendmail(me,to_list,msg.as_string()) 
  25.         s.close() 
  26.         return True 
  27.     except Exception,e: 
  28.         print str(e) 
  29.         return False 
  30.      
  31. if __name__ == "__main__": 
  32.     if send_mail(mail_to_list, 'Test for python_mail', "aaaaaaaaaaaaaaa"): 
  33.         print "send success!" 
  34.     else
  35.         print "send fail!"  

获取本机IP地址的PY接口:

getip.py

  1. #!/usr/bin/python 
  2. #coding:utf-8 
  3.  
  4. import socket 
  5. import fcntl 
  6. import struct 
  7. def get_ip_address(ifname): 
  8.     s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) 
  9.     return socket.inet_ntoa(fcntl.ioctl( 
  10.         s.fileno(), 
  11.         0x8915,  # SIOCGIFADDR 
  12.         struct.pack('256s', ifname[:15]) 
  13.     )[20:24]) 
  14.      
  15. if __name__ == "__main__": 
  16.     print get_ip_address('eth0') 
  17.     print get_ip_address('lo') 

 

THE END