python案例1---修改haproxy的配置文件

运用的知识点:

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

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

      1.os.rename('文件1','文件2')

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

  3.python函数的定义和调用

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

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

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

 

  

代码:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
#!/usr/bin/env python
# -*- coding:utf-8 -*-
#导入2个模块json和os
import   json,os
 
#定义fetch函数,并同时传入我们指定backend参数,此参数代表backend名
def  fetch(backend):
     flag  =  False  #定义flag,目的是为了后面是否取可用的backend
     fetch_list = []   #定义空列表,目的是为了后面将取出的backend信息存储在此列表里面
 
     #以读的形式打开'ha'文件,一行一行读,然后将我们指定backend读取到列表里面
     #最后返回装有我们指定backend信息的列表
     with  open ( 'ha' ) as obj:
         for  line  in  obj:
             if  line.strip()  = =  "backend %s"  % (backend):
                 flag  =  True
                 continue
             if  line.strip().startswith( 'backend' ):
                 flag = False
             if  flag  and  line.strip():
                 fetch_list.append(line.strip())
 
     return  fetch_list
 
#定义add1函数,并同时传入参数dict_info,此字典参数里面包含我们要传入的server相关信息
def  add1(dict_info):
 
     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" ])
 
     #将backend获取,并存储在fetch_list列表里面
     fetch_list = fetch(backend_title)
 
     #如果要插入的backend存在
         #1.首先设定2个标志:flag和has_write
             #flag:用于找到要添加的backend下的内容
             #has_write:用于判断fetch_list中的内容是否重新写入到了文件
         #2.遍历源文件ha:
             #1首先逐行读取,并同时进行判断,
                 # 1.判断backend是否已经找到,先把backend写入文件,然后找到后将flag设置为True,最后跳出本次循环
                 # 2.判断找到的backend里面的record是否为空,若为空将flag设置为False
                 # 3.对flag进行判断
                     #1.将flag为True的record从fetch_list写入到文件中,在这里面同时要进行是否已经写入的判断,即定义has_write的作用
                     #2.若flag为False,就直接写入新文件,即将非列表里的内容直接写入
 
     if  fetch_list:
         flag = False
         has_write  =  False
         with  open ( 'ha' ) as read_obj, open ( 'ha.new' , '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  =  False
                 if  flag:
                     for  new_line  in  fetch_list:
                         if  not  has_write:
                             temp  =  "%s%s"  % ( " " * 8 ,new_line)
                             write_obj.write(temp)
                             has_write = True
                 else :
                     write_obj.write(line)
 
 
     else :
     #如果要插入的backend不存在
         #分2部分写入:
             # 1部分.从ha文件里面直接读取,并同时将读取的内容写入新的文件ha.new
             # 2部分.将新的backend和context信息,直接写到新的文件ha.new的文件尾部
         with  open ( 'ha' ) as read_obj, open ( 'ha.new' , '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)
 
     #将原文件ha改名备用文件为ha.bak,即将ha下线
     #将新文件ha.new改名为线上文件ha,即将ha.new上线
     os.rename( 'ha' , 'ha.bak' )
     os.rename( 'ha.new' , 'ha' )
 
 
#删除函数
def  delete(dict_info):
     #1.首先获取用户的记录并以列表的形式存储,以便于我们调用
     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()函数,得到包含backend内容的列表
     fetch_list  =  fetch(del_backend)
     #判断1:若列表为空:直接返回
     if  not  fetch_list:
         return
     #判断2:否则不为空
     else :
         #判断1:若用户输入的server信息不在fetch_list里面,直接返回
         if  context_record  not  in  fetch_list:
             print  "your server message is not exists"
             return
         #判断2:若输入的server在fetch_list里面,就将此server信息从列表中移除
         else :
             fetch_list.remove(context_record)
 
         #逐行从ha文件中读取,并同时进行判断
         #核心思想:
             #首先边读取、边判断、边写入,
             # 判断其实就是我们自己输入的backend,找到后将列表fetch_list信息写入到新的文件里面。
             # 其余的非列表的内容直接从源文件ha写到新文件ha.new即可
         with  open ( 'ha' , 'r' ) as read_obj, open ( 'ha.new' , '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)
 
     #将原文件ha改名备用文件为ha.bak,即将ha下线
     #将新文件ha.new改名为线上文件ha,即将ha.new上线
     os.rename( 'ha' , 'ha.bak' )
     os.rename( 'ha.new' , 'ha' )
 
 
"""
s='{"backend":"ttt.oldboy.org","record":{"server":"100.1.7.9","weight":"20","maxconn":"3000"}}'
data_info=json.loads(s)
add1(data_info)
#delete(data_info)
"""
 
#主函数
if  __name__  = =  "__main__" :
     #获取用户的操作选择
     print  u "1.获取记录\n2.添加记录\n3.删除记录\n"
     select_num = raw_input ( "plz input number:" )
 
     #根据用户的选择,进行调用函数
     if  select_num  = =  "1" :
         backend  =  raw_input ( "plz input backend:" )
         fetch_list  =  fetch(backend)
         for  in  fetch_list:
             print  i
     else :
         print  "plz input like this example:"
         print  '{"backend":"ttt.oldboy.org","record":{"server":"100.1.7.9","weight":"20","maxconn":"3000"}}'
         data_str  =  raw_input ( "plz input record>>>" )
         data_dict  =  json.loads(data_str)
         if  select_num  = =  "2" :
             add1(data_dict)
         elif  select_num  = =  "3" :
             delete(data_dict)
         else :
             print  "input error"

转载自:http://www.cnblogs.com/jachy/p/4983246.html

你可能感兴趣的:(Python)