python 的RE @igogo

[無邊框]

319

python 的RE

“

python re

„

分享此文連結 //n.sfs.tw/12349

分享連結 python 的RE@igogo

(文章歡迎轉載，務必尊重版權註明連結來源)

2018-04-24 15:03:38 最後編修

2018-04-20 11:29:47 By igogo

有多個記錄檔需要更動, 其中記錄檔主要內容長這樣

$ORIGIN 2.5.8.8.2.0.1.0.0.2.ip6.arpa.
$TTL 86400   ; 24 hours
@ SOA   dns.tc.edu.tw. nc-net.tc.edu.tw. (
   2018030603 ; serial
   36000   ; refresh (10 hours)
   3600   ; retry (1 hours)
   36000   ; expire (10 hours)
   43200   ; minimum (12 hours)
)
@ NS   dns.tc.edu.tw.
@ NS   dns2.tc.edu.tw.
@ NS   dns3.tc.edu.tw.

$ORIGIN 2.5.8.8.2.0.1.0.0.2.ip6.arpa.
0.0   NS   dns.tc.edu.tw.
0.0   NS   dns2.tc.edu.tw.
0.0   NS   dns3.tc.edu.tw.
3.2   NS   dns.hpes.tc.edu.tw.
4.2   NS   dns.kkes.tc.edu.tw.

0.0   NS   dns.tc.edu.tw.
0.0   NS   dns2.tc.edu.tw.
0.0   NS   dns3.tc.edu.tw.
3.2   NS   dns.hpes.tc.edu.tw.
4.2   NS   dns.kkes.tc.edu.tw.

想要改成

0.0   NS   dns.tc.edu.tw.
0.0   NS   dns2.tc.edu.tw.
0.0   NS   dns3.tc.edu.tw.
3.2   NS   dns.tc.edu.tw.
3.2   NS   dns2.tc.edu.tw.
3.2   NS   dns3.tc.edu.tw.
4.2   NS   dns.tc.edu.tw.
4.2   NS   dns2.tc.edu.tw.
4.2   NS   dns3.tc.edu.tw.

用RE來比對的原則如下

讀取每一行為line , 不符合蒐集符合加工

m = re.match("^[\d\w]\.[\d\w][\s|\t]+NS", line)

開頭第一個字元為數字或英文, 第二個為 . 第三個是數字或英文[\s|\t]+NS

NS前有可能是tab or space

前三行不動數1,2,3, 其它送到nsfactory加工

import os
import re

files = (os.listdir("files"))

def nsfactory(line):
content = ""

for i in range(3):
if (i == 0):
content += "{}\tdns.tc.edu.tw.\n".format(line)
if (i == 1):
content += "{}\tdns2.tc.edu.tw.\n".format(line)
if (i == 2):
content += "{}\tdns3.tc.edu.tw.\n".format(line)
return content

for file in files:
filename = "files/{}".format(file)
content = ""
count = 0
repeat = 0

with open(filename) as fp:
for line in fp:
m = re.match("^[\d\w]\.[\d\w][\s|\t]+NS", line)
if (m):
count += 1
if (count > 3):
# print(m.group())
content += nsfactory(m.group())
else:
content += line

else:
# print(line)
content += line

filename = "ok/{}".format(file)
with open(filename, "w") as fw:
fw.write(content)

pythonre.zip

END