mirror of
https://github.com/pinry/pinry.git
synced 2025-11-14 00:55:43 +01:00
e2a38f8a10
There has been some refactoring going on in the pinry.pins.models module. The upload_to code has been refactored into its own function, images have been moved to their own models - otherwise the number of fields in the Pin model would skyrocket. Also ModelManagers have been written to move image fetching and generating outside of models.
58 lines
2.0 KiB
Python
58 lines
2.0 KiB
Python
from django import forms
|
|
|
|
from taggit.forms import TagField
|
|
|
|
from .models import Pin
|
|
|
|
|
|
class PinForm(forms.Form):
|
|
url = forms.CharField(label='URL', required=False)
|
|
image = forms.ImageField(label='or Upload', required=False)
|
|
description = forms.CharField(label='Description', required=False, widget=forms.Textarea)
|
|
tags = TagField()
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super(forms.Form, self).__init__(*args, **kwargs)
|
|
|
|
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
|