mirror of
https://github.com/pacnpal/thrillwiki_django_no_react.git
synced 2025-12-20 10:51:09 -05:00
fix location add
This commit is contained in:
@@ -1,6 +1,7 @@
|
||||
from django import forms
|
||||
from decimal import Decimal, InvalidOperation, ROUND_DOWN
|
||||
from .models import Park
|
||||
from location.models import Location
|
||||
|
||||
|
||||
class ParkForm(forms.ModelForm):
|
||||
@@ -207,8 +208,14 @@ class ParkForm(forms.ModelForm):
|
||||
'postal_code': self.cleaned_data.get('postal_code'),
|
||||
}
|
||||
|
||||
# Set location data to be saved with the park
|
||||
park.set_location(**location_data)
|
||||
# Handle location: update if exists, create if not
|
||||
if park.location.exists():
|
||||
location = park.location.first()
|
||||
for key, value in location_data.items():
|
||||
setattr(location, key, value)
|
||||
location.save()
|
||||
else:
|
||||
Location.objects.create(content_object=park, **location_data)
|
||||
|
||||
if commit:
|
||||
park.save()
|
||||
|
||||
Reference in New Issue
Block a user