利用Python 脚本 格式化文件，可传递两个传递参数
#!/usr/bin/env python
#encoding: utf-8
#author: xiaofangliu
importos
importsys
printos.getcwd()
#os.environ.update({"DJANGO_SETTINGS_MODULE": "djapi.settings"})
#pro_dir = os.getcwd() # 如果放在project目录，就不需要在配置绝对路径了
#sys.path.append('/Users/xiaofangl/Downloads/huasheng/hasan/djapi')
#
os.environ['DJANGO_SETTINGS_MODULE']='djapi.settings.settings'#项目的settings
importdjango
django.setup()
printsys.path
sys.path.append(os.path.dirname(__file__))
fromapp01.modelsimportDnsHost
fromapp01.modelsimportDnsFiles
#print (DnsFiles.objects.all())
"""
执行脚本 并携带两个参数，第一个是ip地址，第二个是文件名,脚本跑完 生成的zone文件多一行回车 手动去掉
将dns zone文件每行格式化
host + '\t' + ' ' + 'IN' + '\t' + ' ' + type + '\t' + ' ' + value
"""
sou_namelist={
'172': ['zoneName',''],
'192': ['zoneName','']
}
dis_namelist=['172_huashenghaoche.com.zone','192_huashenghaoche.com.zone','192_huashenghaoche.net.zone',
'192_huashenghaoche.work.zone']
#执行脚本 并携带两个参数，第一个是ip地址，第二个是文件名
classFormatFiles(object):
def__init__(self):
#title = 'Reset Password' if not title else title
self.source_host=sys.argv[1]
self.source_filename=str(sys.argv[2])
self.res={}
printself.source_filename,self.source_host
#将sou_name按格式生成dis_name
def_gen_filename(self):
ip=self.source_host.split('.')[0]
name=self.source_filename
dis_filename=ip+'_'+name
returndis_filename
#格式化写入，this class not use!
def_wirte_file(self,file_name,data):
try:
withopen(file_name,'w+')asf:
f.write(data)
except:
self.res['status']=False
#loger.error('format_file write false!!')
def_create_files(self,host,name,res,f):
#f = False
iff:
try:
reco=DnsHost.objects.filter(desc=host[0])
foriinreco:
#print i.host, i.desc
f.file_to_host.add(i.id)
except:
res['status']=False
res['message']='_create_files file failed ..'
else:
try:
files=DnsFiles()
files.filename=name
files.save()
reco=DnsHost.objects.filter(desc=host[0])
foriinreco:
#print i.host, i.desc
files.file_to_host.add(i.id)
except:
res['status']=False
res['message']='_create_files file failed ..'
returnres
#同步完文件后 写入数据库
def_write_db_format_file(self):
res={
'status':True,
'message':''
}
if'192'inself.source_host:
host=[]
host.append('lan_dns主机组')
elif'172'inself.source_host:
host=[]
host.append('wan_dns主机组')
f=DnsFiles.objects.filter(filename=self.source_filename)
iff:
is_f=DnsFiles.objects.get(filename=self.source_filename)
res=self._create_files(host,self.source_filename, res, is_f)
else:
res=self._create_files(host,self.source_filename, res, f)
returnres
#读写文件把源文件格式化后写入新文件
def_run_format_file(self):
#为真时的结果 if 判定条件 else 为假时的结果
#self.source_filename if self.source_filename else self.source_filename = '172_huashenghaoche.com.zone'
dis_name=self._gen_filename()
resu=self._write_db_format_file()
print'+++++++++++++++++++++++++++++++'
print'_write_db_format_file', resu
i=1
withopen(self.source_filename,'r')asf:
withopen('../'+dis_name,'w+')asfw:
line=f.readline()
fw.write(line)
whileline:
line=f.readline()
i+=1
ifi<10:
fw.write(line)
elifi>=10:
#print '------------------------------'
one_line=line.strip()
#print 'i:', i, one_line.split(' ')
one_line2=one_line.replace('\t','')
#print 'i:', i, one_line2.split(' ')
one_line3='\t'.join(filter(lambdax: x, one_line2.split('')))
#print 'i:', i, one_line3
iflen(one_line3)>0:
fw.write(one_line3)
fw.write('\n')
printself.source_filename,'---', dis_name
#format_file = FormatFiles()
#format_file._run_format_file()
#返回格式化数据，test
deftest_file(name,dis_filename):
printos.getcwd()
res={'status':True}
i=0
try:
withopen(name,'r')asf:
tmp=[]
withopen(dis_filename,'w')asfw:
line=f.readline()
tmp.append(line)
fw.write(line)
whileline:
line=f.readline()
i+=1
ifi<10:
fw.write(line)
tmp.append(line)
elifi>=10:
#print '------------------------------'
one_line=line.strip()
#print 'i:', i, one_line.split(' ')
one_line2=one_line.replace('\t','')
#print 'i:', i, one_line2.split(' ')
one_line3='\t'.join(filter(lambdax: x, one_line2.split('')))
#print 'i:', i, one_line3
iflen(one_line3)>0:
tmp.append(one_line3)
fw.write(one_line3)
fw.write('\n')
printlen(tmp)
except:
res['status']=False
res['message']='格式化失败-->>test_file'
#logging.info(res)
returnres
defwrite_in_db(host,names):
res={
'status':True,
'message':''
}
print'name', host, names,type(host)
f=DnsFiles.objects.filter(filename=names)
print'write_in_db', f
#res = self._create_files(host, self.source_filename, res, f)
iff:
try:
reco=DnsHost.objects.filter(desc=host)
foriinreco:
printi.host, i.desc
is_f=DnsFiles.objects.get(filename=names)
is_f.file_to_host.add(i.id)
except:
res['status']=False
res['message']='_create_files file failed ..'
else:
try:
files=DnsFiles()
files.filename=names
files.save()
reco=DnsHost.objects.filter(desc=host)
foriinreco:
printi.host, i.desc
files.file_to_host.add(i.id)
except:
res['status']=False
res['message']='_create_files file failed ..'
returnres
if__name__=="__main__":
format_file=FormatFiles()
format_file._run_format_file()
#test_file('../172_huashenghaoche.com.zone')
#test = FormatFiles()