码迷,mamicode.com
首页 > 编程语言 > 详细

【python】获取高德地图省市区县列表

时间:2015-08-03 22:21:26      阅读:320      评论:0      收藏:0      [点我收藏+]

标签:

项目中需要用省市区来进行检索,原想高德地图肯定会有API来获得这些数据,结果没有找到,有一个接口好像可以用,但是会附带大量的边界坐标点。

所以就不如自己把高德的省市区列表扒下来,自己写接口来完成这个功能。

看到高德地图的js的demo里面有这样的展示页面:http://lbs.amap.com/api/javascript-api/example/u/2001-2/,所以我就直接利用它来分析。

1. 省列表

省的列表是直接写死在这个界面里的,所以我也照搬,把省都直接写死:

provinceList = [北京市, 天津市, 河北省, 山西省, 内蒙古自治区, 辽宁省, 吉林省,黑龙江省, 上海市, 江苏省, 浙江省, 安徽省, 福建省, 江西省, 山东省,河南省, 湖北省, 湖南省, 广东省, 广西壮族自治区, 海南省, 重庆市,四川省, 贵州省, 云南省, 西藏自治区, 陕西省, 甘肃省, 青海省, 宁夏回族自治区, 新疆维吾尔自治区, 台灣, 香港特别行>政区, 澳门特别行政区];

2. 市列表

从市列表开始,就要扒高德地图的接口了,首先模仿浏览器的访问设置几个Http头:

