import os #導入os模塊
def search(): #定義查找函數
with open('haproxy.txt','r') as f: #只讀方式打開文件
value = input('請輸入您要查詢的信息:')
backend = 'backend %s' % value #字符串拼接
records = [] #定義空列表
tag = False #將標簽設置為false
for line in f: #循環打開文件
if line.strip() == backend: #當line等于backend，將標簽改為true
tag = True
continue #跳出當前循環
if line.startswith('backend'): #當line以backend開頭，跳出整個循環
break
if tag == True: #標簽為true，將line去除首尾空值，加入到records列表中
records.append(line.strip())
print(records) #打印列表
def add(): #定義add函數
adds = input('請輸入你要添加的內容：')
adds = eval(adds) #將adds轉換為集合
backend = '\nbackend %s \n' % adds['backend'] #拼接字符串
record = ' server {server} {server} weight {weight} maxconn {maxconn}\n'.format(
server = adds['record']['server'],
weight = adds['record']['weight'],
maxconn = adds['record']['maxconn']
)
records =[] #定義records空列表
with open('haproxy.txt','r') as r_file,\
open('haproxy_add.txt','w+') as w_file: #只讀方式打開haproxy，生成新文件haproxy_add
tag = True #標簽設置為true
for r_line in r_file: #循環文件
if r_line.strip() == backend.strip(): #當兩個值相等，
tag = False #將標簽設為false
records = r_file.readlines() #并將這一行的后面所有值加入到records列表中
w_file.write(backend) #將backend寫入新文件
break #跳出for循環
w_file.write(r_line) #將r_line寫入新文件
if tag == True: #當tag為true時
w_file.write(backend) #寫入backend和record
w_file.write(record)
if tag == False: #當tag為false時
if record not in records: #如果record不在records列表內
records.insert(0,record) #把record的值加入到列表的第一個位置
for line in records: #遍歷records列表
w_file.write(line) #將列表的值寫入到新文件
os.rename('haproxy.txt','haproxy.addbak') #將文件備份
os.rename('haproxy_add.txt','haproxy.txt')
def remove(): #定義remove函數
removes = input('請輸入你要刪除的內容：')
removes = eval(removes) #將removes轉換為集合
backend = '\nbackend %s\n' % removes['backend'] #拼接字符串
record = ' server {server} {server} weight {weight} maxconn {maxconn}\n'.format(
server = removes['record']['server'],
weight = removes['record']['weight'],
maxconn = removes['record']['maxconn']
)
with open('haproxy.txt','r') as r_file,\
open('haproxy_remove.txt','w+') as re_file: #只讀方式打開haproxy，生成新文件haproxy_remove
tag = True #定義標簽為True
for r_line in r_file: #遍歷文件
if backend.strip() == r_line.strip() or r_line.strip() == record.strip():
tag = False #當兩個字符串有一個等于r_line時，將標簽改為false
continue #跳出當前循環
if r_line.startswith('backend') or r_line.strip() != record.strip():
tag = True #當r_line開頭為backend或者record不等于r_line時，將標簽改為true
if tag == True: #當標簽為true，寫入新文件
re_file.write(r_line)
os.rename('haproxy.txt','haproxy.removbak') #備份文件
os.rename('haproxy_remove.txt','haproxy.txt')
def alter(): #定義alter函數
alters = input('請輸入你要修改的原的內容：')
alters_new = input('請輸入你要修改的新內容：')
alters = eval(alters) #將兩個值轉為字典
alters_new = eval(alters_new)
backend = '\nbackend %s\n' % alters['backend'] #拼接字符串
record = ' server {server} {server} weight {weight} maxconn {maxconn}\n'.format(
server=alters['record']['server'],
weight=alters['record']['weight'],
maxconn=alters['record']['maxconn']
)
backends = '\nbackend %s\n' % alters_new['backend']
records = ' server {server} {server} weight {weight} maxconn {maxconn}\n'.format(
server=alters_new['record']['server'],
weight=alters_new['record']['weight'],
maxconn=alters_new['record']['maxconn']
)
with open('haproxy.txt', 'r') as r_file, \
open('haproxy_alter.txt', 'w+') as al_file: #只讀方式打開haproxy，生成新文件haproxy_alter
for r_line in r_file: #遍歷文件
if backend.strip() == r_line.strip(): #當原值為輸入的原值，修改為新值
r_line = r_line.replace(backend,backends)
if r_line.strip() == record.strip():
r_line = r_line.replace(record,records)
al_file.write(r_line) #寫入新文件
os.rename('haproxy.txt','haproxy.alterbak') #備份文件
os.rename('haproxy_alter.txt','haproxy.txt')
menu = ['1.查找','2.新增','3.刪除','4.修改','q.退出'] #定義菜單
menu_func = {'1':search, #定義功能字典
'2':add,
'3':remove,
'4':alter,
}
while True: #開始循環
print('-----------------------------------')
for i in menu: #打印菜單
print(i)
choise = input('請選擇您的操作方式(輸入對應編號)：')
if choise.isdigit(): #判斷輸入是否為數字
if int(choise) <= len(menu_func) and int(choise) >0: #判斷輸入數字是否在菜單內
menu_func[choise]() #調用功能字典內對應數字的函數
else: #為數字但不再菜單內，提示
print('請輸入對應的編號!\n')
elif choise == 'q': #輸入q退出
print('退出成功!')
break
else: #不為數字提示
print('請輸入對應的編號!\n')
作業有點亂，別噴我！！！！！我盡力了- -
轉載于:https://www.cnblogs.com/judge9/p/6903392.html
)
