Added extra protocol checking.

pinry/pins/forms.py

@@ -8,15 +8,31 @@ class PinForm(forms.ModelForm):
 
     def clean_url(self):
         data = self.cleaned_data['url']
+
+        # 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.")
+
+        # Check if pin already exists
         try:
             Pin.objects.get(url=data)
             raise forms.ValidationError("URL has already been pinned!")
         except Pin.DoesNotExist:
-            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.")
-            return data
+            protocol = data.split(':')[0]
+            if protocol == 'http':
+                opp_data = data.replace('http://', 'https://')
+            elif protocol == 'https':
+                opp_data = data.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_data)
+                raise forms.ValidationError("URL has already been pinned!")
+            except Pin.DoesNotExist:
+                return data
 
     class Meta:
         model = Pin