python 的RE @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

python 的RE

URL Link //n.sfs.tw/12349

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