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	21	fields = ['name','link_url','link_description']
22	22
23	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	33	class Meta:
25	34	model = Link
26	35	fields = ['name','link_url','link_description']
...	...