import os
import re
from typing import List
def replace(a,a_swp):
a_bak=a+'.bak'
os.rename(a,a_bak)
os.rename(a_swp,a)
os.remove(a_bak)
def fetch(domain):
domain=domain
if not domain:
domain=input('please input your domain:').strip()
result=[]
with open('remap.config','r') as read_f:
for read_line in read_f:
tmpdata=read_line.strip()
if re.search(domain,tmpdata) :
result.append(tmpdata)
print(tmpdata)
else:
continue
if not result:
print("not found")
return result
def change():
data=input('please input your data:').strip()
data_split=data.split() #分割输入数据取domain,数据格式为map http://domain http://srcip ，则分割后取1下标的数据
olddata=fetch(data_split[1])
if not olddata:
with open('remap.config', 'r') as read_f,open('remap.config.swp', 'w') as write_f:
for read_line in read_f:
write_f.write(read_line)
write_f.write(data.strip('"')+'\n')
else:
with open('remap.config', 'r') as read_f,open('remap.config.swp', 'w') as write_f:
for read_line in read_f:
tmpdata = read_line.strip()
if re.search(data_split[1], tmpdata):
write_f.write(data.strip('"')+'\n')
else:
write_f.write(read_line)
replace("remap.config","remap.config.swp")
def add():
data=input('please input your data:').strip()
with open('remap.config', 'r') as read_f, open('remap.config.swp', 'w') as write_f:
for read_line in read_f:
write_f.write(read_line)
write_f.write(data.strip('"') + '\n')
replace("remap.config", "remap.config.swp")
def delete():
data = input('please input your data:').strip()
olddata=fetch(data)
if olddata:
with open('remap.config', 'r') as read_f, open('remap.config.swp', 'w') as write_f:
for read_line in read_f:
tmpdata = read_line.strip()
if re.search(data, tmpdata):
continue
else:
write_f.write(read_line)
replace("remap.config", "remap.config.swp")
else:
print("needn't delete")
list='''what do you want ?:
1.fetch
2.change
3.add
4.delete
5.exit
'''
list_dic={
1:fetch,
2:change,
3:add,
4:delete,
5:exit
}
while True:
print(list)
choice=input('you choice>>: ').strip()
if not choice or not choice.isdigit():
continue
if choice == '5':
break
list_dic[int(choice)]()
如图：remap.config格式如下
map http://baidu.com/ http://1.1.1.17/ map http://www.baidu.com/ http://1.1.1.17/ regex_map http://[a-z0-9].baidu.com/ http://1.1.1.11/ map http://285166825.cdn.baidu.com/ http://1.1.1.11/ map http://1731178188.cdn.baidu.com/ http://1.1.1.11/ map http://907607712.cdn.baidu.com/ http://1.1.1.11/ map http://jmbaihacker.cdn.baidu.com/ http://1.1.1.11/ map http://13991833.cdn.baidu.com/ http://1.1.1.11/ map http://89802850.cdn.baidu.com/ http://1.1.1.11/ map http://jbsdnsn.cdn.baidu.com/ http://1.1.1.11/ map http://myl666.cdn.baidu.com/ http://1.1.1.11/ map http://cdn-w.cdn.baidu.com/ http://1.1.1.11/
本程序意在熟悉python的函数、文件处理、控制模块功能
个人思路历程以及遇到的问题：
定义四个功能模块，add()等用pass占位
打印询问菜单，为了简洁用'''形式
用while true 循环，发现没退出功能，返回列表，增加一个退出选项5
用choice变量保存input输入的值，发现无法直接从原有询问列表list提取对应的操作选项，于是使用一个字典list_dic保存选项与操作的对应关系
发现用户可能不输入数字，于是用str.isdigit()判断客户输入，非数字则跳出本次循环
开始处理查询功能fetch，发现字符串匹配判断需要用到强大的re模块，此处我用
-
if re.search(domain,tmpdata):
意为：如果domain在tmpdata中，则返回匹配对象，加上if则可判断对象是否存在，也就是有没有tmpdata里面有没有匹配到domain 更多参考：https://docs.python.org/zh-cn/3/library/re.html
处理change功能，发现必先查询，因为我这里默认限制修改功能数据格式为"map http://domain http://srcip“格式，而fetch功能默认只能查找domain，则对change功能的输入用str.split()进行分割提取domain，此处发现，input输入带空格的字符串必须用引号引起来，否则报错
对fetch功能进行修改：1.如果有参数值则查找的值（即data变量）为参数值，否则输入一个值 2.构建一个空列表result，若查找到值，则用append()方法追加，最后return返回
文件写入时发现没换行，用+'\n'形式解决
write_f.write(data.strip('"')+'\n')
12.从change()功能模块中摘取对应功能代码略加修改生成add和delete模块
13.发现文件覆盖功能多次使用，提取出来做成replace函数调用