#!/usr/bin/env python
#-*- coding:UTF-8 -*-
"""
@Item : Rsync Backup
@Author : Villiam Sheng
@Group : Linux Group
@Date : 2012-08-13
@Funtion:
rsync_conf: Configuration rsync server, and the server
#res[os.path.join(root,fpath)] = int((time.time() - os.stat(os.path.join(root,fpath)).st_ctime) / 86910)
#int((time.time() - os.stat(os.path.join(root,fpath)).st_ctime) / 86910)
#try:
# res[os.path.join(root,fpath)] = time.strptime((fpath.split('_')[2]),'%Y%m%d%H%M%S').tm_yday #截取文件名,得到时间天数,
#except Exception,e:
# print e
#res[os.path.join(root,fpath)] = int((time.time() - os.stat(os.path.join(root,fpath)).st_ctime) / 86910)
Central idea:
"""
import os,sys,time,re,socket,threading,json,base64,traceback,ConfigParser,fcntl,struct
from rsync_log import rsync_log
from rsync_post import rsync_post
from statvfs import F_BLOCKS,F_BAVAIL,F_BSIZE
pcg = 0
""" 生成斐波那契数列"""
lists = []
a,b = 0,1
while b <= 365:
a,b = b ,a+b
lists.append(b)
class rsync_thread(threading.Thread):
def __init__(self,path):
threading.Thread.__init__(self)
self.log = rsync_log()
self.path = path
""" 计算当前磁盘的使用百分比"""
def disk(self):
try:
vfs = os.statvfs(self.path)
disk_full = int(vfs[F_BLOCKS]*vfs[F_BSIZE]/1024/1024/1024)
disk_free = int(vfs[F_BAVAIL]*vfs[F_BSIZE]/1024/1024/1024)
return '%.1f'%(float(disk_full - disk_free) / disk_full * 100)
except:
self.log.log_info('rsync_info.err','dfile.disk',traceback.print_exc())
return traceback.print_exc()
def run(self):
global pcg
old_df = [] # 上一年的删除历史文件
new_df = [] # 今年的删除历史文件
sf = [] # 保留的历史文件
res = {} # 所有文件的天数及文件的路径
rs = 0 # 删除文件的总和
size = [] # 获取删除文件的大小
msize = [] # 今天备份所有文件的大小
tday_size = [] # 今天备份文件的大小
ms = 0 # 今天备份文件的总和
year = time.localtime().tm_year
""" 得到文件的天数，以文件名作为key，天数作为value """
for root,dirs,files in os.walk(self.path):
for fpath in files:
res[os.path.join(root,fpath)] = time.localtime(os.stat(os.path.join(root,fpath)).st_ctime).tm_yday
""" 判断文件的天数是否符合斐波那契数列，符合条件append到sf列表中,不符合append df列表中 """
for v,k in res.items():
if k in lists:
sf.append(k)
self.log.log_info('log_info.save','dfile.rsync_thread','%s:::%s'%(v,k))
elif k not in lists:
if year != time.localtime(os.stat(v).st_ctime).tm_year:
old_df.append({v:k})
else:
new_df.append({v:k})
"""
try:
for s in range(len(new_df)):
for f,k in new_df[s].items():
tday_size.append(k)
if max({}.fromkeys(tday_size).keys()) == k:
msize.append(os.path.getsize(f))
except:
pass
"""
c = []
pcg = float(self.disk())
""" 判断今天是否有新的文件备份，在删除的列表中删除最后一天的数据，但必须保证磁盘的使用百分比大于 %55 """
if time.localtime().tm_yday in res.values():
if len(old_df) != 0:
for s in range(len(old_df)):
for f,k in old_df[s].items():
c.append(k)
for s in range(len(old_df)):
for f,k in old_df[s].items():
if min({}.fromkeys(c).keys()) == k and pcg > 91:
size.append(os.path.getsize(f))
os.system('rm -frv %s' %f)
self.log.log_info('log_info.delete','remove cmd','rm -frv %s %s'%(f,k))
elif pcg <= 91:
break
pcg = float(self.disk())
elif len(new_df) != 0:
for s in range(len(new_df)):
for f,k in new_df[s].items():
c.append(k)
for s in range(len(new_df)):
for f,k in new_df[s].items():
if min({}.fromkeys(c).keys()) == k and pcg > 91:
size.append(os.path.getsize(f))
os.system('rm -frv %s' %f)
self.log.log_info('log_info.delete','remove cmd','rm -frv %s %s'%(f,k))
elif pcg <= 91:
break
pcg = float(self.disk())
for s in size:
rs += s
#for m in msize:
# ms += m
self.log.log_info('log_info.delete','Disk release %s %s MB'%(self.path,rs /1024/1024),'Disk append %s %s MB'%(self.path,ms /1024/1024))
else:
self.log.log_info('log_info.delete','Disk files ',' %s No update file' %self.path)
sys.exit()
class rsync_dfile(object):
def __init__(self):
self.log = rsync_log()
self.rsync_post = rsync_post()
def work(self):
fp = open('/proc/mounts','r')
m_info=fp.readlines()
fp.close()
data = {}
sections = []
for i in m_info:
if i.find('data=ordered') != -1 or i.find('mfs') != -1 or i.find('nfs') != -1:
if os.path.ismount(str(i.split()[1])):
if str(i.split()[1]) != '/':
if str(i.split()[1]) != '/root':
if str(i.split()[1]) != '/var':
if len(i.split()[1]) != 1:
if not i.find('sunrpc') != -1:
rs_thread = rsync_thread(i.split()[1])
rs_thread.start()
while threading.active_count() > 1:
time.sleep(1)
conf = ConfigParser.ConfigParser()
conf.read('/etc/rsyncd.conf')
try:
for i in conf.sections():
if i != 'global':
sections.append(i)
for i in sections:
vfs = os.statvfs(conf.get(i,'path'))
disk_full = int(vfs[F_BLOCKS]*vfs[F_BSIZE]/1024/1024/1024)
disk_free = int(vfs[F_BAVAIL]*vfs[F_BSIZE]/1024/1024/1024)
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
try:
ip = socket.inet_ntoa(fcntl.ioctl(s.fileno(),0x8915,struct.pack('24s','eth0'))[20:24])
except:
ip = socket.inet_ntoa(fcntl.ioctl(s.fileno(),0x8915,struct.pack('24s','eth1'))[20:24])
t_info={'flag':0,'store_ip':ip,'store_module':i,'store_path':conf.get(i,'path'),'disk_full':disk_full,'disk_free':disk_free,'action':'rsync_renew'}
data['param'] = base64.b64encode(json.dumps(t_info))
self.rsync_post.work(data)
self.log.log_info('rsync_info.err','dfile.work',t_info)
except Exception,e:
t_info={'flag':1,'store_ip':ip,'store_module':i,'store_path':conf.get(i,'path'),'disk_full':disk_full,'disk_free':disk_free,'action':'rsync_renew'}
data['param'] = base64.b64encode(json.dumps(t_info))
self.rsync_post.work(data)
self.log.log_info('rsync_info.err','dfile.work',e)
if __name__ == "__main__":
rs = rsync_dfile()
while True:
rs.work()
if pcg <= 91:
break
