add simple survey support

survey/forms.py

@@ -0,0 +1,82 @@
+from django import forms
+from django.forms import ModelMultipleChoiceField
+from django.contrib.auth.models import User
+from generic.widgets import UserSearchWidget
+from .models import Survey, SurveyQuestion, SurveyResponse, SurveyAnswer
+
+class SurveyForm(forms.ModelForm):
+    author = ModelMultipleChoiceField(
+        queryset=User.objects.all(),
+        widget=UserSearchWidget(),
+        required=False,
+        help_text="Users who can view and manage this survey when it is set to private."
+    )
+
+    class Meta:
+        model = Survey
+        fields = ["name", "description", "active", "anonymous", "authors_only", "author"]
+
+class SurveyQuestionForm(forms.ModelForm):
+    class Meta:
+        model = SurveyQuestion
+        fields = ["text", "question_type", "choices", "required", "position"]
+        widgets = {
+            "choices": forms.Textarea(attrs={"rows": 4, "placeholder": "Enter each choice on a new line"}),
+        }
+
+class SurveyTakeForm(forms.Form):
+    def __init__(self, *args, **kwargs):
+        self.survey = kwargs.pop("survey")
+        self.questions = list(self.survey.questions.all())
+        super().__init__(*args, **kwargs)
+        
+        for q in self.questions:
+            field_name = f"question_{q.pk}"
+            if q.question_type == SurveyQuestion.QuestionType.TEXT:
+                self.fields[field_name] = forms.CharField(
+                    label=q.text,
+                    required=q.required,
+                    widget=forms.Textarea(attrs={"rows": 3, "class": "form-control"})
+                )
+            elif q.question_type == SurveyQuestion.QuestionType.CHOICE:
+                choices = [(c, c) for c in q.get_choices_list()]
+                self.fields[field_name] = forms.ChoiceField(
+                    label=q.text,
+                    required=q.required,
+                    choices=[("", "Select an option")] + choices,
+                    widget=forms.Select(attrs={"class": "form-select"})
+                )
+            elif q.question_type == SurveyQuestion.QuestionType.RATING:
+                self.fields[field_name] = forms.ChoiceField(
+                    label=q.text,
+                    required=q.required,
+                    choices=[(str(i), str(i)) for i in range(1, 6)],
+                    widget=forms.RadioSelect(attrs={"class": "form-check-input"}),
+                )
+
+    def save(self, user=None, cid=None, content_type=None, object_id=None, pre_or_post='GENERAL'):
+        response_user = None if self.survey.anonymous else user
+        response_cid = None if self.survey.anonymous else cid
+        
+        response = SurveyResponse.objects.create(
+            survey=self.survey,
+            user=response_user,
+            cid=response_cid,
+            content_type=content_type,
+            object_id=object_id,
+            pre_or_post=pre_or_post
+        )
+        
+        for q in self.questions:
+            field_name = f"question_{q.pk}"
+            val = self.cleaned_data.get(field_name)
+            if val is not None and val != "":
+                ans = SurveyAnswer(response=response, question=q)
+                if q.question_type == SurveyQuestion.QuestionType.TEXT:
+                    ans.text_answer = val
+                elif q.question_type == SurveyQuestion.QuestionType.CHOICE:
+                    ans.choice_answer = val
+                elif q.question_type == SurveyQuestion.QuestionType.RATING:
+                    ans.rating_answer = int(val)
+                ans.save()
+        return response