84 lines
2.5 KiB
Python
84 lines
2.5 KiB
Python
import logging
|
|
from typing import Dict
|
|
from uuid import uuid4
|
|
|
|
from django.contrib.auth import get_user_model
|
|
from django.conf import settings
|
|
from django.db import models
|
|
from django.urls import reverse
|
|
from django_extensions.db.models import TimeStampedModel
|
|
from scrobbles.mixins import ScrobblableMixin
|
|
|
|
logger = logging.getLogger(__name__)
|
|
BNULL = {"blank": True, "null": True}
|
|
User = get_user_model()
|
|
|
|
|
|
class GeoLocation(ScrobblableMixin):
|
|
COMPLETION_PERCENT = getattr(settings, "LOCATION_COMPLETION_PERCENT", 100)
|
|
|
|
uuid = models.UUIDField(default=uuid4, editable=False, **BNULL)
|
|
lat = models.FloatField()
|
|
lon = models.FloatField()
|
|
altitude = models.FloatField(**BNULL)
|
|
|
|
class Meta:
|
|
unique_together = [["lat", "lon", "altitude"]]
|
|
|
|
def __str__(self):
|
|
if self.title:
|
|
return self.title
|
|
|
|
return f"{self.lat} x {self.lon}"
|
|
|
|
def get_absolute_url(self):
|
|
return reverse(
|
|
"locations:geo_location_detail", kwargs={"slug": self.uuid}
|
|
)
|
|
|
|
@property
|
|
def truncated_lat(self):
|
|
return float(str(self.lat)[:-3])
|
|
|
|
@property
|
|
def truncated_lan(self):
|
|
return float(str(self.lon)[:-3])
|
|
|
|
@classmethod
|
|
def find_or_create(cls, data_dict: Dict) -> "GeoLocation":
|
|
"""Given a data dict from GPSLogger, does the heavy lifting of looking up
|
|
the location, creating if if doesn't exist yet.
|
|
|
|
"""
|
|
# TODO Add constants for all these data keys
|
|
if "lat" not in data_dict.keys() or "lon" not in data_dict.keys():
|
|
logger.error("No lat or lon keys in data dict")
|
|
return
|
|
|
|
data_dict["lat"] = float(data_dict.get("lat", ""))
|
|
data_dict["lon"] = float(data_dict.get("lon", ""))
|
|
data_dict["altitude"] = float(data_dict.get("alt", ""))
|
|
|
|
location = cls.objects.filter(
|
|
lat=data_dict.get("lat"),
|
|
lon=data_dict.get("lon"),
|
|
altitude=data_dict.get("alt"),
|
|
).first()
|
|
|
|
if not location:
|
|
location = cls.objects.create(
|
|
lat=data_dict.get("lat"),
|
|
lon=data_dict.get("lon"),
|
|
altitude=data_dict.get("alt"),
|
|
)
|
|
return location
|
|
|
|
|
|
class RawGeoLocation(TimeStampedModel):
|
|
user = models.ForeignKey(User, on_delete=models.CASCADE)
|
|
lat = models.FloatField()
|
|
lon = models.FloatField()
|
|
altitude = models.FloatField(**BNULL)
|
|
speed = models.FloatField(**BNULL)
|
|
timestamp = models.DateTimeField(**BNULL)
|