Exemple #1
0
    def __init__(self,version=None,year=None):
        Data.__init__(self)
        # 设置数据库
        self.database = AdminDatabase()

        # 设置最新版本
        self.latestversion = self.database.version()[-1]

        # 设置版本和年份
        if (version is None) and (year is None):
            self.version = self.latestversion
            self.year = re.split('_',self.version)[0]
        elif version is not None:
            self.version = version
            self.year = re.split('_',self.version)[0]
        else:
            self.year = str(year)
            self.version = self.database.version(self.year)[-1]
Exemple #2
0
class AdminData:
    '''AdminData类表示行政区域数据

    :param str version: 颁布的行政区划的版本号
    :param int,str year: 颁布的行政区划的年份
    :var AdminDatabase database: 行政区划数据库
    :var str latestversion: 最新颁布的行政区划版本号
    :var str year: 年份
    :var str version: 版本号
    :var list Province: 某个版本的所有省级行政区划
    :var list Prefecture: 某个版本的所有地级行政区划
    :var list County: 某个版本的所有县级行政区划
    :var list ProvincePrefecture: 某个版本的所有省级和地级行政区划
    :var list ProvincePrefectureCounty: 某个版本的所有省级地级县级区划

    通用接口__getitem__

    :param str,list key: 查询条件。其中的key表示表示区域名称,方法如下:
    (1)省级区域,用名字直接表示,例如ad[u'北京'];
    (2)地级区域,用省级、地级名称表示,例如ad[u'浙江',u'嘉兴'];(3)县级区域,用省级、地级、县级名称表示,例如ad[u'湖北', u'恩施',u'来凤'];
    (4)如果是本级行政区域加上下级行政区域,下级行政区域用f(first)表示,例如表示浙江省及其所有地级行政区域,用ad[u'浙江',u'f'];
    (5)如果是本级行政区域加上下下级行政区域,下下级行政区域用s(second)表示,例如表示浙江省及其所有县级行政区域,用ad[u'浙江',u's'];
    (6)如果是本级行政区域加上下级及下下级行政区域,下级及下下级行政区域用b(both)表示,例如表示浙江省及其所有地县级行政区域,用ad[u'浙江',u'b']。返回值是数据库中查询得到的行政区划的列表。
    '''
    def __init__(self,version=None,year=None):
        Data.__init__(self)
        # 设置数据库
        self.database = AdminDatabase()

        # 设置最新版本
        self.latestversion = self.database.version()[-1]

        # 设置版本和年份
        if (version is None) and (year is None):
            self.version = self.latestversion
            self.year = re.split('_',self.version)[0]
        elif version is not None:
            self.version = version
            self.year = re.split('_',self.version)[0]
        else:
            self.year = str(year)
            self.version = self.database.version(self.year)[-1]

    # 通用行政区划查询接口
    def __getitem__(self, key):
        '''通用行政区划查询接口

        :param key:
        :return: 查询结果
        '''
        # f to get all first level
        # s to get all second level
        # b to get all first and second level
        if isinstance(key,str):
            if re.match('^s$',key):
                return self.Province
            if re.match('^t$',key):
                return self.Prefecture
            if re.match('^f$',key):
                return self.County
            return self.get_province_by_name(key)

        if isinstance(key,tuple) and len(key) < 2:
            if re.match('^s$',key[0]):
                return self.Province
            if re.match('^t$',key[0]):
                return self.Prefecture
            if re.match('^f$',key[0]):
                return self.County
            return self.get_province_by_name(key[0])

        if isinstance(key,tuple):
            if len(key) < 3:
                if re.match(key[1],u'f') is not None:
                    result = self.get_prefecture_children(key[0])
                    return self._sorted(result)
                elif re.match(key[1],u's') is not None:
                    prefectures = self.get_prefecture_children(key[0])
                    result = []
                    for item in prefectures:
                        result.extend(self.get_county_children(key[0],item['region']))
                    return self._sorted(result)
                elif re.match(key[1],u'b') is not None:
                    prefectures = self.get_prefecture_children(key[0])
                    result = []
                    for item in prefectures:
                        result.append(item)
                        result.extend(self.get_county_children(key[0],item['region']))
                    return self._sorted(result)
                else:
                    return self.get_prefecture_by_name(key[0],key[1])
            else:
                if re.match(key[2],u'f') is not None:
                    result = self.get_county_children(key[0],key[1])
                    return self._sorted(result)
                else:
                    return self.get_county_by_name(key[0],key[1],key[2])

    def get_by_acode(self,acode,year=None):
        '''通过行政区划代码(acode)查询行政区划

        :param str acode: 行政区划代码
        :param int,str year: 年份
        :return: 查询结果
        :rtype: list
        '''
        if year is None:
            return list(self.database.find(acode=acode,version=self.version))
        else:
            version = self.database.version(year)[-1]
            return list(self.database.find(acode=acode,version=version))

    def get_province_by_name(self,province):
        '''通过省级名称查询行政区划

        :param str province: 省级行政区划名称
        :return: 省级行政区划单位
        :rtype: list
        '''
        province_pattern = u'省|市|自治区|维吾尔自治区|回族自治区|壮族自治区'
        province = re.split(province_pattern,re.sub('\s+','',province))[0]
        mprovince = '^' + province +'$'
        result = [item for item in self.Province if re.match(mprovince,re.split(province_pattern,item['region'])[0]) is not None]
        if len(result) < 1:
            return []
        return  result

    def get_prefecture_by_name(self,province,prefecture):
        '''通过省级地级名称查询行政区划

        :param str province: 省级行政区划名称
        :param str prefecture: 地级行政区划名称
        :return: 地级行政区划单位
        :rtype: list
        '''
        prefectures =  self.get_prefecture_children(province)
        result = [item for item in prefectures if re.match(prefecture,item['region']) is not None]
        if len(result) < 1:
            return []
        return result

    # 获得一个县级单位
    def get_county_by_name(self,province,prefecture,county):
        '''通过省级地级县级名称查询行政区划

        :param str province: 省级行政区划名称
        :param str prefecture: 地级行政区划名称
        :param str county: 县级行政区划名称
        :return: 县级行政区划单位
        :rtype: list
        '''
        counties =  self.get_county_children(province,prefecture)
        result = [item for item in counties if re.match(county,item['region']) is not None]
        if len(result) < 1:
            return []
        return result

    def get_prefecture_children(self,province):
        '''通过省级名称获得其辖区下的所有地级行政区划

        :param str province: 省级行政区划的名称
        :return: 某省级单位辖区下的所有地级区划单位
        :rtype: list
        '''
        province_found = self.get_province_by_name(province)
        if len(province_found) < 1:
            print('Can not find ',province)
            raise NameError
        if len(province_found) > 1:
            print('Tow much: ',province)
            raise NameError
        prefecture = self.database.find(parent=province_found[0]['_id'],version=self.version,sorts=[('acode',ASCENDING)])
        return list(prefecture)

    def get_county_children(self,province,prefecture):
        '''通过省级和地级区域名称获得其辖区下的所有县级行政区划

        :param str province: 省级行政区划名称
        :param str prefecture: 地级行政区划名称
        :return: 某个地级行政单位辖区下的所有县级区划单位
        :rtype: list
        '''
        # to find province item
        prefecture_found = self.get_prefecture_by_name(province,prefecture)
        if len(prefecture_found) < 1:
            print('Can not find ',province,'.',prefecture)
            raise NameError
        if len(prefecture_found) > 1:
            print('Tow much: ',province,'.',prefecture)
            raise NameError
        county = self.database.find(parent=prefecture_found[0]['_id'],version=self.version,sorts=[('acode',ASCENDING)])
        return list(county)

    # 设置版本
    def set_version(self,version):
        '''设置版本号

        :param str version: 颁布的行政区划版本号
        :return: 无返回值
        '''
        self.version = version
        self.year = re.split('_',self.version)[0]

    # 设置年份
    def set_year(self,year):
        '''设置年份

        :param str year: 年份
        :return: 无返回值
        '''
        self.year = str(year)
        self.version = self.database.version(self.year)[-1]

    # 所有的省级单位
    @property
    def Province(self):
        return self._sorted(list(self.database.find(adminlevel=2,version=self.version)))

    # 所有的省级单位
    @property
    def Prefecture(self):
        return self._sorted(list(self.database.find(adminlevel=3,version=self.version)))

    # 所有的省级单位
    @property
    def County(self):
        return self._sorted(list(self.database.find(adminlevel=4,version=self.version)))

    # 获得省级和地级单位
    @property
    def ProvincePrefecture(self):
        result = []
        provinces = self.Province
        for province in provinces:
            result.append(province)
            result.extend(self.get_prefecture_children(province['region']))
        return result

    # 获得省级、地级和县级单位
    @property
    def ProvincePrefectureCounty(self):
        result = []
        provinces = self.Province
        for province in provinces:
            result.append(province)
            prefectures = self.get_prefecture_children(province['region'])
            for prefecture in prefectures:
                result.append(prefecture)
                result.extend(self.get_county_children(province['region'],prefecture['region']))
        return result

    # 辅助排序函数
    def _sorted(self,regions):
        if len(regions) < 1:
            return []
        return sorted(regions,key = lambda x:x['acode'])