python 第三天 编写文件查询、添加、删除
需求如下：
输出：
1
、获取ha记录
2
、增加ha记录
3
、删除ha记录
num
=
raw_input
(
'请输入操作序号：'
)
如果用户输入的
1
：
read
=
raw_input
(
'请输入backend：'
) 如输入：www.oldboy.org
讲配置文件 backend www.oldboy.org 节点下的所有记录获取到，并输入到终端
如果用户输入的
2
：
read
=
raw_input
(
'请输入要新加的记录：'
) 如： {
"backend"
:
"test.oldboy.org"
,
"record"
:{
"server"
:
"100.1.7.9"
,
"weight"
:
20
,
"maxconn"
:
30
}}
# 讲read字符串转换成 字典类型
read_dict
=
json.loads(read)
backend_title
=
read_dict[
'backend'
]
# 去配置文件中找到指定的节点：
# backend test.oldboy.org
# 如果已经存在，
# 则在此节点下添加根据用输入构造出的记录，例如：
server
100.1
.
7.9
100.1
.
7.9
weight
20
maxconn
3000
# 如果不存在，
# 则添加backend节点和记录，例如：
backend test.oldboy.org
server
100.1
.
7.9
100.1
.
7.9
weight
20
maxconn
3000
# （可选）可以再对节点下记录进行判断是否已经存在
如果用户输入的
3
：
read
=
raw_input
(
'请输入要删除的记录：'
) 如： {
"backend"
:
"test.oldboy.org"
,
"record"
:{
"server"
:
"100.1.7.9"
,
"weight"
:
20
,
"maxconn"
:
30
}}
# 讲read字符串转换成 字典类型
read_dict
=
json.loads(read)
backend_title
=
read_dict[
'backend'
]
#去配置文件中找到指定的节点，并在删除指定记录，如：
backend test.oldboy.org
server
100.1
.
7.10
100.1
.
7.10
weight
20
maxconn
3000
server
100.1
.
7.9
100.1
.
7.9
weight
20
maxconn
3000
# 删除掉
# （可选）如果backend下所有的记录都已经被删除，那么将当前 backend test.oldboy.org 也删除掉。
思路如下：
文件处理：
1、查询、、、、判断用户输入，查询关键字，判断是否存在关键字符串，存在打印，不存在则输出
2、增加、、、、判断是否存在backend字段：
1、存在则判断是否存在要添加字段，存在无动作，不存在则添加。
2、判断是否存在要添加的记录，存在无动作，不存在则添加。
3、删除、、、、定位到删除的backend关键字
1、找到相应记录，然后删除，判断删除后此backend字段是否为空，为空则删除backend字段。
代码实现如下：
# -*- coding: utf-8 -*-
__author__ = 'weibinf'
import json #导入模块
import os #导入模块
def cat():
read = raw_input("\033[31m请输入backend： \033[0m").strip()
strr = "backend" + " " + read #取得输入字段
obj = open("ha",'r') #打开方式
i = 0
for item in obj: #取得字段是否存在
if strr.strip() == item.replace("\n",'').strip():
i += 1
if i > 0: #不为0
exitt = False
obj.seek(0) #从文件开头重新读取
while not exitt: #循环读取
line = obj.readline().strip() #去掉本行中的空格
if strr.strip() == line.replace("\n",'').strip(): #判断查询字符串和本行字符串是否相等
print "\033[34m -\033[0m" * 30
print "\033[34m %s \033[0m"%line
obj.seek(obj.tell())
while not exitt:
line = obj.readline()
if "backend" in line:#判断本行是否包含backend字段
exitt = True #包含则退出
else:
print "\033[34m %s \033[0m"%line
else:
print "\033[31m 你输入的字符串不在文本中。。。 \033[0m" #不存在打印
def edit():
read = raw_input("\033[31m请输入要新加的记录： \033[0m").strip() #获取输入字符串
read_dict = json.loads(read) #转换成字典
backend_title = read_dict['backend'] #取出key值
strr = "backend" + " " + str(backend_title) #取出字符串
strr1 = " " * 8 + "server" + " " + str(read_dict['record']['server']) + " " + str(read_dict['record']['server']) + " " + "weight" + " " + str(read_dict['record']['weight']) + " " + "maxconn" + " " + str(read_dict['record']['maxconn'])
with open("ha",'r') as obj , open("haha",'a+') as obj2: #打开两个文件
i = 0
for item in obj:
if strr.strip() == item.replace("\n",'').strip(): #判断字段是否存在
i += 1
qa = obj.tell()
if i == 0: #不存在则重新写入到文件
obj.seek(0)
while True:
line = obj.readline()
aa = obj.tell()
if qa == aa:
obj2.write(line)
obj2.write("\n")
obj2.write('%s\n'%strr)
obj2.write('%s\n'%strr1)
obj2.flush()
break
else:
obj2.write(line)
elif i == 1: #字段存在
exitt = False
obj.seek(0)
while not exitt:
line = obj.readline()
obj2.write(line)
bb = obj.tell()
if strr.strip() == line.replace("\n",'').strip(): #判断字符串和本行是否相等
obj.seek(bb)
while not exitt:
line = obj.readline()
cc = obj.tell()
if strr1.replace("\n",'').replace(" ",'').strip() != line.replace("\n",'').replace(" ",'').strip(): #字符串和本行是否相等
if "backend" in line: #本行包含backend关键字
num = obj.tell() - len(line) - 1
obj2.seek(num)
obj2.write('%s\n'%strr1)
obj2.write(line)
obj.flush()
elif qa == cc: #判断文件是否已经是结尾
obj2.write(line)
obj2.write('%s\n'%strr1)
obj.flush()
exitt = True
else: #其余结果
obj2.write(line)
obj2.flush()
os.rename("ha","ha.bak") #重新命名文件
os.rename("haha","ha")
def dele():
read = raw_input("\033[31m请输入要删除的记录： \033[0m").strip()
read_dict = json.loads(read)
backend_title = read_dict['backend']
strr = "backend" + " " + str(backend_title)
strr1 = " " * 8 + "server" + " " + str(read_dict['record']['server']) + " " + str(read_dict['record']['server']) + " " + "weight" + " " + str(read_dict['record']['weight']) + " " + "maxconn" + " " + str(read_dict['record']['maxconn'])
with open("ha",'r+') as obj , open("hadel",'a+') as obj2: #打开两个文件
i = 0
for item in obj:
if strr.strip() == item.replace("\n",'').strip(): #判断文件中是否包含关键字符串
i += 1
qa = obj.tell()
if i == 0:
print "\033[31m你输入的记录不存在。。。 \033[0m" #不存在输出
elif i == 1: #存在关键字符串
exitt = False #设置全局关键退出字段
obj.seek(0) #回到文件开头
while not exitt:
line = obj.readline()
bb = obj.tell()
if strr.strip() != line.replace("\n",'').strip(): #字符串和本行是否相等
obj2.write(line)
obj2.flush()
else:
if qa == bb : #是否是文件结尾
obj2.write(line)
exitt = True
else:
xs = line
while not exitt:
line = obj.readline()
bb = obj.tell()
if strr1.replace("\n",'').replace(" ",'').strip() != line.replace("\n",'').replace(" ",'').strip() : #本行和关键字符串不相等
if qa == bb:
obj2.write(line)
exitt = True
elif "backend" in line: #本行包含关键字
obj2.write(line)
obj2.flush()
else:
obj2.write(xs)
obj2.write(line)
obj2.flush()
else:
continue
os.rename("ha","hadel.bak")
os.rename("hadel","ha")
if __name__ == '__main__':
print """\033[31m
1、获取ha记录
2、增加ha记录
3、删除ha记录
\033[0m"""
while True:
num = raw_input("\033[31m请输入相应编号：\033[0m").strip()
if num == "1":
cat()
elif num == "2":
edit()
elif num == "3":
dele()
else:
break