Link validation included on updatelink form [Issue:#122]

Gustavo Bernardo
1 parent 93297df7
Showing 1 changed file with 9 additions and 0 deletions Show diff stats
links/forms.py
@@ -21,6 +21,15 @@ class CreateLinkForm(forms.ModelForm):
         fields = ['name','link_url','link_description']
 class UpdateLinkForm(forms.ModelForm):
+    def clean_link_url(self):
+        link_url = self.cleaned_data['link_url']
+        try:
+            response = requests.head(link_url)
+            if response.status_code >= 400:
+                raise forms.ValidationError(_('Invalid url!'))
+        except requests.ConnectionError:
+            raise forms.ValidationError(_('Invalid url!'))
+        return link_url
     class Meta:
         model = Link
         fields = ['name','link_url','link_description']
	@@ -21,6 +21,15 @@ class CreateLinkForm(forms.ModelForm):		@@ -21,6 +21,15 @@ class CreateLinkForm(forms.ModelForm):
21	fields = ['name','link_url','link_description']	21	fields = ['name','link_url','link_description']
22		22
23	class UpdateLinkForm(forms.ModelForm):	23	class UpdateLinkForm(forms.ModelForm):
		24	+ def clean_link_url(self):
		25	+ link_url = self.cleaned_data['link_url']
		26	+ try:
		27	+ response = requests.head(link_url)
		28	+ if response.status_code >= 400:
		29	+ raise forms.ValidationError(_('Invalid url!'))
		30	+ except requests.ConnectionError:
		31	+ raise forms.ValidationError(_('Invalid url!'))
		32	+ return link_url
24	class Meta:	33	class Meta:
25	model = Link	34	model = Link
26	fields = ['name','link_url','link_description']	35	fields = ['name','link_url','link_description']