python脚本实战1：rsync客户端推送

前段时间想有空写个rsync client脚本 需要收集一下rsync执行结果，正好昨天部门其他team有这个需求，今天抽时间写了一个。

 1 #!/bin/env python26
 2 #-*- coding: utf-8 -*-
 3 '''
 4 Created on 2012-11-9
 5 
 6 @author: Jin
 7 
 8 '''
 9 
10 import os
11 import sys
12 import subprocess
13 import logging
14 
15 #rsync config
16 localdir  = '/tmp/testdir/'
17 rsync_exe = '/usr/bin/rsync '
18 pwdfile   = '/etc/rsync/rsync.passwd'
19 argv      = '-vzrtopg --delete --password-file='+pwdfile
20 USER      = 'rsync_store'
21 IP        = '@1.1.1.1'
22 MODULE    = '::store'
23 
24 #vars
25 homedir   = '/home/rsyncstore/'
26 logfile   = os.path.basename(sys.argv[0])+'.log'
27 logpath   = homedir+logfile
28 
29 #logconfig
30 logging.basicConfig(level=logging.DEBUG,
31                     format='%(asctime)s %(name)-12s %(levelname)-8s %(message)s',
32                     datefmt='%Y-%m-%d %H:%M',
33                     filename=logpath,
34                     filemode='a')
35 
36 console = logging.StreamHandler()
37 console.setLevel(logging.INFO)
38 formatter = logging.Formatter('%(asctime)s %(name)-12s %(levelname)-8s %(message)s')
39 console.setFormatter(formatter)
40 logging.getLogger('').addHandler(console)
41 
42 
43 def is_run(processname):
44     '''Get processname status'''
45     cmd='/usr/bin/pgrep '+ processname+' > /dev/null 2>&1'
46     try:
47         #pstat=subprocess.Popen(cmd,shell=True,stdin=subprocess.PIPE)
48         pstat=subprocess.Popen(cmd,shell=True)
49     except StandardError,e:
50         logging.error("Get run status failed:: %d (%s) ,exit run!" % (e.errno, e.strerror))
51         sys.exit(1)
52     else:
53         pstat.wait()
54         if pstat.returncode == 0:
55             return True
56         else:
57             return False
58         
59 
60 def rsync_store():
61     '''Put localfile to rsync server'''
62     cmd=rsync_exe+argv+' '+localdir+' '+USER+IP+MODULE
63     if is_run('rsync'):
64         logging.warning("rsync is running!")
65     else:
66         print 'to run'
67         try:
68             pstat=subprocess.Popen(cmd,shell=True,stderr=subprocess.PIPE,stdout=subprocess.PIPE)
69         except StandardError,e: 
70             logging.error("Run script failed:: %d (%s) ,exit run!" % (e.errno, e.strerror))
71             sys.exit(1)
72         else:
73             pstat.wait()
74             if pstat.returncode == 0:
75                 logging.info("Run script Successful,exit code is %s" % pstat.returncode)
76                 lines=pstat.stdout.readlines()
77                 dellist=[]
78                 rsync_dict={}
79                 for line in lines:
80                     if 'deleting' in line:
81                         dellist.append(line.split()[1])
82                 delcount=len(dellist)
83                 addlist=[i.rstrip('\n') for i in lines[delcount+2:-3]]
84                 addcount=len(addlist)
85                 speed=' '.join(lines[-2].split()[-2:])
86                 logging.info("Result,delcount:%d,dellist:%s,addcount:%d,addlist:%s,speed:%s" % (delcount,dellist,addcount,addlist,speed))
87                     
88             else:
89                 logging.error("Run script End,exit code is %s,With reason <%s>" % (pstat.returncode,pstat.stderr.read().rstrip('\n')))
90                 
91 
92 if __name__=='__main__':
93     rsync_store()