New feature: administrative region access

Author: zeoFura

readme.md

@@ -9,6 +9,8 @@
 - [下载天地图切片](src/download_pic/pic_download.py)
     - download_pic 下载函数
     - merge_pic 合并函数
+- [天地图行政区获取](src/tianditu/ez_region.py)
+    - download_region 行政区下载函数
 - [根据颜色进行分类提取](src/classify/groupIMG.py)
     - img_color_cov 提取函数
 

resource/tianditu_zhongguo_region.json

@@ -0,0 +1,3 @@
+#! /usr/bin/env python
+# -*- coding: utf-8 -*-
+# __file__: __init__.py

resource/tianditu_zhongguo_region_2.json

@@ -0,0 +1,141 @@
+#! /usr/bin/env python
+# -*- coding: utf-8 -*-
+# __file__: administrative_region
+# 天地图行政区获取
+import datetime
+import requests
+import json
+import codecs
+import pymongo
+
+key = "a4ee5c551598a1889adfabff55a5fc27"
+
+client = pymongo.MongoClient(host='localhost', port=27017)
+
+db = client.lvyou
+collection = db.tianditu
+
+
+def points_calc(shi, points):
+    """
+    将文本坐标转换成浮点坐标
+    :param shi: 市级行政区名称
+    :param points: "116.403 39.972,116.402 39.97,116.405 39.97,116.405 39.964,"
+    :return: [[x1,y1],[x2,y2]]
+    """
+    x_list = []
+    y_list = []
+    for i in points:
+        s = [x for x in i['region'].split(",")]
+        s = [x.split(' ') for x in s]
+        for j in s:
+            x_list.append(eval(j[0]))
+            y_list.append(eval(j[1]))
+
+    res = [[x, y] for x, y in zip(x_list, y_list)]
+
+    polygon = {'type': 'Polygon', 'coordinates': [res]}
+    return polygon
+
+
+def xianggang_calc(chi):
+    """
+    处理香港的数据
+    :param chi:
+    :return:
+    """
+
+    for i in chi['child']:
+        sheng_point = i['points']
+        sheng_name = i['name']
+        sheng_reg = points_calc(sheng_name, sheng_point)
+        i['geom_tey'] = sheng_reg
+
+    pass
+
+
+def do_zizhiqu(zz):
+    """
+    处理自治区
+    :param zz: 自治区
+    :return:
+    """
+    for ii in zz:
+        # 市区
+        s = calc_rege(ii)
+        ii['geom_tey'] = s
+        cs = ii.get('child')
+        if cs:
+            for c in cs:
+                saa = calc_rege(c)
+                c['geom_tey'] = saa
+    pass
+
+
+def calc_rege(chi):
+    """
+    计算区域
+    :param chi:
+    :return:
+    """
+    aomen = chi['name']
+    aomen_point = chi.get('points')
+    if aomen_point:
+        aomen_reg = points_calc(aomen, aomen_point)
+        return aomen_reg
+    else:
+        return None
+
+
+def run():
+    """
+    运行函数
+    :return:
+    """
+    r = requests.get(
+        'http://api.tianditu.gov.cn/administrative?postStr={"searchWord":"中国","searchType":"1","needSubInfo":"true","needAll":"true","needPolygon":"true","needPre":"true"}&tk=a4ee5c551598a1889adfabff55a5fc27')
+
+    data = json.loads(r.text)
+
+    data['spider_time'] = datetime.datetime.now().strftime('%Y-%m-%d %H-%M-%S')
+    data = data['data']
+    for item in data:
+
+        child_list = item['child']
+
+        for chi in child_list:
+            # 处理的是省
+            chi_reg = calc_rege(chi)
+            chi['geom_tey'] = chi_reg
+            if chi['name'] == '澳门':
+                aomen_reg = calc_rege(chi)
+                chi['geom_tey'] = aomen_reg
+            elif chi['name'] == '香港':
+                xianggang_calc(chi)
+            else:
+                chi_chi = chi['child']
+                sheng_point = chi['points']
+                sheng_name = chi['name']
+
+                if '自治区' in sheng_name:
+                    do_zizhiqu(chi_chi)
+
+                else:
+                    for ii in chi_chi:
+                        # 市区
+                        s = calc_rege(ii)
+                        ii['geom_tey'] = s
+
+                        cs = ii.get('child')
+                        if cs:
+                            for c in cs:
+                                saa = calc_rege(c)
+                                c['geom_tey'] = saa
+
+            collection.insert(chi)
+
+    pass
+
+
+if __name__ == '__main__':
+    run()

src/tianditu/__init__.py

@@ -0,0 +1,31 @@
+#! /usr/bin/env python
+# -*- coding: utf-8 -*-
+# __file__: ez_region
+
+import requests
+import json
+import pymongo
+
+def download_region():
+    """
+    简易的天地图行政区下载
+    :return: null
+    """
+
+    client = pymongo.MongoClient(host='localhost', port=27017)
+
+    db = client.lvyou
+    collection = db.tianditu_zhongguo_region
+
+    r = requests.get(
+        'http://api.tianditu.gov.cn/administrative?postStr={"searchWord":"中国","searchType":"1","needSubInfo":"true","needAll":"true","needPolygon":"true","needPre":"true"}&tk=a4ee5c551598a1889adfabff55a5fc27')
+
+    data = json.loads(r.text)
+    data = data['data']
+    for item in data:
+        child_list = item['child']
+        for chi in child_list:
+            collection.insert(chi)
+
+if __name__ == '__main__':
+    download_region()