【python】获取高德地图省市区县列表
项目中需要用省市区来进行检索，原想高德地图肯定会有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方法不存在，有这个错误的话不用理会，可以正常运行，如果不设置那么在数据库插入的时候会报下面的一个错误：
UnicodeEncodeError:'latin-1' codec can't encode character
最后是上述方法的调用：
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）。
