python實現mongo自動新增地圖索引, 自動修正遇到的問題
阿新 • • 發佈:2018-11-09
mongo新增地圖索引時常會遇到兩個問題: 座標點衝突和座標點重合, 所以之前寫了個指令碼解決一下, 主要思路就是:
當新增地圖索引報錯時, 根據error資訊,判斷座標點衝突or 座標點重合的地方, 根據正則修正座標點, 然後通過遞迴or迴圈的方式, 重新嘗試, 程式碼如下:
-*- coding: utf-8 -*- import pymongo import re conn = pymongo.MongoClient(host='127.0.0.1', port=27017) citytest2 = conn."db_name"."collection_name" # 連線表 i = 0 # 用來統計刪除座標點次數及防止棧溢位 def decoordinates(citytest2): ''' 名稱: decoordinates 功能: 新增座標索引,若發生座標衝突,則刪除靠後的座標,直至成功(座標點相同問題尚未新增) return: None ''' try: global i citytest2.create_index([("polygons", "2dsphere")]) print(i) except Exception as e: global i e = str(e) retest = re.compile(r'\[(.*?)\]') if "Duplicate vertices" in str(e): # 座標重合問題 retest2 = re.compile(r'and (\d+)') test2 = int(retest2.findall(str(e))[0]) # 獲取出錯在第幾條 print test2 elif "locations in degrees" in str(e): # 座標衝突問題 retest2 = re.compile(r'(\d+?)\s+?cross') test2 = retest2.findall(str(e))[0] # 獲取出錯在第幾條 i += 1 idtest = re.compile(r'_id:\s+(\d+)') idtest = idtest.findall(str(e))[0] # 獲取出錯id test = ' [' + retest.findall(str(e))[int(test2)].strip() + '],' # 獲取該條出錯的座標 errortest = citytest2.find_one({"_id": int(idtest)}) e = eval(str(errortest).replace(test, '')) citytest2.update({"_id": int(idtest)}, e) if i < 1000: decoordinates(citytest2) if name == 'main': decoordinates(citytest2)