56 lines
1.5 KiB
Python
56 lines
1.5 KiB
Python
from django import forms
|
|
from django.contrib.auth.forms import UserCreationForm
|
|
|
|
from map.models import FriendLocation, Friend
|
|
|
|
|
|
class LocationForm(forms.ModelForm):
|
|
class Meta:
|
|
model = FriendLocation
|
|
fields = [
|
|
'latitude',
|
|
'longitude',
|
|
'start_date',
|
|
'end_date',
|
|
'friend',
|
|
]
|
|
widgets = {'friend': forms.HiddenInput()}
|
|
|
|
def __init__(self, request, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.request = request
|
|
|
|
def clean_friend(self):
|
|
return self.request.user
|
|
|
|
|
|
class LocationSharingForm(forms.ModelForm):
|
|
class Meta:
|
|
model = Friend
|
|
fields = [
|
|
'shares_location_to',
|
|
]
|
|
|
|
def __init__(self, request, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.request = request
|
|
self.fields['shares_location_to'].queryset = Friend.objects.exclude(pk=request.user.pk)
|
|
self.fields['shares_location_to'].label = 'Share location to'
|
|
self.fields['shares_location_to'].widget.attrs = {
|
|
'data-size': 20,
|
|
'class': 'selectpicker',
|
|
'data-live-search': 'true',
|
|
'data-actions-box': 'true',
|
|
}
|
|
|
|
|
|
class FriendCreationForm(UserCreationForm):
|
|
class Meta(UserCreationForm.Meta):
|
|
model = Friend
|
|
fields = UserCreationForm.Meta.fields + (
|
|
'first_name',
|
|
'last_name',
|
|
'email',
|
|
)
|
|
|