码迷,mamicode.com
首页 > 编程语言 > 详细

python学习之路之案例1(修改haproxy的配置文件)

时间:2015-11-23 10:10:23      阅读:297      评论:0      收藏:0      [点我收藏+]

标签:源文件   配置文件   知识点   python   record   

整个案例运用的知识点:

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

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

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

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

  3.python函数的定义和调用

  4.标志位的灵活运用:

            1.flag用于找backend

            2.haswrite用于找record和判断是否已经将记录写入到文件

  5.python基本语法的使用:

            1.for循环

            2.if...else....判断

  6.python文件的操作的基本使用:

            1.with open(‘文件1‘,‘模式‘) as obj1,open(‘文件2‘,‘模式‘) as obj2


整个代码的中心思想:

    1.设置2个文件:源文件ha和终文件ha.new(整个过程是:边读边写)

            1.源文件ha用于提供读

            2.终文件ha.new用于提供写

    2.上线、下线的巧妙应用:

            1.下线:将源文件ha  rename 成备用文件ha.bak

            2.上线:将终文件ha.new rename 成新的配置文件ha

 

配置文件(即ha):

global       
        log 127.0.0.1 local2
        daemon
        maxconn 256
        log 127.0.0.1 local2 info
defaults
        log global
        mode http
        timeout connect 5000ms
        timeout client 50000ms
        timeout server 50000ms
        option  dontlognull

listen stats :8888
        stats enable
        stats uri       /admin
        stats auth      admin:1234

frontend jachy.org
        bind 0.0.0.0:80
        option httplog
        option httpclose
        option  forwardfor
        log global
        acl www hdr_reg(host) -i www.jachy.org
        use_backend www.jachy.org if www

backend www.jachy.org
        server 100.1.7.9 100.1.7.9 weight 20 maxconn 3000



代码实现的功能:

    1.查找ha文件中指定backend下的后端主机列表信息

    2.增加ha文件中指定backend以及backend后端主机列表信息

    3.删除ha文件中指定backend以及backend后端主机列表信息

代码:

#!/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.jachy.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 i in fetch_list:
            print i
    else:
        print "plz input like this example:"
        print ‘{"backend":"ttt.jachy.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"

  


python学习之路之案例1(修改haproxy的配置文件)

标签:源文件   配置文件   知识点   python   record   

原文地址:http://jachy.blog.51cto.com/10625380/1715829

踩
(0)
赞
(0)
   
举报
评论 一句话评论(0)
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!