mirror of
https://github.com/pinry/pinry.git
synced 2025-11-14 09:05:41 +01:00
65 lines
2.1 KiB
Python
65 lines
2.1 KiB
Python
from django import forms
|
|
|
|
from .models import Pin
|
|
|
|
|
|
class PinForm(forms.ModelForm):
|
|
url = forms.CharField(label='URL', required=False)
|
|
image = forms.ImageField(label='or Upload', required=False)
|
|
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super(forms.ModelForm, self).__init__(*args, **kwargs)
|
|
self.fields.keyOrder = (
|
|
'url',
|
|
'image',
|
|
'description',
|
|
)
|
|
|
|
|
|
def check_if_image(self, data):
|
|
# Test file type
|
|
image_file_types = ['png', 'gif', 'jpeg', 'jpg']
|
|
file_type = data.split('.')[-1]
|
|
if file_type.lower() not in image_file_types:
|
|
raise forms.ValidationError("Requested URL is not an image file. "
|
|
"Only images are currently supported.")
|
|
|
|
def clean(self):
|
|
cleaned_data = super(PinForm, self).clean()
|
|
|
|
url = cleaned_data.get('url')
|
|
image = cleaned_data.get('image')
|
|
|
|
if url:
|
|
self.check_if_image(url)
|
|
try:
|
|
Pin.objects.get(url=url)
|
|
raise forms.ValidationError("URL has already been pinned!")
|
|
except Pin.DoesNotExist:
|
|
protocol = url.split(':')[0]
|
|
if protocol == 'http':
|
|
opp_url = url.replace('http://', 'https://')
|
|
elif protocol == 'https':
|
|
opp_url = url.replace('https://', 'http://')
|
|
else:
|
|
raise forms.ValidationError("Currently only support HTTP and "
|
|
"HTTPS protocols, please be sure "
|
|
"you include this in the URL.")
|
|
|
|
try:
|
|
Pin.objects.get(url=opp_url)
|
|
raise forms.ValidationError("URL has already been pinned!")
|
|
except Pin.DoesNotExist:
|
|
pass
|
|
elif image:
|
|
pass
|
|
else:
|
|
raise forms.ValidationError("Need either a URL or Upload.")
|
|
|
|
return cleaned_data
|
|
|
|
class Meta:
|
|
model = Pin
|
|
exclude = ['submitter']
|