匹配指定的ip属于哪个地址段

匹配指定的ip属于哪个地址段

作者:梅劲松
程序提供:hoxide

因为写DNS服务器，来根据客户端的访问,返回指定ip地址.所以需要用到匹配ip属于哪个地址段的功能.hoxide帮忙用python实现了出来,在这里表示感谢.代码如下 :
注意:hoxide注明本程序为MIT 授权.

class NetPool:
    def __init__(self):
        self.Pool = dict()
        self.Pool['net'] = '0.0.0.0'
       
    def add(self, ip):
        a = map(int, ip.split('.'))
        p = self.Pool
        for i in range(len(a)-1, -1, -1):
            if a[i] != 0 :
                k = i+1
                break
        for i in range(k):
            ia = a[i]
            try:
                p = p[ia]
            except:
                p[ia] = dict()
                p = p[ia]
        p['net'] = ip
    def get(self, ip):
        a = map(int, ip.split('.'))
        p = self.Pool
        st = list()
        for ia in a:
            try:
                if ia == 0:
                    st.append(p)
                p = p[ia]
            except:
                break
        st.append(p)
        while st:
            p = st.pop()
            try:
                return p['net']
            except:
                pass
        return 'no such net'

def test():
    import pprint
    pp = pprint.PrettyPrinter()
    P = NetPool()
    Net = [
        '11.1.1.0',
        '12.3.2.0',
        '11.1.2.0',
        '12.0.0.0',
        '11.0.1.0',
        '11.0.0.0'
        ]
    IP = [
        '11.1.1.2',
        '12.3.2.5',
        '11.1.2.2',
        '12.5.7.8',
        '11.0.1.9',
        '11.253.9.1',
        '254.254.255.255'
        ]
    #print 'NET'
    for ip in Net:
        print 'NET: ', ip
        P.add(ip)
    for ip in Net:
        print 'NET: ', ip
        P.add(ip)
    pp.pprint(P.Pool)
    for ip in IP:
        print ip, ' in ', P.get(ip)
       
if __name__ == '__main__':
    test()