Added forms for creating and editing a task

1 files changed, 53 insertions, 0 deletions

diff --git a/pytask/taskapp/forms.py b/pytask/taskapp/forms.py
new file mode 100644
index 0000000..3804d2a
--- /dev/null
+++ b/pytask/taskapp/forms.py
@@ -0,0 +1,53 @@
+from django import forms
+from pytask.taskapp.models import Task, WorkReport
+
+class TaskCreateForm(forms.ModelForm):
+    class Meta:
+        model = Task
+        fields = ['title', 'desc', 'tags_field', 'pynts']
+
+    def clean_title(self):
+
+        data = self.cleaned_data['title'].strip()
+
+        if not data:
+            raise forms.ValidationError("This field is required")
+
+        try:
+            Task.objects.exclude(status="DL").get(title__iexact=data)
+            raise forms.ValidationError("Another task with same title exists")
+        except Task.DoesNotExist:
+            return data
+
+    def clean_desc(self):
+
+        data = self.cleaned_data['desc'].strip()
+
+        if not data:
+            raise forms.ValidationError("This field is required")
+
+        return data
+
+class EditTaskForm(forms.ModelForm):
+    class Meta:
+        model = Task
+        fields = ['title', 'desc', 'tags_field', 'pynts']
+
+    def clean_desc(self):
+        data = self.cleaned_data['desc'].strip()
+        if not data:
+            raise forms.ValidationError("Enter some description for the task")
+
+        return data
+
+    def clean_title(self):
+        data = self.cleaned_data['title'].strip()
+        try:
+            prev_task = Task.objects.exclude(status="DL").get(title__iexact=data)
+            if prev_task.id != self.instance.id:
+                raise forms.ValidationError("Another task with same title exists")
+            else:
+                return data
+        except Task.DoesNotExist:
+            return data
+