send_headers = {
        Accept:*/*,
        Accept-Encoding:gzip, deflate, sdch,
        Accept-Language:zh-CN,zh;q=0.8,
        Connection:keep-alive,
        Host:restapi.amap.com,
        Referer:http://lbs.amap.com/fn/iframe/?id=3556,
}

然后是根据省名获取市列表的函数:

def getCity(province):
    print "Province->" + province
    url = "http://restapi.amap.com/v3/config/district?subdistrict=1&extensions=all&level=province&key=608d75903d29ad471362f8c58c550daf&s=rsv3&output=json&callback=jsonp_" + getJsonP() + "_&keywords=" + province;
    data = getData(url)
    item = data[districts][0];
    if item[citycode] == []:
        item[citycode] = ‘‘
    save([item[citycode], item[adcode], item[name], item[center], item[level], ‘‘])
    for item in data[districts][0][districts]:
        save([item[citycode], item[adcode], item[name], item[center], item[level], ‘‘])
        while 1:
            try:
                getDistrict(item)
                break
            except Exception, e:
                print retry: + item[name] + "->" + str(e)

这里有跨域访问用的getJsonP、发起网络请求的getData、存数据库的save,这三个函数的具体实现一会再说,先看其它逻辑,高德服务端返回数据是长得这个样子的:

技术分享

districts域就是搜索的结果,因为这是按名称进行搜索,而不是id,所以很可能会搜出来多个结果,当然省名不会重,但一样得到的是一个JSONArray,所以用data[‘districts‘][0]来得到省的信息,除了直辖市之外,省级单位的citycode都是空的,用item[‘citycode‘] = ‘‘给它一个空字符串做默认值,然后调用save方法把它存入数据库,在上图中也看到了,data[‘districts‘][0]里面还有一个districts域,这里面就是所有市的信息,同样把它们存入数据库,然后针对每个市,去获取区的列表,这里有可能会出现网络问题,所以加上了失败重试。

下面依次来看上面几个关键方法的实现:

1)getJsonP:

跨域访问是通过jsonp回调的方式进行的,函数名都是jsonp_xxxxx_()的形式,这个xxxxx是一个变化的随机数字,我采用依次递增的形式来生成这个jsonp的数字:

jsonp = 9999
def getJsonP():
    global jsonp
    jsonp = jsonp + 1
    if (jsonp > 99999):
        jsonp = 10000
    return str(jsonp)

这里我保证这个数字是一个5位数(理论上几位都可以),因为在图里可以看到,返回值是jsonp_xxxxx_({aaa})的形式,我们要分析json串需要把前后没用的字符去掉,如果长度固定就很好删了。

2)getData:

发起网络请求并把返回的数据转成JSONObject:

def getDataWithEx(url):
    req = urllib2.Request(url,headers=send_headers)
    r = urllib2.urlopen(req,timeout=30)

    if r.info().get(Content-Encoding) == gzip:
        buf = StringIO(r.read())
        f = gzip.GzipFile(fileobj=buf)
        data = f.read()
    else:
        data = r.read()
    return json.loads(data[13:-1])

def getData(url):
    while 1:
        try:
            response = getDataWithEx(url)
            break
        except Exception, e:
            print retry: + url + " with error " + str(e)
    return response

这里也加入了失败重试,一般的数据都要用gzip进行解码,但是我忘了是哪个了,好像是有一个区的数据很特别没有用gzip,所以这里要分开判断,getJsonP里面介绍了,返回的数据前后是写无用的字符,用data[13:-1]删掉。

3)save

这是往MySQL里存数据的函数,没有特殊的点需要介绍,直接上代码:

def save(value):
    try:
        conn=MySQLdb.connect(host=www.xxx.com,user=‘xxx,passwd=‘xxx,port=3306,charset="utf8")
        conn.select_db(‘test)
        cur=conn.cursor()
        cur.execute("insert into amap(citycode, adcode, name, center, level, areacode) values(%s,%s,%s,%s,%s,%s)",value)
        conn.commit()
        cur.close()
        conn.close()
    except MySQLdb.Error,e:
        print "Mysql Error %d: %s" % (e.args[0], e.args[1])

3. 区列表

跟获取城市列表相似:

def getDistrict(city):
    print "->City->" + city[name]
    url = "http://restapi.amap.com/v3/config/district?subdistrict=1&extensions=all&level=city&key=608d75903d29ad471362f8c58c550daf&s=rsv3&output=json&callback=jsonp_" + getJsonP() + "_&keywords=" + city[name]
    data = getData(url)
    for possible in data[districts]:
        if possible[adcode] == city[adcode]:
            for item in possible[districts]:
                save([item[citycode], item[adcode], item[name], item[center], item[level], ‘‘])
                while 1:
                    try:
                        getBusiness(item)
                        break
                    except Exception, e:
                        print retry: + item[name] + "->" + str(e)
            break

和获取市列表稍微不同的是这里不能直接data[‘districts‘][0]了,按市、区搜就有可能重名了,所以用城市的adcode来匹配,针对匹配上的市,遍历区的信息存入数据库,然后再针对区搜索商圈。

4. 商圈列表

跟获取市列表和获取区列表没有太大的区别:

def getBusiness(district):
    print "->->District->" + district[name]
    url = "http://restapi.amap.com/v3/config/district?subdistrict=1&extensions=all&level=district&key=608d75903d29ad471362f8c58c550daf&s=rsv3&output=json&callback=jsonp_" + getJsonP() + "_&keywords=" + district[name]
    data = getData(url)
    for possible in data[districts]:
        if possible[adcode] == district[adcode]:
            values=[]
            for item in possible[districts]:
                values.append((item[citycode], item[adcode], item[name], item[center], item[level], item[areacode]))
            saveAll(values)
            break

唯一不同的是直接用saveAll一次性存储所有的数据而不是一条条的save:

def saveAll(values):
    try:
        conn=MySQLdb.connect(host=www.xxx.com,user=xxx,passwd=xxx,port=3306,charset="utf8")
        conn.select_db(test)
        cur=conn.cursor()
        cur.executemany("insert into amap(citycode, adcode, name, center, level, areacode) values(%s,%s,%s,%s,%s,%s)",values)
        conn.commit()
        cur.close()
        conn.close()
    except MySQLdb.Error,e:
        print "Mysql Error %d: %s" % (e.args[0], e.args[1])

关键的逻辑就都介绍完了,下面是头部库的导入和编码设置:

# -*- coding: utf-8 -*-
#encoding=utf-8

import urllib2
import sys, json
from StringIO import StringIO
import gzip
import MySQLdb
import datetime

reload(sys)
sys.setdefaultencoding(utf-8)

这里要说的一点是:不知道为什么编译器会报错说setdefaultencoding方法不存在,有这个错误的话不用理会,可以正常运行,如果不设置会出现乱码。

最后是上述方法的调用:

starttime = datetime.datetime.now()
for province in provinceList:
    getCity(province)
print over
endtime = datetime.datetime.now()
print (endtime - starttime).seconds

扒数据的时间很长,n多小时,我忘了具体是多久了,而且为了避免出错我是把所有省分成了好几段分别扒的。

如果不想自己扒数据,想直接得到省市区的数据,可以直接下载我转存的sql文件。

也可以临时使用我的接口:

获取省列表:http://www.yxbilu.com/sites/tools/amap/province;

获取市列表:http://www.yxbilu.com/sites/tools/amap/city?adcode=110000(省的adcode);

获取区列表:http://www.yxbilu.com/sites/tools/amap/district?adcode=110100(市的adcode)。

 

【python】获取高德地图省市区县列表

标签:

原文地址:http://www.cnblogs.com/smarterplanet/p/4700375.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!