From b914d2f4bbfb11ea74ecb675bafd341ef2c3f4cc Mon Sep 17 00:00:00 2001 From: orbitroom Date: Wed, 23 Mar 2016 17:05:38 -0400 Subject: [PATCH] Added Form Validation --- srp/form.py | 16 ++++++++++++++-- 1 file changed, 14 insertions(+), 2 deletions(-) diff --git a/srp/form.py b/srp/form.py index 52463abe..400fbfb2 100755 --- a/srp/form.py +++ b/srp/form.py @@ -1,6 +1,7 @@ from django import forms + class SrpFleetMainForm(forms.Form): fleet_name = forms.CharField(required=True, label="Fleet Name") fleet_time = forms.DateTimeField(required=True, label="Fleet Time") @@ -8,8 +9,19 @@ class SrpFleetMainForm(forms.Form): class SrpFleetUserRequestForm(forms.Form): - killboard_link = forms.CharField(required=True, label="ZKillboard Link") additional_info = forms.CharField(required=False, label="Additional Info") + killboard_link = forms.CharField( + label="Zkillboard Link", + max_length=255, + required=True + + ) + + def clean_killboard_link(self): + data = self.cleaned_data['killboard_link'] + if "zkillboard.com" not in data: + raise forms.ValidationError("Invalid Link. Please use Zkillboard.com") + return data class SrpFleetUpdateCostForm(forms.Form): @@ -17,4 +29,4 @@ class SrpFleetUpdateCostForm(forms.Form): class SrpFleetMainUpdateForm(forms.Form): - fleet_aar_link = forms.CharField(required=True, label="After Action Report Link") + fleet_aar_link = forms.CharField(required=True, label="After Action Report Link") \ No newline at end of file