利用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()