python小工具_自動生成nodelist

工具開發背景：
將一個含有很多基站ip的文件拆分成多個batch文件。

$ cat Node_list | wc -l
3230

$python Generate_node_list.py -N 50 -F node_list
可將含有多個node Ip 的文件拆分成50個一組的新文件。

代碼如下，文件爲 Generate_node_list.py

#!/usr/bin/python
import sys, getopt

def GetOption(argv):
    nodes_count = ''
    nodes_file = ''
    try:
        opts, args = getopt.getopt(argv, "hN:F:",["Num=","File="])
    except getopt.GetoptError:
        print ('Error arg input -N <number of nodes>')
        sys.exit(2)

    for opt, arg in opts:
        if opt == "-h":
            print ('input python Generate_node_list.py -N <number of nodes>')
            sys.exit()
        elif opt in ("-N", "--Num"):
            nodes_count = arg
        elif opt in ("-F", "--File"):
            nodes_file = arg
    return ( nodes_file, nodes_count,)

if __name__ == '__main__':
    option = GetOption(sys.argv[1:])
    node_list_file, nodes_per_batch = option
    nodes = open(node_list_file,'r')
    nodelist = nodes.readlines()
    nodelist_cp = nodelist[:]
    num = 0
    while num <= int(round(len(nodelist)))/int(nodes_per_batch):
        line = 0
        for i in nodelist_cp:
            batchfile = open('batch_' + str(num+1),'a+')
            batchfile.write(i)
            line += 1
            if line == int(nodes_per_batch):
                for j in range(int(nodes_per_batch)):
                    nodelist_cp.pop(0)
                break
        num += 1
        print('The total number of nodes in this file is ' + str(line))