Python学习day3作业
#!/usr/bin/env python # _*_coding:utf-8_*_'''
* Created on 2016/11/7 21:24.
* @author: Chinge_Yang.
'''
import shutil
import json
def list_function():
print("Please choice the> print("【1】.Fetch haproxy.cfg backend infomation.")
print("【2】.Add haproxy.cfg backend infomation.")
print("【3】.Delete haproxy.cfg backend infomation.")
print("End".center(50, "#"))
def fetch(backend):
# 取出backend相关server信息
result = []# 定义结果列表
with open("haproxy.cfg", "r", encoding="utf-8") as f:# 循环读取文件
flag = False# 标记为假
for line in f:
# 以backend开头
line = line.strip()
if line.startswith("backend") and line == "backend " + backend:
flag = True# 读取到backend开头的信息，标记为真
continue
# 下一个backend开头
if flag and line.strip().startswith("backend"):
flag = False
break
# server信息添加到result列表
if flag and line.strip():
result.append(line.strip())
return result
def writer(backend, record_list):
with open("haproxy.cfg", "r") as old, open("new.cfg", "w") as new:
flag = False
for line in old:
if line.strip().startswith('backend') and line.strip() == "backend " + backend:
flag = True
new.write(line)
for new_line in record_list:
new.write(" " * 4 + new_line + "\n")
continue# 跳到下一次循环，避免下一个backend写二次
if flag and line.strip().startswith("backend"):# 下一个backend
flag = False
new.write(line)
continue# 退出此循环，避免server信息二次写入
if line.strip() and not flag:
new.write(line)
def add(backend, record):
global change_flag
record_list = fetch(backend)# 查找是否存在记录
if not record_list:# backend不存在, record不存在
with open('haproxy.cfg', 'r') as old, open('new.cfg', 'w') as new:
for line in old:
new.write(line)
# 添加新记录
new.write("\nbackend " + backend + "\n")
new.write(" " * 4 + record + "\n")
print("\033[32;1mAdd done\033[0m")
change_flag = True
else:# backend存在，record存在
if record in record_list:
print("\033[31;1mRecord already in it,Nothing to do!\033[0m")
change_flag = False
else:# backend存在，record不存在
record_list.append(record)
writer(backend, record_list)
print("\033[32;1mAdd done\033[0m")
change_flag = True
def delete(backend, record):
global change_flag
record_list = fetch(backend)# 查找是否存在记录
if not record_list:# backend不存在, record不存在
print("Not match the backend,no need delete!".center(50, "#"))
change_flag = False
else:# backend存在，record存在
if record in record_list:
record_list.remove(record)# 移除元素
writer(backend, record_list)# 写入
print("\033[31;1mDelete done\033[0m")
change_flag = True
else:# backend存在，record不存在
print("Only match backend,no need delete!".center(50, "#"))
change_flag = False
return change_flag
def output(servers):
# 输出指定backend的server信息
print("Match the server info:".center(50, "#"))
for server in servers:
print("\033[32;1m%s\033[0m" % server)
print("#".center(50, "#"))
def input_json():
# 判断输入，要求为json格式
continue_flag = False
while continue_flag is not True:
backend_record = input("Input backend info(json):").strip()
try:
backend_record_dict = json.loads(backend_record)
except Exception as e:
print("\033[31;1mInput not a json data type!\033[0m")
continue
continue_flag = True
return backend_record_dict
def operate(action):
global change_flag
if action == "fetch":
backend_info = input("Input backend info:").strip()
result = fetch(backend_info)# 取出backend信息
if result:
output(result)# 输出获取到的server信息
else:
print("\033[31;1mNot a match is found!\033[0m")
elif action is None:
print("\033[31;1mNothing to do!\033[0m")
else:
backend_record_dict = input_json()# 要求输入json格式
for key in backend_record_dict:
backend = key
record = backend_record_dict
if action == "add":
add(backend, record)
elif action == "delete":
delete(backend, record)
if change_flag is True:# 文件有修改，才进行文件更新
# 将操作结果生效
shutil.copy("haproxy.cfg", "old.cfg")
shutil.copy("new.cfg", "haproxy.cfg")
result = fetch(backend)
output(result)# 输出获取到的server信息
def judge_input():
# 判断输入功能编号，执行相应步骤
input_info = input("Your input number:").strip()
if input_info == "1":# 查询指定backend记录
action = "fetch"
elif input_info == "2":# 添加backend记录
action = "add"
elif input_info == "3":# 删除backend记录
action = "delete"
elif input_info == "q" or input_info == "quit":
exit("Bye,thanks!".center(50, "#"))
else:
print("\033[31;1mInput error!\033[0m")
action = None
return action
def main():
exit_flag = False
while exit_flag is not True:
global change_flag
change_flag = False
list_function()
action = judge_input()
operate(action)
if __name__ == '__main__':
main()
页:
[1]