MongoDB Nearby Search

DEFAULT_GEO_RADIUS = 50
EARTH_RADIUS_IN_KM = 6371
EARTH_RADIUS_IN_MILES = 3959


if comparator == "within":
    comparator = "$within"
    params = value.split(",")

    if len(params) > 3:
        coords = [float(c) for c in params]
        value = {
            "$box": zip(*[coords[i::2] for i in range(2)]),
        }
    else:
        if len(params) > 2:
            lon, lat, radius = params
        else:
            lon, lat = params
            radius = DEFAULT_GEO_RADIUS

        value = {
            "$center": [[float(lon), float(lat)], float(radius) / EARTH_RADIUS_IN_KM],
        }
elif comparator == "near":
    comparator = "$near"
    params = value.split(",")

    if len(params) > 2:
        lon, lat, radius = params
    else:
        lon, lat = params
        radius = DEFAULT_GEO_RADIUS

    value = [float(lon), float(lat), float(radius) / EARTH_RADIUS_IN_KM]