Fix and refactor statistics app

1 files changed, 48 insertions, 0 deletions

diff --git a/statistics_app/forms.py b/statistics_app/forms.py
new file mode 100644
index 0000000..cd04cd3
--- /dev/null
+++ b/statistics_app/forms.py
@@ -0,0 +1,48 @@
+from django import forms
+from django.utils import timezone
+
+from workshop_app.models import states, WorkshopType
+
+
+class FilterForm(forms.Form):
+    from_date = forms.DateField(
+        widget=forms.DateInput(
+            attrs={'type': 'date', 'class': 'form-control'}
+        )
+    )
+    to_date = forms.DateField(
+        widget=forms.DateInput(
+            attrs={'type': 'date', 'class': 'form-control'}
+        )
+    )
+    workshop_type = forms.ModelChoiceField(
+        queryset=WorkshopType.objects.all(), required=False,
+        widget=forms.Select(attrs={'class': 'form-control'})
+    )
+    state = forms.ChoiceField(
+        choices=states, required=False,
+        widget=forms.Select(attrs={'class': 'form-control'})
+    )
+    show_workshops = forms.BooleanField(
+        help_text="Show my workshops only", required=False,
+    )
+    sort = forms.ChoiceField(
+        choices=(("date", "Oldest"), ("-date", "Latest")),
+        help_text="Sort by",
+        widget=forms.Select(attrs={'class': 'form-control'})
+    )
+    def __init__(self, *args, **kwargs):
+        start = kwargs.pop("start") if "start" in kwargs else None
+        end = kwargs.pop("end") if "end" in kwargs else None
+        selected_state = kwargs.pop("state") if "state" in kwargs else None
+        selected_type = kwargs.pop("type") if "type" in kwargs else None
+        show_workshops = (kwargs.pop("show_workshops")
+                            if "show_workshops" in kwargs else None)
+        sort = kwargs.pop("sort") if "sort" in kwargs else None
+        super(FilterForm, self).__init__(*args, **kwargs)
+        self.fields["from_date"].initial = start
+        self.fields["to_date"].initial = end
+        self.fields["state"].initial = selected_state
+        self.fields["workshop_type"].initial = selected_type
+        self.fields["show_workshops"].initial = show_workshops
+        self.fields["sort"].initial = sort