Esempio n. 1
0
    def add(self,transfer_cmd):
        add_flag = True
        print('This is add,example add -g xxx -h x.x.x.x')
        #print(transfer_cmd.index["-g"])

        if len(transfer_cmd) < 5:
            #如果输入不小于5,说明格式不对
            print('格式错误请重试')
        else:
            if add_flag:
                index_g = transfer_cmd.index('-g')
                #判断出 '-g'的下标,以便取出组名
                #print(index_g)
                index_h = transfer_cmd.index('-h')
                #判断出'-h'的下标,以便取出后面的主机地址
                #print(index_h)
                group = transfer_cmd[index_g+1:index_h]
                #将组名加入group_list,组名就等于 index_g+1:index_h 之间的值
                print(group)
                host_list = transfer_cmd[index_h+1:]
                #将下标为index_h+1后面的元素放入host_list,为主机地址列表
                print(host_list)
                host = config_handle.read()
                for i in host:
                    if i in host.keys():
                        #host.keys().append(host_list)
                        pass
                    else:
                        raw = input('你想新建一个分组吗? [y/n]').strip()
                        if raw == 'y':
                            host[i] = host_list
                            print(host)
                            config_handle.write(host)
                        else:
                            print('那就返回吧 !')
Esempio n. 2
0
 def show(self,transfer_cmd):
     '''
     输出分组信息等等
     :param transfer_cmd:
     :return:
     '''
     print('This is all of hosts: ')
     configure = config_handle.read()
     #从config_handle读取分组信息
     if len(transfer_cmd) == 1:
         #如果只输入了'show'打印所有分组信息
         print("{:*^50}".format("groups"))
         #print(configure)
         for k in configure:
             #循环打印读取字典的key,也就是分组组名
             print(k)
         print("*" * 50)
     elif '-g' in transfer_cmd:
         #如果用户输入带有'-g',说明指定了分组
         group_list = transfer_cmd[transfer_cmd.index("-g")+1:]
         #获取组名,用户输入信息,'-g' 下标后面的字符 就是组名,赋给 group_list这个列表
         print(group_list)
         try:
             for i in group_list:
                 print("{:*^50}".format(i))
                 for j in configure[i]:
                     print(j)
                 print("*" * 50)
         except (ValueError, KeyError):
             print("invalid group name!")
     else:
         print("invalid instructions!")