zoukankan      html  css  js  c++  java
  • python作业day3修改配置文件

    思维还有点乱,撸代码到深夜,先上代码吧.(我是跟着武sir的思路的)

    流程图:

    配置文件操作


    代码(有注释):

      1 #!/usr/bin/env python
      2 # -*- coding:utf-8 -*-
      3 import json,os
      4 def login():
      5     flag = False
      6     while True:
      7         username = input("Please enter a user name:")
      8         l = open('name_lock.txt','r')
      9         for lline in l.readlines():
     10             lline = lline.strip()
     11             if username == lline:
     12                 print("账号被锁")
     13                 flag = True
     14                 break
     15         if  flag == True:
     16             break
     17         #while True:
     18         u = open('name.txt','r')
     19         for uline in u.readlines():
     20             user,password,mony = uline.strip().split(':')
     21             #print(user)
     22 
     23             if username == user:
     24 
     25                 i = 0
     26                 while  i < 3:
     27                     passwd = input('Please enter a password:')
     28                     i +=1
     29 
     30                     if passwd == password:
     31                         print('欢迎%s登陆管理平台' % username)
     32                         flag = True
     33                         break
     34 
     35                     else:
     36                         if i >= 3:
     37                             with open('name_lock.txt','a') as l_2:
     38                                 l_2.write(username + '
    ')
     39                             exit("试了太多次,将被锁定,请联系管理员")
     40                         print('密码输入错误,还有%d次机会' % (3 - i))
     41                 break
     42         else:
     43             print("用户输入错误,请重新输入")
     44         if flag == True:
     45             break
     46 
     47 def chazhao(backend):
     48     cz_list = []
     49     flag = False
     50     with open('ha.txt','r') as cz_1:                                        #打开旧配置文件
     51         for line in cz_1:
     52             cz_line = line.strip()
     53             if cz_line == "backend %s" % backend:                           #如果读取出的行等于输入的backend
     54                 flag = True
     55                 continue
     56             if flag == True and cz_line.startswith('backend'):              #如果flag=True并且行开头为backend
     57                 break
     58             if flag == True and cz_line.startswith('server'):               #如果flag= True并且行开头为server
     59                 cz_list.append(cz_line)
     60     return cz_list
     61 
     62 def add_1(dict_info):
     63     backend_title = dict_info.get('backend')
     64     crrent_title = "backend %s" %backend_title
     65     crrent_record = "server %s %s weight %s maxconn %s" %(dict_info['record']['server'], dict_info['record']['server'], dict_info['record']['weight'], dict_info['record']['maxconn'])
     66     cz_list = chazhao(backend_title)
     67     if cz_list:
     68         #backend存在
     69         if crrent_record in cz_list:    #如果新的数据在旧的配置文件中
     70             pass        #不操作
     71         else:           #否则,进行以下操作
     72             cz_list.append(crrent_record)    #插入数据
     73         #处理完成之后的中间部分
     74         #读取旧配置文件,写入新的配置文件中
     75         #边读边写
     76         #新处理后的配置文件写入新的配置文件中
     77         flag = False
     78         flag_1 = False
     79         with open('ha.txt','r') as ha_1,open('ha_new.txt','w') as ha_2:     #打开旧配置文件,打开新配置文件
     80             for line in ha_1:
     81                 if line.strip() == crrent_title:                                  #如果读取的行等于输入的backend
     82                     flag = True
     83                     ha_2.write(line)                                              #将读取的行写入新文件
     84                     continue                                                    #重新进行循环
     85                 if flag == True and line.startswith('backend'):             #如果标志位等于True并且此行开头为‘backend’
     86                     flag = False                                                 #设置标志位为False(因为循环到了下一个‘backend’)
     87                 if flag == True and line.startswith('server'):                #把已经处理完的数据,写入新配置文件中
     88                      if not flag_1:
     89                         for nwe_line in cz_list:
     90                             temp = ("	"+nwe_line + "
    ")
     91                             ha_2.write(temp.expandtabs())                         #将新的数据按行写入新配置文件中
     92                         flag_1 = True
     93                 else:
     94                     ha_2.write(line)
     95     else:
     96         #backend不存在,添加记录和backend
     97         with open('ha.txt','r') as ha_1,open('ha_new.txt','w') as ha_2:
     98             for line in ha_1:
     99                 ha_2.write(line)
    100             ha_2.write("
    " + crrent_title + "
    ")
    101             temp = ("	"+crrent_record + "
    ")
    102             ha_2.write(temp.expandtabs())
    103     os.rename("ha.txt","ha.txt.back")                                   #将旧配置文件备份
    104     os.rename("ha_new.txt","ha.txt")                                    #将新配置文件命名为线上使用的文件名
    105     os.rename("ha.txt.back","ha_new.txt")
    106 
    107 def delt(dict_info):
    108     backend_title = dict_info.get('backend')
    109     crrent_title = "backend %s" %backend_title
    110     crrent_record = "server %s %s weight %s maxconn %s" %(dict_info['record']['server'], dict_info['record']['server'], dict_info['record']['weight'], dict_info['record']['maxconn'])
    111     cz_list = chazhao(backend_title)
    112     if cz_list:                                         #如果列表有数据
    113         if crrent_record in cz_list:       #如果数据在列表中
    114             print(cz_list)
    115             del cz_list[cz_list.index(crrent_record)]   #删除这个列表中找到的对应数据
    116         print(cz_list)
    117         with open('ha.txt','r') as ha_1,open('ha_new.txt','w') as ha_2:#打开新旧配置文件
    118             flag = False
    119             flag_1 = False
    120             for line in ha_1:                                       #读取行在旧配置文件中
    121                 line_strip = line.strip()                           #将读取的行前后空格删除
    122                 if line_strip == crrent_title:                      #如果删除空格后的行数据等于输入的backend
    123                     if len(cz_list)>0:                              #如果列表中还有数据
    124                         ha_2.write(crrent_title + '
    ')             #将backend写回去(前面删除的时候会把当前backend删除掉)
    125                     flag = True
    126                     continue                                        #跳出循环
    127                 if flag == True and line_strip.startswith('backend'):  #如果flag等于True并且行开头为backend
    128                     flag = False
    129                 if flag == True and line_strip.startswith('server'):    #如果flag= True并且行开头为server执行下面步骤
    130                     if not flag_1:
    131                         for i in cz_list:
    132                             temp = ("	"+ i + "
    ")
    133                             ha_2.write(temp.expandtabs())
    134                         flag_1 = True
    135 
    136                 else:
    137                     ha_2.write(line)
    138     else:
    139         return
    140     os.rename("ha.txt","ha.txt.back")
    141     os.rename("ha_new.txt","ha.txt")
    142     os.rename("ha.txt.back","ha_new.txt")
    143 def main():
    144     login()
    145     while True:
    146         print("1.查看 2.添加 3.删除 4.退出")
    147         number = input("请输入序号:")
    148 
    149         if number == "1":
    150             backend_1=input("请输入backend:")
    151             cz = chazhao(backend_1)
    152             print(cz)
    153             continue
    154         elif number == "2":
    155             backend_1=input("请输入backend:")
    156             server_1 = input("请输入server:")
    157             weight_1 = input("请输入weight:")
    158             maxconn_1 = input("请输入maxconn:")
    159             s = '{"backend":"%s","record":{"server":"%s","weight":"%s","maxconn":"%s"}}'%(backend_1,server_1,weight_1,maxconn_1)
    160             data_dict = json.loads(s)
    161             add_1(data_dict)
    162             cz_1 = chazhao(backend_1)
    163             print(cz_1)
    164             continue
    165         elif number == "3":
    166             backend_1=input("请输入backend:")
    167             cz = chazhao(backend_1)
    168             print(cz)
    169             server_1 = input("请输入server:")
    170             weight_1 = input("请输入weight:")
    171             maxconn_1 = input("请输入maxconn:")
    172             s = '{"backend":"%s","record":{"server":"%s","weight":"%s","maxconn":"%s"}}'%(backend_1,server_1,weight_1,maxconn_1)
    173             data_dict = json.loads(s)
    174             delt(data_dict)
    175             cz_1 = chazhao(backend_1)
    176             print(cz_1)
    177             continue
    178         elif number == "4":
    179             exit()
    180         else:
    181             print("输入有误")
    182             continue
    183 if __name__ =='__main__':
    184     main()
    配置文件操作
  • 相关阅读:
    CentOS6.7 mysql5.6.33修改数据文件位置
    win8 win10 安装msi 提示2502、2503的错误代码
    2016年国内开源镜像站点汇总
    eclipse创建本地maven
    maven添加sqlserver的jdbc驱动包
    CentOS6.5下RPM方式安装mysql5.6.33
    linux中~和/的区别
    Linux命令学习(22):ss命令
    Linux命令学习(21):netstat命令
    Linux命令学习(20):traceroute命令
  • 原文地址:https://www.cnblogs.com/spykids/p/5180838.html
Copyright © 2011-2022 走看看