上次已经进行了图解。本次将展示一些主要的函数：
##################################### 1 服务器组展示函数：sys_show() #################### def sys_show(file): sys_zone_name = 'backend' ### 搜索关键字, 系统组名称 ### srv_zone_name = '' ### 搜索关键字，服务器组的名称 ，例如：Oracle ### svr_name = 'server' ###搜索关键字，服务器关键字### srv_zone_names_list = [] ### 服务器组名称sys_zone_name组成的列表 ### srv_zone_names_dict = {} ### 服务器组名称sys_zone_name组成的字典，返回该值 ### svr_zones_dict = defaultdict(list) ### 各服务器组名称（srv_zone_name）和对应服务器详细参数（svr_record_dict{}）组成的字典 ，返回该值### svr_record_dict = OrderedDict() ### 包含服务器详细参数有序字典 ### svr_record_list = [] ### svr_record_dict{} 的列表形式### svr_name_flag = False ### 标记是否找到该服务组（srv_zone_name）的标志### with open(file, mode='r', encoding='UTF-8') as cfg_file: for line in cfg_file.readlines(): line = line.strip('\n') if re.match(sys_zone_name, line): ### 系统组名称 sys_zone_name 如果在line中 ### srv_zone_name = re.split('\s+', line)[1] ### 取得 srv_zone_name值 ### srv_zone_names_list.append(srv_zone_name) ### 将srv_zone_name值存入列表 srv_zone_names_list中 ### svr_name_flag = True ### 找到 对应的系统 sys_zone_name ### elif svr_name_flag and re.match('\s+server', line): ### 在该系统 sys_zone_name下，如果存在svr_name关键字 ### svr_record_list = re.split('\s+', line) ### 该行以 空格 为分隔符 将具体信息 存入 svr_record_list[]中 ### svr_record_dict = OrderedDict() ### 在定义新参数前，归位svr_record_dict，，否则会使用下面的新参数覆盖所有原先的赋值### svr_record_dict['server_name'] = svr_record_list[2] svr_record_dict['IP_address'] = svr_record_list[4] svr_record_dict['weight'] = svr_record_list[6] svr_record_dict['maxconn'] = svr_record_list[8] ###取出列表svr_record_list 的信息，存入svr_record_dict{}字典中### svr_zones_dict[srv_zone_name].append(svr_record_dict) ### 将srv_zone_name 与 svr_record_dict关联 ### else: svr_name_flag = False for k, v in enumerate(srv_zone_names_list, 1): srv_zone_names_dict[k] = v ###对srv_zone_names_list进行新字典赋值### print(' \033[34m %d . %-6s 服务器组 \033[0m' % (k, v)) ###输出 带编号的 服务器组 名称### return srv_zone_names_dict, svr_zones_dict ###返回srv_zone_names_dict字典 和svr_zones_dict 字典### #########################################################################################
效果图：
下面是服务器展示函数：
#################################### 4 服务器展示函数 srv_record_show () ################
def srv_record_show(srv_zone_name, svr_zones_dict):
############对sys名称进行遍历并加上数字编号###########################
print('\n================================================================================')
print('(%s)服务器信息如下 : ' % srv_zone_name)
print('%-5s %-18s %-18s %-18s %-18s' % ('id', 'name', 'address', 'weight', 'maxconn'))
svr_detail = svr_zones_dict[srv_zone_name]
for k, v in enumerate(svr_detail, 1): ###用enumerate进行unit_record展示###
print('%-5s ' % k, end='')
for kk, vv in v.items():
print('%-18s' % vv, end='|')
print()
print('\n=========================================================================')
#########################################################################################
效果图如下：
#################################### 8 配置文件回写函数 wrt2cfg() #######################
def wrt2cfg(file, svr_zones_all_dict):
newfile = '%s.new' % file ###定义回写的新文件###
svr_name_flag = False
###同时开启2个文件，分别进行读写###
with open(file, 'r', encoding='UTF-8') as read_file, open(newfile, 'w', encoding='UTF-8') as write_file:
for line in read_file:
if re.match('backend', line): ###读取read_file，匹配至backend行时###
write_file.write(line) ###将本行插入到write_file中。###
srv_zone_name = re.split('\s+', line)[1]
for srv_dict in svr_zones_all_dict[srv_zone_name]: ###对svr_zones_all_dict字典进行遍历###
server_line = '\t server {server_name} IP {IP_address} weight {weight} maxconn {maxconn}\n'
write_file.write(server_line.format(**srv_dict)) ###将srv_zone_name下的server行插入到write_file###
svr_name_flag = True ###读取到read_file中以server开头的行，将该行svr_name_flag标记为真###
elif svr_name_flag and re.match('\s+server', line):
###读取read_file，匹配至server开头的行时跳过，对write_file不做任何操作###
pass
else:
write_file.write(line) ###将read_file其他的行直接插入到write_file中###
svr_name_flag = False ###读取read_file，未发现以server开头的行时，将该行server_flag标记为FALSE###
print('\033[39m 配置文件 更新成功\033[0m')
os.system('move %s %s.bak ' % (file, file)) ###对源文件read_file进行备份###
os.system('move %s %s' % (newfile, file)) ##对新生成的write_file进行改名###
#########################################################################################
#################################### 9 更新服务器数据函数 update_svr_dict(）#############
def update_svr_dict (new_svr_dict,update_select_num) :
old_select_num = update_select_num
(new_svr_dict['server_name'], update_select_num) = name_filter(update_select_num) ###对name检查###
if old_select_num == update_select_num :
(new_svr_dict['IP_address'], update_select_num) = ipaddress_filter(update_select_num) ###对IP检查###
if old_select_num == update_select_num :
(new_svr_dict['weight'], update_select_num) = number_filter('权重值',update_select_num) ###对权重检查###
if old_select_num == update_select_num :
(new_svr_dict['maxconn'], update_select_num) = number_filter('最大连接数',update_select_num) ###对最大连接数检查###
if old_select_num == update_select_num :
print (new_svr_dict['server_name'], new_svr_dict['IP_address'], new_svr_dict['weight'], new_svr_dict['maxconn'])
return (new_svr_dict,update_select_num)
#########################################################################################
主程序如下：
##################################主程序开始#############################################
if __name__ == '__main__':
flag = True ### 脚本运行的标志 ###
sys_zone_name_ky = 'backend' ### 系统的 名称 ###
srv_zone_name_vl = '' ### 服务器组的名称 例如：Oracle
sys_cfg_file = 'configuretion.cfg' ###指定sys的配置文件###
svr_zones_all_dict = {} ### svr_zones_all_dict字典 记录各服务器的详细信息 ###
srv_zone_all_names_dict = {} ### svr_name_ky 字典（由svr_name_ky列表和 数字编号 组成的字典）###
global_opt2 = ''
srv_zone_name_vl_num = ''
while flag: ### 循环M ###
os.system('cls') ###清屏###
print('\n \033[38m ================================================================')
print(' \033[36m 欢迎使用 数据库服务器 配置文件管理平台：\033[0m')
print('\033[32m ================================================================')
print(''' \033[39m 1. 查询具体服务器\033[0m
\033[34m 2. 添加具体服务器\033[0m
\033[39m 3. 修改服务器列表\033[0m
\033[34m 4. 删除服务器列表\033[0m
\033[31m 0. 退出\033[0m ''')
print('================================================================')
select_num = input('\033[30m 请按 数字键（0~4） 选择：\033[0m')
##################################### 0 退出 exit ###############################
if select_num == '0':
flag = False
continue
while select_num in ['1','2','3','4'] : ### 循环 0 ###
if select_num == '1':temp_vl = '查看' #temp_vl 为下一步显示对应选定的操作而设置的临时参数
if select_num == '2':temp_vl = '添加'
if select_num == '3':temp_vl = '修改'
if select_num == '4':temp_vl = '删除'
old_select_num = select_num
print('\n %-8s ZONE 服务器组信息如下 : ' % sys_zone_name_ky)
(srv_zone_all_names_dict, svr_zones_all_dict) = sys_show(sys_cfg_file)
print('\033[32m ================================================================')
input_content = input('\033[39;1m请输入进行\033[33;1m%s\033[0m操作的服务器组的编号或名称' \
'(返回主菜单,请输入B or b)：\033[0m' % temp_vl )
############################## 对输入值进行过滤 #############################
(srv_zone_name_vl, tmp_flag) = content_filter(input_content, srv_zone_all_names_dict)
if srv_zone_name_vl: ###显示对应 srv_zone 下的服务器列表###
srv_record_show(srv_zone_name_vl, svr_zones_all_dict)
if tmp_flag == False : break
##################################### 1 查询服务器信息 ######################
if select_num == '1': continue
##################################### 2. 更新服务器信息 #####################
while select_num == old_select_num and srv_zone_name_vl: ### 循环1 ###
if select_num in [ '3', '4' ]: ###判断修改、删除选项中输入的服务器编号准确性###
srv_zone_name_vl_num = input('\033[32;1m请输入\033[33;1m%s\033[0m的服务器的编号,((N|n)返回上级菜单)：\033[0m'% temp_vl)
if srv_zone_name_vl_num in ['N', 'n']: break
if len(srv_zone_name_vl_num) != 0:
srv_zone_name_vl_num = input_str2int(srv_zone_name_vl_num)
#####################指定具体的server编号进行判断#####
elif srv_zone_name_vl_num not in range(1, len(svr_zones_all_dict[srv_zone_name_vl]) + 1):
print('超出范围，请重新输入！！！')
continue ###srv_zone_name_vl_num不在在服务器组个数范围之内，提示并重新选择###
if select_num in [ '2','3' ]: ###添加和修改服务器时进行的操作步骤：更新服务器列表###
update_srv_dict = OrderedDict()
(update_srv_dict, select_num) = update_svr_dict(update_srv_dict,select_num)
if select_num in [ '4' ]:
print(svr_zones_all_dict[srv_zone_name_vl][srv_zone_name_vl_num - 1]) #显示被选中删除的服务器列表#
if select_num == old_select_num : ###如果在以上操作中没有进行取消操作，则选择参数不会变化，维持原来的用户操作。###
update_opt = input('(Y|y)\033[33;1m%s\033[0m此服务器,任意键取消：' % temp_vl)
if update_opt in ['y','Y']:
if select_num == '2':
svr_zones_all_dict[srv_zone_name_vl].append(update_srv_dict)
if select_num == '3':
svr_zones_all_dict[srv_zone_name_vl][srv_zone_name_vl_num - 1] =update_srv_dict
if select_num == '4':
del svr_zones_all_dict[srv_zone_name_vl][srv_zone_name_vl_num - 1]
wrt2cfg(sys_cfg_file, svr_zones_all_dict)
srv_record_show(srv_zone_name_vl, svr_zones_all_dict)
print('\033[33;1m%s\033[0m成功！' % temp_vl)
else : ###如果在添加、修改、删除过程中，选择了取消，则select_num=5 ，对应操作皆被取消###
print("\033[33;1m%s\033[0m被取消" % temp_vl)
global_opt2 = input('(Y|y)继续\033[33;1m%s\033[0m本组服务器,（N|n）返回重新选择服务器组，（B|b）返回主菜单：' % temp_vl)
###以上所有操作完成后，根据用户需求，修正select_num 数值，提示用户下一步操作。###
if global_opt2 in ['B', 'b']: ### 遇到select_num = '5'时，返回主菜单。###
select_num = '5'
print("返回主菜单。")
elif global_opt2 in ['N', 'n']: ###用户继续选择服务器组，继续进行对原来的操作select_num = old_select_num,。###
select_num = old_select_num
break
else: select_num = old_select_num ###在原来选择的服务器组中，继续进行操作。###
else:
sys.exit('\033[36m 程序退出，欢迎使用，谢谢! \033[0m')
#########################################################################################
部分截图如下：
