python高德 查询县_Python获取高德地图POI

# -*- encoding: utf-8 -*-

# coding:utf-8

from xlwt import Workbook

from tempfile import TemporaryFile

import urllib.request

import xml.dom.minidom as minidom

import xlwt

#结果目录

inforst = '/home/yuhz/PycharmProjects/untitled2/POI/'

file_name = 'result.txt' # write result to this file

#keyword = 'XXX' urllib.parse.quote(keyword)

url_amap = 'http://restapi.amap.com/v3/place/text?&keywords=&types=010800&city=370600&citylimit=true&&output=xml&offset=20&page=1&key=&extensions=base'

#facility_type = r'types=170300' # factory facilities

#region = r'city=120113' # beichen of tianjin

each_page_rec = 20 # results that displays in one page

which_pach = r'page=1' # display which page

xml_file = 'tmp.xml' # xml filen name

#写入Excel(定义Excel表头)

book = Workbook()

sheet1 = book.add_sheet('Sheet 1')

row1 = sheet1.row(0)

row1.write(0,'CORPNAME')

row1.write(1,'ADDRESS')

row1.write(2,'TEL')

row1.write(3,'pname')

row1.write(4,'cityname')

row1.write(5,'adname')

row1.write(6,'location')

sheet1.col(0).width = 10000

sheet1.col(1).width = 10000

sheet1.col(2).width = 10000

sheet1.col(3).width = 5000

sheet1.col(4).width = 5000

sheet1.col(5).width = 5000

sheet1.col(6).width = 5000

#自动换行

style = xlwt.easyxf('align: wrap on')

# get html by url and save the data to xml file

def getHtml(url):

page = urllib.request.urlopen(url)

html = page.read()

try:

# open xml file and save data to it

with open(xml_file, 'wb') as xml_file_handle:

xml_file_handle.write(html)

except IOError as err:

print("IO error: " + str(err))

return -1

return 0

# phrase data from xml

def parseXML(index):

total_rec = 1 # record number

# open xml file and get data record

try:

with open(file_name, 'a') as file_handle:

dom = minidom.parse(xml_file)

root = dom.getElementsByTagName("response") # The function getElementsByTagName returns NodeList.

for node in root:

total_rec = node.getElementsByTagName("count")[0].childNodes[0].nodeValue

pois = node.getElementsByTagName("pois")

for poi in pois[0].getElementsByTagName('poi'):

name = poi.getElementsByTagName("name")[0].childNodes[0].nodeValue

try:

address = poi.getElementsByTagName("address")[0].childNodes[0].nodeValue

except IndexError:

address = ""

try:

tel = poi.getElementsByTagName("tel")[0].childNodes[0].nodeValue

except IndexError:

tel = ""

try:

pname = poi.getElementsByTagName("pname")[0].childNodes[0].nodeValue

except IndexError:

pname = ""

try:

cityname = poi.getElementsByTagName("cityname")[0].childNodes[0].nodeValue

except IndexError:

cityname = ""

try:

adname = poi.getElementsByTagName("adname")[0].childNodes[0].nodeValue

except IndexError:

adname = ""

location = poi.getElementsByTagName("location")[0].childNodes[0].nodeValue

#写入Excel

index = index + 1

row1 = sheet1.row(index)

row1.write(0, name, style)

row1.write(1, address, style)

row1.write(2, tel, style)

row1.write(3, pname, style)

row1.write(4, cityname, style)

row1.write(5, adname, style)

row1.write(6, location, style)

except IOError as err:

print

"IO error: " + str(err)

return total_rec

if __name__ == '__main__':

index = 0

if getHtml(url_amap) == 0:

print

'parsing page 1 ... ...'

# parse the xml file and get the total record number

total_record_str = parseXML(index)

total_record = int(total_record_str)

if (total_record % each_page_rec) != 0:

page_number = total_record / each_page_rec + 2

else:

page_number = total_record / each_page_rec + 1

# retrive the other records

for each_page in range(2, int(page_number)):

index = index + 20

print('parsing page ' + str(each_page) + ' ... ...')

url_amap = url_amap.replace('page=' + str(each_page - 1), 'page=' + str(each_page))

getHtml(url_amap)

total_record_str = parseXML(index)

total_record = int(total_record_str)

if total_record == 0:

break

else:

print

'error: fail to get xml from amap'

# 保存Excel

book.save(inforst + 'result.xls')

book.save(TemporaryFile())