python3 高德常用方法归类

高德秘钥的设置以及检测

import requests
import time
import csv
from math import ceil


class KEY:
    def __init__(self):
        self.keys = list(csv.reader(open(u'高德秘钥.csv', 'r')))
        self.keys_length = len(self.keys)
        self.i = 0

    def get_key(self):
        return self.keys[self.i][1]

    def next_key(self):
        self.i += 1
        if self.i == self.keys_length:
            self.i = 0
            now_time = time.localtime(time.time())
            time.sleep((23-now_time[3])*3600+(60-now_time[4])*60)
        return self.get_key()

    def check_out(self):
        lon = 104.12159
        lat = 30.65447
        while self.i < self.keys_length:
            print(self.i)
            base_url = 'https://restapi.amap.com/v3/assistant/coordinate/convert'
            _params_ = {'locations': '{},{}'.format(lon, lat),
                        'coordsys': 'gps', 'output': 'json', 'key': self.keys[self.i][1]}

            infocode = int(requests.get(base_url, _params_).json()['infocode'])
            if infocode == 10001:
                print(self.keys[self.i])

            self.i += 1
            time.sleep(1)

代码测试

k = KEY()
k.check_out()

常用方法总结

class Method:
    def __init__(self):
        self.keys = KEY()
        self.k = self.keys.get_key()

    # 访问网址
    def get_response(self, base_url_, params_):
        """
        :param base_url_:  网址
        :param params_: 访问网址的参数字典
        :return: 访问网址的响应内容
        """
        while True:
            try:
                response_ = requests.get(base_url_, params_)
                response_ = response_.json()
                infocode = int(response_['infocode'])
                if infocode != 10000:
                    print(infocode, self.k)
                    self.k = self.keys.next_key()
                    params_['key'] = self.k
                else:
                    break
            except requests.exceptions.ConnectionError:
                print('wait a moment')
                for i in range(60, 0, -1):
                    print("倒计时{}秒".format(i), flush=True, end=""*5)
                    time.sleep(1)
            except Exception as e:
                print(e)
        return response_

    # gps坐标转化高德坐标
    def exchange(self, coordinate):
        """
        :param coordinate: 坐标 经度，纬度|经度，纬度……
        :return: 高德坐标
        """
        base_url = 'https://restapi.amap.com/v3/assistant/coordinate/convert'
        _params_ = {'locations': coordinate, 'coordsys': 'gps', 'output': 'json', 'key': self.k}

        return self.get_response(base_url, _params_)['locations'].replace(';', '|')

    # 获取行政区信息
    def get_region(self, *args, coordinate='',  batch=False):
        """
        :param coordinate: 坐标字符串
        :param args: 坐标 经度，纬度|经度，纬度……
        :param batch: 批量查询
        :return: 省，市，区（县）,详细地址
        """
        if coordinate == '':
            temp = []
            for i in range(0, len(args), 2):
                temp.append('{},{}'.format(args[i], args[i+1]))
            str_coordinate = '|'.join(temp)
        else:
            str_coordinate = coordinate

        base_url = 'https://restapi.amap.com/v3/geocode/regeo'
        _params_ = {'location': self.exchange(str_coordinate), 'extensions': 'all',
                    'output': 'json', 'key': self.k, 'batch': str(batch).lower()}

        response = self.get_response(base_url, _params_)

        if not batch:
            province = response['regeocode']['addressComponent']['province']
            city = response['regeocode']['addressComponent']['city']
            district = response['regeocode']['addressComponent']['district']
            address = response['regeocode']['formatted_address']
            pois = [[r['type'].split(';')[0], float(r['distance'])] for r in response['regeocode']['pois'][:3]]

            return province, city, district, address, pois
        else:
            regeocodes = response['regeocodes']
            result = []
            for r in regeocodes:
                result.append(
                    (
                        r['addressComponent']['province'],
                        r['addressComponent']['city'],
                        r['addressComponent']['district'],
                        r['formatted_address'],
                        [[i['type'].split(';')[0], float(i['distance'])] for i in r['pois'][:3]]
                    )
                )
            return result

    # 获取POI信息（多边形搜索--矩形搜索）
    def get_poi(self, lon, lat, lon_gap, lat_gap, poi_type, output_poiname=False):
        """
        :param lon: 矩形中心经度
        :param lat: 矩形中心纬度
        :param lon_gap: 矩形经度间隔
        :param lat_gap: 矩形纬度间隔
        :param poi_type: 查询POI类型
        :param output_poiname: 输出POI name
        :return: 相应poi数量
        """
        base_url = 'https://restapi.amap.com/v3/place/polygon'
        params = {'polygon': self.exchange('{},{}|{},{}'.
                                           format(lon-lon_gap/2, lat+lat_gap/2, lon+lon_gap/2, lat-lat_gap/2)),
                  'types': poi_type, 'output': 'json', 'key': self.k, 'page': 1, 'offset': 20}

        response = self.get_response(base_url, params)
        n = int(response['count'])

        if output_poiname:
            print('Crawling POI {} name'.format(poi_type))
            name = ['{}({})'.format(e['name'], e['typecode']) for e in response['pois']]
            while params['page'] < ceil(n/20):
                params['page'] += 1
                name += ['{}({})'.format(e['name'], e['typecode']) for e in self.get_response(base_url, params)['pois']]

            return n, name
        else:
            return n

    def get_distance(self, o_lon, o_lat, d_lon, d_lat):
        base_url = 'https://restapi.amap.com/v3/direction/driving'
        parameters = {
            'key': self.k, 'strategy': 2, 'nosteps': 1, 'extensions': 'base',
            'origin': self.exchange('{},{}'.format(o_lon, o_lat)),
            'destination': self.exchange('{},{}'.format(d_lon, d_lat))
        }

        response = self.get_response(base_url, parameters)

        return float(response['route']['paths'][0]['distance'])

代码测试

lon_ = 104.12159
lat_ = 30.65447
m = Method()
print(m.get_region(lon_, lat_, lon_, lat_, batch=True))

lon_ = 104.12159
    lat_ = 30.65447
    m = Method()
    print(m.get_region(lon_, lat_, lon_, lat_, batch=True))