python作业编程_python基础:day3作业

修改haproxy配置文件

基本功能:

1.获取记录

2.添加记录

3.删除记录

代码结构:

三个函数一个主函数

知识点:

1.python简单数据结构的使用：列表、字典等

2.python两个模块的使用：os和json

a.os.rename('文件1'，'文件2')

b.json完成自动识别字典、列表，并识别后进行自动转换

3.python函数的定义和调用

4.标志位的灵活运用：flag和haswrite分别用于找backend的record和判断是否已经将记录写入到文件

5.python基本语法的使用:for循环、if...else....判断

6.python文件的操作的基本使用：with open('文件1','模式') as obj1,open('文件2','模式') as obj2

流程图:

代码:

#!/usr/bin/env python3

#antuor:Alan

import os

import json

"""

把文件按从上到下的顺序分为上部分,要处理部分,下部分;

step1: 上部分从原配置文件写入-->新配置文件

step2: 遇到要处理部分则写入-->fetch_list

step3: 下部分从原配置文件写入-->新配置文件

step4: fetch_list入-->新配置文件

"""

def fetch(backend): ###定义fetch函数，并同时传入我们指定backend参数，此参数代表backend名

flag = False ###定义flag为False，目的是为了后面是否取可用的backend

fetch_list=[] ###定义空列表，目的是为了后面将取出的backend信息存储在此列表里面

with open('配置文件.txt') as obj: ###打开配置文件

for line in obj: ###一行行读取

if line.strip() == "backend %s" %(backend): ###line.stri()去掉空格和换行符

flag = True ###读到要取的记录flage改为True

continue ###结束本次循环,下面代码不执行,开始新的循环

if flag and line.strip().startswith('backend'): ###如果已经取到记录,则又读到backend开头的数据则不取

flag=False

if flag and line.strip(): ###判断flag为True,且不是空行(布尔值非空是True)

fetch_list.append(line.strip()) ###把数据加入列表

return fetch_list ###返回值

def add1(dict_info): ###定义add1函数，同时传入参数dict_info,字典参数里面包含要传入的server信息

backend_title = dict_info.get("backend") ###插入的backend的名称(变量初始化)

context_title = "backend %s" %(backend_title) ###插入backend整个字段(变量初始化)

record_title = dict_info["record"] ###要插入的记录(变量初始化)

context_record = "server %s %s weight %s maxconn %s" %(record_title["server"],record_title["server"],record_title["weight"],record_title["maxconn"])

fetch_list=fetch(backend_title) ###把读取的记录放入列表(变量初始化)

if fetch_list: ###判断列表是否为空

flag=False ###标志位

has_write = False ###标志位

with open('配置文件.txt','r') as read_obj,open('新配置文件.txt','w') as write_obj:

for line in read_obj:

if line.strip() == context_title:

write_obj.write("\n"+line)

flag=True

continue

if flag and line.startswith('backend'): ###flag为True,如果在读到backend信息,则不处理

flag = False

if flag:

for new_line in fetch_list:

if not has_write:

temp = "%s%s" %(" "*8,new_line) ###把列表的记录赋值给temp

write_obj.write(temp)

has_write=True

else:

write_obj.write(line)

else: ###如果fetch_list为空

with open('配置文件.txt') as read_obj,open('新配置文件.txt','w') as write_obj:

for line in read_obj:

write_obj.write(line)

write_obj.write("\n"+context_title+"\n")

temp=" "*8+context_record+"\n"

write_obj.write(temp)

os.rename('配置文件.txt','配置文件.txt') ###将原文件配置文件改名备用文件为配置文件b

os.rename('新配置文件.txt','配置文件.txt') ###将新配置文件改名为配置文件

def delete(dict_info): ###删除函数

del_backend = dict_info["backend"]

del_record = dict_info["record"]

context_title = "backend %s" %(del_backend)

context_record = "server %s %s weight %s maxconn %s" %(del_record["server"],del_record["server"],del_record["weight"],del_record["maxconn"])

fetch_list = fetch(del_backend)

if not fetch_list:

return

else:

if context_record not in fetch_list:

print ("your server message is not exists")

return

else:

fetch_list.remove(context_record)

with open('配置文件.txt','r') as read_obj,open('新配置文件.txt','w') as write_obj:

flag = False

has_write = False

for line in read_obj:

if line.strip() == context_title:

write_obj.write(line)

flag = True

continue

if flag and line.startswith('backend'):

flag = False

if flag:

if not has_write:

print (fetch_list)

for new_line in fetch_list:

temp = "%s%s\n" %(" "*8,new_line)

write_obj.write(temp)

has_write = True

else:

write_obj.write(line)

os.rename('配置文件.txt','配置文件.txt')

os.rename('新配置文件.txt','配置文件.txt')

s='{"backend":"ttt.oldboy.org","record":{"server":"100.1.7.9","weight":"20","maxconn":"3000"}}' ###要添加的记录信息

if __name__ == "__main__": ###主函数,入口

print (u"1.获取记录\n2.添加记录\n3.删除记录\n") ###获取用户的操作选择

select_num=input("请输入需要进行的操作编号:")

if select_num == "1": ###根据用户的选择，进行调用函数

backend = input("请输入backend信息:")

fetch_list = fetch(backend)

for i in fetch_list:

print ("配置信息:%s" % i)

else:

print ("请按以下格式输入:")

print ('{"backend":"ttt.oldboy.org","record":{"server":"100.1.7.9","weight":"20","maxconn":"3000"}}')

data_str = input("请输入信息>>>")

data_dict = json.loads(data_str)

if select_num == "2":

add1(data_dict)

elif select_num == "3":

delete(data_dict)

else:

print ("输入有误!!!!!")