いまMongoDBと、その地理空間クエリをいろいろ試しているところです。なかなか面白いので、PyMongoでどう使っているかをまとめてみようと思います！

まずはおなじみのpip install pymongoが必要です。最初にMongoクライアントを用意しますが、これはとても簡単です。

from pymongo import MongoClient

client = MongoClient()

なかなか便利で、デフォルトではMongoClientはmongodb://localhost:27017に接続します。

ただ、実際には次のようにしておくとよいかもしれません。

url = os.environ.get("MONGO_URL", "mongodb://localhost:27017")
client = MongoClient(url)

地理空間検索を機能させるには、ドキュメントのフィールドにMongoDBのインデックスを作成する必要があります。ここでは2dsphereインデックスを作成します。

from pymongo import MongoClient, GEOSPHERE, DESCENDING

client = MongoClient()
db = client.foo
db.bar.create_index([("location", GEOSPHERE)])

あとはドキュメントを挿入してクエリを実行するだけです :) locationフィールドだけを持つドキュメントを挿入してみます。

db.bar.insert_one({
	“location”: {
		"coordinates" : [ 
			51.4982563,
			-0.0861183
    		],
    		"type" : "Point"
	}
})

locationフィールドはGeoJSON形式である必要があります。GeoJSONは地理的特徴を表現するためのJSON標準です。これがあるとドキュメントの検索がとても簡単になります！

db.bar.find({
	"location": {
		"$near": {
			"$geometry": {"type": "Point", "coordinates": [LAT, LONG]},
			"$maxDistance": range,
		}
	}
})