威步加密狗共享配置脚本

很久没写博客了，今天随便写个linux下加密狗共享配置的脚本。

# -*- coding:utf-8 -*-
import os
import re
import ConfigParser
cp = ConfigParser.ConfigParser()

def server():
    cp.set('General','IsNetworkServer','1')
    cp.set('General','NetworkAccessFsb','1')
    cp.set('General','Client1','localhost')
    ips = raw_input("请输入需要共享的客户端的IP个数： ")
    for i in range(0,int(ips)):
        client = "Client"+str(i+2)
        ip = raw_input("请输入客户端ip地址")
        cp.set('General',client,ip)
    cp.write(open("/etc/wibu/CodeMeter/Server.ini","w"))
    print u"配置服务端完成"
    
def client():
    ips = raw_input("请输入服务端的个数： ")
    for i in range(0,int(ips)):
        ServerSearchList = "ServerSearchList\Server"+str(i+1)
        ip = raw_input("请输入服务端ip地址")
        cp.add_section(ServerSearchList)
        cp.set(ServerSearchList,'Address',ip)
    cp.write(open("/etc/wibu/CodeMeter/Server.ini","w"))
    print u"配置客户端完成"

def add_server():
    conf = open('/etc/wibu/CodeMeter/Server.ini','r')
    lines =conf.readlines()
    count = 0
    for line in lines:
        if re.match('Client',line,flags=0):
            count += 1
        elif re.match('client',line,flags=0):
            count += 1
        else:
            pass
    print count
    cp.set('General','IsNetworkServer','1')
    cp.set('General','NetworkAccessFsb','1')
    cp.set('General','Client1','localhost')
    ips = raw_input("请输入需要增加共享的客户端的IP个数： ")
    for i in range(0,int(ips)):
        client = "Client"+str(i+count+1)
        ip = raw_input("请输入客户端ip地址")
        cp.set('General',client,ip)
    cp.write(open("/etc/wibu/CodeMeter/Server.ini","w"))
    print u"服务端增加共享的IP完成"
    conf.close()
    
def del_server():
    conf = open('/etc/wibu/CodeMeter/Server.ini','r')
    lines =conf.readlines()
    count = 0
    dic_ips = {}
    for line in lines:
        if re.match('Client',line,flags=0):
            count += 1
            dic_ips[str(count)]=line
        elif re.match('client',line,flags=0):
            count += 1
            dic_ips[str(count)]=line
        else:
            pass
    if len(dic_ips) > 0:
        for key in dic_ips:
            print key ,' ',dic_ips[key]
        print u"请输入要删除共享客户端的的序号，多个序号用英文逗号隔开，删除全部请输入all"
        del_number = raw_input()
        get_number = del_number.split(',')
        print get_number
        for number in get_number:
            if number == 'all':
                for key in dic_ips:
                    ip_key = dic_ips[key].split(' = ')[0]
                    print ip_key
                    #cp.remove_option('General', ip_key)
                    print cp.remove_option('General', ip_key)
                break
            else:
                ip_key = dic_ips[str(number)].split(' = ')[0]
                print ip_key
                print cp.remove_option('General', ip_key)
    else:
        print u"没有配置访问ip"
    cp.write(open("/etc/wibu/CodeMeter/Server.ini","w"))
    conf.close()
    
if __name__ == '__main__':
    os.popen('service codemeter stop')
    while(True):
        type = raw_input("输入数字：0配置服务端，1配置客户端,2服务端追加访问ip，3服务端删除访问ip，4退出")
        type = int(type)
        cp.read('/etc/wibu/CodeMeter/Server.ini')
        if type == 0:
            server()
        elif type == 1:
            client()
        elif type == 2:
            add_server()
        elif type == 3:
            del_server()
        elif type == 4:
            print u"退出加密狗共享配置"
            break
        else:
            print u"您输入的代码不正确，请重新输入"
    os.popen('service codemeter start')