Closed tejon-melero closed 6 years ago
I think it would be fairly easy to implement geopoint field. Below is very simple implementation offcourse it is not a full one.
from motorengine.fields import *
from pymongo import GEOSPHERE, GEO2D
from motorengine.query.base import QueryOperator
from motorengine.utils import serialize, deserialize
class GeoPointField(BaseField):
def __init__(self, *args, **kwargs):
super(GeoPointField, self).__init__(*args, **kwargs)
def validate(self, lst):
if not isinstance(lst, (list, )):
return False
return True
def to_son(self, lst):
longitude = lst[1]
latitude = lst[0]
value = {"type": "Point", "coordinates": [latitude, longitude]}
return value
def from_son(self, jsn):
valued = jsn
longitude = valued.get("coordinates")[1]
latitude = valued.get("coordinates")[0]
return [latitude, longitude]
class GeoNearOperator(QueryOperator):
def to_query(self, field_name, value):
return {
field_name: {
"$near": {
"$geometry": {
"type": "Point",
"coordinates": list(value[0])
},
"$minDistance": value[1]
}
}
}
def get_value(self, field_name, raw_value):
return raw_value
class GeoSphearNearOperator(QueryOperator):
EARTH_RADIOUS = 3963.2
def to_query(self, field_name, value):
return {
field_name: {
"$geoWithin": {
"$centerSphere": [list(value[0]),
value[1]/self.EARTH_RADIOUS]
}
}
}
def get_value(self, field_name, raw_value):
return raw_value
class Events(MotorEngineDocument):
__indexes__ = [('location', GEOSPHERE)]
name = StringField(required=True)
tid = StringField(required=True)
event_on = DateTimeField(required=True)
location = GeoPointField(required=True)
@classmethod
async def nearby(cls, lat, lon, radious, limit=10, skip=0):
results = await cls.objects.limit(limit).skip(skip)\
.filter(location__around=[(lat, lon), radious])\
.find_all()
return results
But you need to be sure that the motorengine operators are updated properly before geo query.
from motorengine.query_builder.transform import OPERATORS
OPERATORS.update({
"near": GeoNearOperator,
"around": GeoSphearNearOperator,
"search": TextSearch
})
@tejon-melero Go ahead and close the issue if it works. Feel free to reopen at anytime.
I have the following model:
I used the mongoengine.GeoPointField to perform queries like find all objects near to given coordinates:
Howewer the GeoPointField field is not available in motorengine. Is it possible to define objects and use queries like this with motorengine? And if not is there a workaround for this kind of use case? related stackoverflow question: https://stackoverflow.com/questions/49215097/what-is-the-equivalent-of-mongoengine-geopointfield-in-motorengine-to-perform-ne