china_ip.py

import csv
import requests
import itertools

from netaddr import cidr_merge, iprange_to_cidrs
from ipaddress import (summarize_address_range,
                       collapse_addresses, IPv4Address)

class AddressBlock(object):
    def __init__(self, country_code):
        self.country_code = country_code

    @property
    def blocks(self):
        if not hasattr(self, '_block'):
            res = requests.get(
                'http://www.nirsoft.net/countryip/{}.csv'.format(self.country_code))
            raw_blocks = [row.split(',') for row in res.text.split('\r\n') if row]
            self._blocks = [{'start': row[0], 'end': row[1]} for row in raw_blocks]

        return self._blocks

    @property
    def networks(self):
        if not hasattr(self, '_networks'):
            # self._networks = collapse_addresses(
                # chain(summarize_address_range(
                    # block['start'], block['end']) for block in self.blocks))
            self._networks = []
            for i, block in enumerate(self.blocks):
                self._networks += summarize_address_range(
                    IPv4Address(block['start']),
                    IPv4Address(block['end']))
                print('{}/{} has done, here are {} networks'.format(i, len(self.blocks), len(self._networks)))
            self._networks = collapse_addresses(self._networks)

        return self._networks



if __name__ == '__main__':
    ab = AddressBlock('cn')
    print(ab.networks)