mirror of
https://github.com/pacnpal/thrillwiki_django_no_react.git
synced 2025-12-20 09:51:09 -05:00
initial geodjango implementation
This commit is contained in:
27
location/migrations/0004_add_point_field.py
Normal file
27
location/migrations/0004_add_point_field.py
Normal file
@@ -0,0 +1,27 @@
|
||||
# Generated by Django 5.1.2 on 2024-11-04 22:30
|
||||
|
||||
import django.contrib.gis.db.models.fields
|
||||
from django.db import migrations
|
||||
|
||||
|
||||
class Migration(migrations.Migration):
|
||||
|
||||
dependencies = [
|
||||
("location", "0003_alter_historicallocation_city_and_more"),
|
||||
]
|
||||
|
||||
operations = [
|
||||
migrations.AddField(
|
||||
model_name="location",
|
||||
name="point",
|
||||
field=django.contrib.gis.db.models.fields.PointField(
|
||||
blank=True,
|
||||
help_text="Geographic coordinates as a Point",
|
||||
null=True,
|
||||
srid=4326,
|
||||
),
|
||||
),
|
||||
migrations.DeleteModel(
|
||||
name="HistoricalLocation",
|
||||
),
|
||||
]
|
||||
52
location/migrations/0005_convert_coordinates_to_points.py
Normal file
52
location/migrations/0005_convert_coordinates_to_points.py
Normal file
@@ -0,0 +1,52 @@
|
||||
# Generated by Django 5.1.2 on 2024-11-04 22:21
|
||||
|
||||
from django.db import migrations, transaction
|
||||
from django.contrib.gis.geos import Point
|
||||
|
||||
def forwards_func(apps, schema_editor):
|
||||
"""Convert existing lat/lon coordinates to points"""
|
||||
Location = apps.get_model("location", "Location")
|
||||
db_alias = schema_editor.connection.alias
|
||||
|
||||
# Update all locations with points based on existing lat/lon
|
||||
with transaction.atomic():
|
||||
for location in Location.objects.using(db_alias).all():
|
||||
if location.latitude is not None and location.longitude is not None:
|
||||
try:
|
||||
location.point = Point(
|
||||
float(location.longitude), # x coordinate (longitude)
|
||||
float(location.latitude), # y coordinate (latitude)
|
||||
srid=4326 # WGS84 coordinate system
|
||||
)
|
||||
location.save(update_fields=['point'])
|
||||
except (ValueError, TypeError):
|
||||
print(f"Warning: Could not convert coordinates for location {location.id}")
|
||||
continue
|
||||
|
||||
def reverse_func(apps, schema_editor):
|
||||
"""Convert points back to lat/lon coordinates"""
|
||||
Location = apps.get_model("location", "Location")
|
||||
db_alias = schema_editor.connection.alias
|
||||
|
||||
# Update all locations with lat/lon based on points
|
||||
with transaction.atomic():
|
||||
for location in Location.objects.using(db_alias).all():
|
||||
if location.point:
|
||||
try:
|
||||
location.latitude = location.point.y
|
||||
location.longitude = location.point.x
|
||||
location.point = None
|
||||
location.save(update_fields=['latitude', 'longitude', 'point'])
|
||||
except (ValueError, TypeError, AttributeError):
|
||||
print(f"Warning: Could not convert point back to coordinates for location {location.id}")
|
||||
continue
|
||||
|
||||
class Migration(migrations.Migration):
|
||||
|
||||
dependencies = [
|
||||
('location', '0004_add_point_field'),
|
||||
]
|
||||
|
||||
operations = [
|
||||
migrations.RunPython(forwards_func, reverse_func, atomic=True),
|
||||
]
|
||||
174
location/migrations/0006_readd_historical_records.py
Normal file
174
location/migrations/0006_readd_historical_records.py
Normal file
@@ -0,0 +1,174 @@
|
||||
# Generated by Django 5.1.2 on 2024-11-04 22:32
|
||||
|
||||
import django.contrib.gis.db.models.fields
|
||||
import django.core.validators
|
||||
import django.db.models.deletion
|
||||
import simple_history.models
|
||||
from django.conf import settings
|
||||
from django.db import migrations, models
|
||||
|
||||
|
||||
class Migration(migrations.Migration):
|
||||
|
||||
dependencies = [
|
||||
("contenttypes", "0002_remove_content_type_name"),
|
||||
("location", "0005_convert_coordinates_to_points"),
|
||||
migrations.swappable_dependency(settings.AUTH_USER_MODEL),
|
||||
]
|
||||
|
||||
operations = [
|
||||
migrations.CreateModel(
|
||||
name="HistoricalLocation",
|
||||
fields=[
|
||||
(
|
||||
"id",
|
||||
models.BigIntegerField(
|
||||
auto_created=True, blank=True, db_index=True, verbose_name="ID"
|
||||
),
|
||||
),
|
||||
("object_id", models.PositiveIntegerField()),
|
||||
(
|
||||
"name",
|
||||
models.CharField(
|
||||
help_text="Name of the location (e.g. business name, landmark)",
|
||||
max_length=255,
|
||||
),
|
||||
),
|
||||
(
|
||||
"location_type",
|
||||
models.CharField(
|
||||
help_text="Type of location (e.g. business, landmark, address)",
|
||||
max_length=50,
|
||||
),
|
||||
),
|
||||
(
|
||||
"latitude",
|
||||
models.DecimalField(
|
||||
blank=True,
|
||||
decimal_places=6,
|
||||
help_text="Latitude coordinate (legacy field)",
|
||||
max_digits=9,
|
||||
null=True,
|
||||
validators=[
|
||||
django.core.validators.MinValueValidator(-90),
|
||||
django.core.validators.MaxValueValidator(90),
|
||||
],
|
||||
),
|
||||
),
|
||||
(
|
||||
"longitude",
|
||||
models.DecimalField(
|
||||
blank=True,
|
||||
decimal_places=6,
|
||||
help_text="Longitude coordinate (legacy field)",
|
||||
max_digits=9,
|
||||
null=True,
|
||||
validators=[
|
||||
django.core.validators.MinValueValidator(-180),
|
||||
django.core.validators.MaxValueValidator(180),
|
||||
],
|
||||
),
|
||||
),
|
||||
(
|
||||
"point",
|
||||
django.contrib.gis.db.models.fields.PointField(
|
||||
blank=True,
|
||||
help_text="Geographic coordinates as a Point",
|
||||
null=True,
|
||||
srid=4326,
|
||||
),
|
||||
),
|
||||
(
|
||||
"street_address",
|
||||
models.CharField(blank=True, max_length=255, null=True),
|
||||
),
|
||||
("city", models.CharField(blank=True, max_length=100, null=True)),
|
||||
(
|
||||
"state",
|
||||
models.CharField(
|
||||
blank=True,
|
||||
help_text="State/Region/Province",
|
||||
max_length=100,
|
||||
null=True,
|
||||
),
|
||||
),
|
||||
("country", models.CharField(blank=True, max_length=100, null=True)),
|
||||
("postal_code", models.CharField(blank=True, max_length=20, null=True)),
|
||||
("created_at", models.DateTimeField(blank=True, editable=False)),
|
||||
("updated_at", models.DateTimeField(blank=True, editable=False)),
|
||||
("history_id", models.AutoField(primary_key=True, serialize=False)),
|
||||
("history_date", models.DateTimeField(db_index=True)),
|
||||
("history_change_reason", models.CharField(max_length=100, null=True)),
|
||||
(
|
||||
"history_type",
|
||||
models.CharField(
|
||||
choices=[("+", "Created"), ("~", "Changed"), ("-", "Deleted")],
|
||||
max_length=1,
|
||||
),
|
||||
),
|
||||
],
|
||||
options={
|
||||
"verbose_name": "historical location",
|
||||
"verbose_name_plural": "historical locations",
|
||||
"ordering": ("-history_date", "-history_id"),
|
||||
"get_latest_by": ("history_date", "history_id"),
|
||||
},
|
||||
bases=(simple_history.models.HistoricalChanges, models.Model),
|
||||
),
|
||||
migrations.RemoveIndex(
|
||||
model_name="location",
|
||||
name="location_lo_latitud_7045c4_idx",
|
||||
),
|
||||
migrations.AlterField(
|
||||
model_name="location",
|
||||
name="latitude",
|
||||
field=models.DecimalField(
|
||||
blank=True,
|
||||
decimal_places=6,
|
||||
help_text="Latitude coordinate (legacy field)",
|
||||
max_digits=9,
|
||||
null=True,
|
||||
validators=[
|
||||
django.core.validators.MinValueValidator(-90),
|
||||
django.core.validators.MaxValueValidator(90),
|
||||
],
|
||||
),
|
||||
),
|
||||
migrations.AlterField(
|
||||
model_name="location",
|
||||
name="longitude",
|
||||
field=models.DecimalField(
|
||||
blank=True,
|
||||
decimal_places=6,
|
||||
help_text="Longitude coordinate (legacy field)",
|
||||
max_digits=9,
|
||||
null=True,
|
||||
validators=[
|
||||
django.core.validators.MinValueValidator(-180),
|
||||
django.core.validators.MaxValueValidator(180),
|
||||
],
|
||||
),
|
||||
),
|
||||
migrations.AddField(
|
||||
model_name="historicallocation",
|
||||
name="content_type",
|
||||
field=models.ForeignKey(
|
||||
blank=True,
|
||||
db_constraint=False,
|
||||
null=True,
|
||||
on_delete=django.db.models.deletion.DO_NOTHING,
|
||||
related_name="+",
|
||||
to="contenttypes.contenttype",
|
||||
),
|
||||
),
|
||||
migrations.AddField(
|
||||
model_name="historicallocation",
|
||||
name="history_user",
|
||||
field=models.ForeignKey(
|
||||
null=True,
|
||||
on_delete=django.db.models.deletion.SET_NULL,
|
||||
related_name="+",
|
||||
to=settings.AUTH_USER_MODEL,
|
||||
),
|
||||
),
|
||||
]
|
||||
@@ -1,8 +1,10 @@
|
||||
from django.contrib.gis.db import models as gis_models
|
||||
from django.db import models
|
||||
from django.contrib.contenttypes.fields import GenericForeignKey
|
||||
from django.contrib.contenttypes.models import ContentType
|
||||
from django.core.validators import MinValueValidator, MaxValueValidator
|
||||
from simple_history.models import HistoricalRecords
|
||||
from django.contrib.gis.geos import Point
|
||||
|
||||
class Location(models.Model):
|
||||
"""
|
||||
@@ -27,7 +29,7 @@ class Location(models.Model):
|
||||
MinValueValidator(-90),
|
||||
MaxValueValidator(90)
|
||||
],
|
||||
help_text="Latitude coordinate",
|
||||
help_text="Latitude coordinate (legacy field)",
|
||||
null=True,
|
||||
blank=True
|
||||
)
|
||||
@@ -38,12 +40,20 @@ class Location(models.Model):
|
||||
MinValueValidator(-180),
|
||||
MaxValueValidator(180)
|
||||
],
|
||||
help_text="Longitude coordinate",
|
||||
help_text="Longitude coordinate (legacy field)",
|
||||
null=True,
|
||||
blank=True
|
||||
)
|
||||
|
||||
# Address components - all made nullable
|
||||
# GeoDjango point field
|
||||
point = gis_models.PointField(
|
||||
srid=4326, # WGS84 coordinate system
|
||||
null=True,
|
||||
blank=True,
|
||||
help_text="Geographic coordinates as a Point"
|
||||
)
|
||||
|
||||
# Address components
|
||||
street_address = models.CharField(max_length=255, blank=True, null=True)
|
||||
city = models.CharField(max_length=100, blank=True, null=True)
|
||||
state = models.CharField(max_length=100, blank=True, null=True, help_text="State/Region/Province")
|
||||
@@ -58,7 +68,6 @@ class Location(models.Model):
|
||||
class Meta:
|
||||
indexes = [
|
||||
models.Index(fields=['content_type', 'object_id']),
|
||||
models.Index(fields=['latitude', 'longitude']),
|
||||
models.Index(fields=['city']),
|
||||
models.Index(fields=['country']),
|
||||
]
|
||||
@@ -73,6 +82,15 @@ class Location(models.Model):
|
||||
location_str = ", ".join(location_parts) if location_parts else "Unknown location"
|
||||
return f"{self.name} ({location_str})"
|
||||
|
||||
def save(self, *args, **kwargs):
|
||||
# Sync point field with lat/lon fields for backward compatibility
|
||||
if self.latitude is not None and self.longitude is not None and not self.point:
|
||||
self.point = Point(float(self.longitude), float(self.latitude))
|
||||
elif self.point and (self.latitude is None or self.longitude is None):
|
||||
self.longitude = self.point.x
|
||||
self.latitude = self.point.y
|
||||
super().save(*args, **kwargs)
|
||||
|
||||
def get_formatted_address(self):
|
||||
"""Returns a formatted address string"""
|
||||
components = []
|
||||
@@ -91,6 +109,29 @@ class Location(models.Model):
|
||||
@property
|
||||
def coordinates(self):
|
||||
"""Returns coordinates as a tuple"""
|
||||
if self.latitude is not None and self.longitude is not None:
|
||||
if self.point:
|
||||
return (self.point.y, self.point.x) # Returns (latitude, longitude)
|
||||
elif self.latitude is not None and self.longitude is not None:
|
||||
return (float(self.latitude), float(self.longitude))
|
||||
return None
|
||||
|
||||
def distance_to(self, other_location):
|
||||
"""
|
||||
Calculate the distance to another location in meters.
|
||||
Returns None if either location is missing coordinates.
|
||||
"""
|
||||
if not self.point or not other_location.point:
|
||||
return None
|
||||
return self.point.distance(other_location.point) * 100000 # Convert to meters
|
||||
|
||||
def nearby_locations(self, distance_km=10):
|
||||
"""
|
||||
Find locations within specified distance in kilometers.
|
||||
Returns a queryset of nearby Location objects.
|
||||
"""
|
||||
if not self.point:
|
||||
return Location.objects.none()
|
||||
|
||||
return Location.objects.filter(
|
||||
point__distance_lte=(self.point, distance_km * 1000) # Convert km to meters
|
||||
).exclude(pk=self.pk)
|
||||
|
||||
Reference in New Issue
Block a user