+
+
+ {% for tag in question.tags.all %}
+ {{ tag.tag_text }}
+ {% endfor %}
+
+
+
+
+
+{% endblock content %}
\ No newline at end of file
diff --git a/polls/tests/base.py b/polls/tests/base.py
index 1e2db85..d1e06d6 100644
--- a/polls/tests/base.py
+++ b/polls/tests/base.py
@@ -1,7 +1,6 @@
from django.utils import timezone
-from django.contrib.auth.models import User
-from ..models import Question, Vote, Choice
+from ..models import Question
def create_question(question_text, day=0):
@@ -11,4 +10,4 @@ def create_question(question_text, day=0):
in the past, positive for questions that have yet to be published).
"""
time = timezone.now() + timezone.timedelta(days=day)
- return Question.objects.create(question_text=question_text, pub_date=time)
\ No newline at end of file
+ return Question.objects.create(question_text=question_text, pub_date=time)
diff --git a/polls/tests/test_detail_views.py b/polls/tests/test_detail_views.py
index db2f3bc..07895ee 100644
--- a/polls/tests/test_detail_views.py
+++ b/polls/tests/test_detail_views.py
@@ -1,5 +1,4 @@
from django.test import TestCase, Client
-from django.utils import timezone
from django.urls import reverse
from django.contrib.auth.models import User
diff --git a/polls/tests/test_question_model.py b/polls/tests/test_question_model.py
index f0cdd07..750dbab 100644
--- a/polls/tests/test_question_model.py
+++ b/polls/tests/test_question_model.py
@@ -102,4 +102,4 @@ class QuestionModelTests(TestCase):
pub_date = timezone.now() - datetime.timedelta(hours=1)
end_date = timezone.now() + datetime.timedelta(hours=2)
question = Question(pub_date=pub_date, end_date=end_date)
- self.assertIs(question.can_vote(), True)
\ No newline at end of file
+ self.assertIs(question.can_vote(), True)
diff --git a/polls/tests/test_search_poll.py b/polls/tests/test_search_poll.py
new file mode 100644
index 0000000..a8709a8
--- /dev/null
+++ b/polls/tests/test_search_poll.py
@@ -0,0 +1,23 @@
+from django.test import TestCase
+from django.urls import reverse
+
+from ..models import Question
+
+
+class SearchPollTest(TestCase):
+ """Test if user search with normal string. It must return same queryset as filter question objects"""
+ def test_search_normal_poll(self):
+ data_1 = {'q': 'what'}
+ data_2 = {'q': 'prefer'}
+ q_1 = 'what'
+ q_2 = 'prefer'
+ response_1 = self.client.get(reverse("polls:search_poll"), data_1)
+ response_2 = self.client.get(reverse("polls:search_poll"), data_2)
+ self.assertQuerysetEqual(response_1.context['results'], Question.objects.filter(question_text__icontains=q_1))
+ self.assertQuerysetEqual(response_2.context['results'], Question.objects.filter(question_text__icontains=q_2))
+
+ def test_search_with_empty(self):
+ """Test if user search with empty string. It must return all question"""
+ data = {'q': ''}
+ response = self.client.get(reverse("polls:search_poll"), data)
+ self.assertQuerysetEqual(response.context['results'], Question.objects.all())
diff --git a/polls/tests/test_sentiment_model.py b/polls/tests/test_sentiment_model.py
index c1ede3b..97bae5f 100644
--- a/polls/tests/test_sentiment_model.py
+++ b/polls/tests/test_sentiment_model.py
@@ -2,7 +2,7 @@ from django.test import TransactionTestCase, Client
from django.contrib.auth.models import User
from .base import create_question
-from ..views import up_down_vote
+
# ! https://stackoverflow.com/questions/24588520/testing-several-integrityerrors-in-the-same-django-unittest-test-case
# * https://stackoverflow.com/questions/44450533/difference-between-testcase-and-transactiontestcase-classes-in-django-test
@@ -32,11 +32,11 @@ class UpDownVoteViewTest(TransactionTestCase):
self.assertEqual(count_up, 0)
self.assertEqual(count_down, 1)
- def test_can_change_up_to_down(self):
+ def test_can_change_down_to_up(self):
self.client.login(username="test_user", password="12345abc")
self.q1.downvote(self.user)
self.q1.upvote(self.user)
count_up = self.q1.sentimentvote_set.filter(vote_types=True).count()
count_down = self.q1.sentimentvote_set.filter(vote_types=False).count()
self.assertEqual(count_up, 1)
- self.assertEqual(count_down, 0)
\ No newline at end of file
+ self.assertEqual(count_down, 0)
diff --git a/polls/tests/test_signup.py b/polls/tests/test_signup.py
index 50278ee..c9bd074 100644
--- a/polls/tests/test_signup.py
+++ b/polls/tests/test_signup.py
@@ -42,4 +42,4 @@ class SignUpTestCase(TestCase):
'password2': 'testpassword123',
}
response = self.client.post(signup_url, data)
- self.assertRedirects(response, reverse("polls:index"))
\ No newline at end of file
+ self.assertRedirects(response, reverse("polls:index"))
diff --git a/polls/tests/test_vote_views.py b/polls/tests/test_vote_views.py
index 393650e..af8b3c3 100644
--- a/polls/tests/test_vote_views.py
+++ b/polls/tests/test_vote_views.py
@@ -5,6 +5,7 @@ from django.contrib.auth.models import User
from .base import create_question
from ..models import Vote, Choice
+
class VoteViewTest(TestCase):
@classmethod
def setUpTestData(cls):
@@ -20,9 +21,9 @@ class VoteViewTest(TestCase):
"""
self.client.login(username=self.user.username, password="aaa123321aaa")
- response = self.client.post(reverse("polls:vote", args=(self.question.id,)),
- {'choice' : self.choice1.id})
-
+ response = self.client.post(reverse("polls:vote", args=(self.question.id,)),
+ {'choice': self.choice1.id})
+
self.assertRedirects(response, reverse("polls:results", args=(self.question.id,)))
self.assertTrue(Vote.objects.filter(user=self.user, question=self.question).exists())
@@ -32,18 +33,18 @@ class VoteViewTest(TestCase):
"""
self.client.login(username=self.user.username, password="aaa123321aaa")
- response = self.client.post(reverse("polls:vote", args=(self.question.id,)),
- {'choice' : 1000})
-
+ response = self.client.post(reverse("polls:vote", args=(self.question.id,)),
+ {'choice': 1000})
+
self.assertRedirects(response, reverse('polls:detail', args=(self.question.id,)))
def test_vote_without_login(self):
"""
Test the vote view when the user is not logged in.
"""
- response = self.client.post(reverse("polls:vote", args=(self.question.id,)),
- {'choice' : self.choice1})
-
+ response = self.client.post(reverse("polls:vote", args=(self.question.id,)),
+ {'choice': self.choice1})
+
self.assertRedirects(response, "/accounts/login/?next=/polls/1/vote/")
def test_vote_voting_not_allowed(self):
@@ -55,9 +56,8 @@ class VoteViewTest(TestCase):
self.question_2 = create_question(question_text="Test not allow", day=10)
self.choice_2 = Choice.objects.create(question=self.question_2, choice_text="Test Choice 2_2")
- response = self.client.post(reverse("polls:vote", args=(self.question_2.id,)),
- {"choice" : self.choice_2.id})
-
+ response = self.client.post(reverse("polls:vote", args=(self.question_2.id,)), {"choice": self.choice_2.id})
+
self.assertRedirects(response, reverse('polls:index'))
def test_vote_with_no_post_data(self):
@@ -81,6 +81,6 @@ class VoteViewTest(TestCase):
response_2 = self.client.post(reverse("polls:vote", args=(self.question.id,)), {"choice": self.choice2.id})
self.assertRedirects(response_2, reverse('polls:results', args=(self.question.id,)))
-
+
self.assertFalse(Vote.objects.filter(user=self.user, question=self.question, choice=self.choice1).exists())
- self.assertTrue(Vote.objects.filter(user=self.user, question=self.question, choice=self.choice2).exists())
\ No newline at end of file
+ self.assertTrue(Vote.objects.filter(user=self.user, question=self.question, choice=self.choice2).exists())
diff --git a/polls/urls.py b/polls/urls.py
index 7b0b4fa..8737085 100644
--- a/polls/urls.py
+++ b/polls/urls.py
@@ -9,6 +9,8 @@ urlpatterns = [
path("
/results/", views.ResultsView.as_view(), name="results"),
path("/vote/", views.vote, name="vote"),
path("signup/", views.SignUpView.as_view(), name="signup"),
- path("upvote/", views.up_down_vote, {'vote_type' : 'upvote'}, name="upvote"),
- path("downvote/", views.up_down_vote, {'vote_type' : 'downvote'}, name="downvote"),
+ path("upvote/", views.up_down_vote, {'vote_type': 'upvote'}, name="upvote"),
+ path("downvote/", views.up_down_vote, {'vote_type': 'downvote'}, name="downvote"),
+ path("search", views.search_poll, name="search_poll"),
+ path("create", views.create_poll, name="create_poll")
]
diff --git a/polls/views.py b/polls/views.py
index e58f1f1..43028ae 100644
--- a/polls/views.py
+++ b/polls/views.py
@@ -2,7 +2,6 @@ import logging
from typing import Any
from django.shortcuts import get_object_or_404, render, redirect
-from django.urls import reverse
from django.views import generic
from django.utils import timezone
from django.urls import reverse_lazy, reverse
@@ -12,7 +11,7 @@ from django.contrib.auth import authenticate, login
from django.contrib.auth.decorators import login_required
from django.db.models import Q
-from .forms import SignUpForm
+from .forms import SignUpForm, PollSearchForm, PollCreateForm
from .models import Choice, Question, Vote
@@ -26,21 +25,21 @@ class IndexView(generic.ListView):
context_object_name = "latest_question_list"
def get_queryset(self):
- """
- Return the last published questions that is published and haven't ended yet.
- """
- now = timezone.now()
- all_poll_queryset = Question.objects.filter(
- Q(pub_date__lte=now) & ((Q(end_date__gte=now) | Q(end_date=None)))
- ).order_by("-pub_date")
+ """
+ Return the last published questions that is published and haven't ended yet.
+ """
+ now = timezone.now()
+ all_poll_queryset = Question.objects.filter(
+ Q(pub_date__lte=now) & ((Q(end_date__gte=now) | Q(end_date=None)))
+ ).order_by("-pub_date")
- trend_poll_queryset = Question.objects.filter(
- Q(pub_date__lte=now) & ((Q(end_date__gte=now) | Q(end_date=None))) & Q(trend_score__gte=100)
- ).order_by("trend_score", "end_date")[:3]
+ trend_poll_queryset = Question.objects.filter(
+ Q(pub_date__lte=now) & ((Q(end_date__gte=now) | Q(end_date=None))) & Q(trend_score__gte=100)
+ ).order_by("trend_score")[:3]
- queryset = {'all_poll' : all_poll_queryset,
- 'trend_poll' : trend_poll_queryset,}
- return queryset
+ queryset = {'all_poll': all_poll_queryset,
+ 'trend_poll': trend_poll_queryset, }
+ return queryset
class DetailView(LoginRequiredMixin, generic.DetailView):
@@ -93,12 +92,16 @@ class DetailView(LoginRequiredMixin, generic.DetailView):
class ResultsView(LoginRequiredMixin, generic.DetailView):
+ """
+ Provide a view for Result page, a Result for the poll contain poll participants
+ number and other statistic such as up, down vote
+ """
model = Question
template_name = "polls/results.html"
def get_context_data(self, **kwargs: Any) -> dict[str, Any]:
context = super().get_context_data(**kwargs)
-
+
user_voted = None
question = self.get_object()
if question.sentimentvote_set.filter(user=self.request.user, question=question, vote_types=True).exists():
@@ -109,7 +112,11 @@ class ResultsView(LoginRequiredMixin, generic.DetailView):
context['user_voted'] = user_voted
return context
+
class SignUpView(generic.CreateView):
+ """
+ View that responsible for Sign Up page.
+ """
form_class = SignUpForm
success_url = reverse_lazy('polls:index')
template_name = 'registration/signup.html'
@@ -146,7 +153,7 @@ def vote(request, question_id):
vote, created = Vote.objects.update_or_create(
user=request.user,
question=question,
- defaults={'choice' : selected_choice}
+ defaults={'choice': selected_choice}
)
if created:
@@ -163,13 +170,15 @@ def vote(request, question_id):
else:
messages.error(request, "Invalid request method.")
return redirect("polls:index")
-
+
@login_required
def up_down_vote(request, question_id, vote_type):
- ip = get_client_ip(request)
+ """
+ A function that control the upvote and downvote request.
+ """
question = get_object_or_404(Question, pk=question_id)
-
+
if request.method == "POST":
if vote_type == "upvote":
if question.upvote(request.user):
@@ -177,15 +186,76 @@ def up_down_vote(request, question_id, vote_type):
elif vote_type == "downvote":
if question.downvote(request.user):
messages.success(request, "You downvoted this Pollπ")
-
+
return redirect(reverse("polls:results", args=(question_id,)))
# https://stackoverflow.com/questions/4581789/how-do-i-get-user-ip-address-in-django
def get_client_ip(request):
+ """
+ Use with logger to get ip of user.
+ """
x_forwarded_for = request.META.get('HTTP_X_FORWARDED_FOR')
if x_forwarded_for:
ip = x_forwarded_for.split(',')[0]
else:
ip = request.META.get('REMOTE_ADDR')
return ip
+
+
+def search_poll(request):
+ """
+ A function that handle the rendering of search result after user search with
+ search bar.
+ """
+ form = PollSearchForm
+
+ results = []
+ q = ''
+ if 'q' in request.GET:
+ form = PollSearchForm(request.GET)
+ if form.is_valid():
+ q = form.cleaned_data['q']
+ # Case insensitive (icontains)
+ results = Question.objects.filter(question_text__icontains=q)
+ # * If user search with empty string then show every poll.
+ if q == '':
+ results = Question.objects.all()
+ return render(request, 'polls/search.html', {'form': form, 'results': results, 'q': q})
+
+
+@login_required
+def create_poll(request):
+ ip = get_client_ip(request)
+ if request.method == 'POST':
+ form = PollCreateForm(request.POST)
+ if form.is_valid():
+ question_text = form.cleaned_data['question_text']
+ pub_date = form.cleaned_data['pub_date']
+ end_date = form.cleaned_data['end_date']
+ short_description = form.cleaned_data['short_description']
+ long_description = form.cleaned_data.get('long_description', '')
+ user_choices = form.cleaned_data['user_choice']
+ tags = form.cleaned_data['tags']
+
+ question = Question.objects.create(
+ question_text=question_text,
+ pub_date=pub_date,
+ end_date=end_date,
+ short_description=short_description,
+ long_description=long_description,
+ )
+
+ choices = user_choices.split(',') # Split with comma
+ for choice_text in choices:
+ Choice.objects.create(question=question, choice_text=choice_text.strip())
+
+ # Add tags to the question
+ question.tags.set(tags)
+ logger.info(f"User {request.user.username} ({ip}) create poll : {question_text}")
+ return redirect('polls:index')
+
+ else:
+ form = PollCreateForm()
+
+ return render(request, 'polls/creation.html', {'form': form})
diff --git a/setup.ps1 b/setup.ps1
index 191362c..013cc64 100644
--- a/setup.ps1
+++ b/setup.ps1
@@ -62,4 +62,5 @@ Write-Host $bottomBorder
python manage.py migrate
python manage.py loaddata data/users.json
python manage.py loaddata data/polls.json
+python manage.py loaddata data/vote.json
python manage.py runserver --insecure
\ No newline at end of file
diff --git a/setup.py b/setup.py
index 9866a7a..7b871ff 100644
--- a/setup.py
+++ b/setup.py
@@ -17,7 +17,7 @@ def check_python_command():
return command
except FileNotFoundError:
continue
-
+
return None
def create_virtual_environment(env_name, python_command):
@@ -31,7 +31,7 @@ def customize_virtual_environment():
def setup_environment_variables(python_command_in_venv):
print("Setting up Django environment variables:")
-
+
# SECRET KEY
generate_secret_key = input("Generate a Django SECRET_KEY? (yes/no): ").strip().lower()
if generate_secret_key == "yes":
@@ -39,7 +39,7 @@ def setup_environment_variables(python_command_in_venv):
'from django.core.management.utils import get_random_secret_key; print(get_random_secret_key())']).decode().strip()
else:
secret_key = input("Enter Django SECRET_KEY: ").strip()
-
+
# DEBUG MODE
while True:
debug_mode = input("Enable DEBUG mode? (True/False): ").strip()
@@ -52,7 +52,7 @@ def setup_environment_variables(python_command_in_venv):
allowed_hosts = input("Enter ALLOWED_HOSTS (comma-separated, or press Enter for default): ").strip()
if not allowed_hosts:
allowed_hosts = "*.ku.th,localhost,127.0.0.1,::1"
-
+
# TZ
available_time_zones = ["Asia/Bangkok", "Japan", "UCT", "CST6CDT", "Custom"]
@@ -73,7 +73,7 @@ def setup_environment_variables(python_command_in_venv):
print("Invalid choice. Please enter a valid number.")
except ValueError:
print("Invalid input. Please enter a valid number.")
-
+
email_host_password = input("Enter EMAIL_HOST_PASSWORD: ").strip()
# SET
@@ -116,7 +116,7 @@ def main():
elif is_windows:
activate_command = os.path.join(".venv", "Scripts", "activate")
subprocess.run([activate_command], shell=True)
-
+
python_command = os.path.join(".venv", "bin", "python") if is_posix else os.path.join(".venv", "Scripts", "python")
else:
print("Not setting up a virtual environment. Using the global Python interpreter.")
@@ -136,6 +136,7 @@ def main():
subprocess.run([python_command, "manage.py", "migrate"])
subprocess.run([python_command, "manage.py", "loaddata", "data/users.json"])
subprocess.run([python_command, "manage.py", "loaddata", "data/polls.json"])
+ subprocess.run([python_command, "manage.py", "loaddata", "data/vote.json"])
start_server = input("Do you want to start the Django server? (yes/no): ").lower()
if start_server == "yes":
@@ -154,19 +155,20 @@ def main():
print(f"==========================Install Requirement==========================")
subprocess.run([python_command_in_venv, "-m", "pip", "install", "-r", "requirements.txt"])
setup_environment_variables(python_command_in_venv)
-
+
subprocess.run([python_command_in_venv, "-m", "pip", "install", "-r", "requirements.txt"], check=True)
subprocess.run([python_command_in_venv, "manage.py", "migrate"], check=True)
subprocess.run([python_command_in_venv, "manage.py", "loaddata", "data/users.json"], check=True)
subprocess.run([python_command_in_venv, "manage.py", "loaddata", "data/polls.json"], check=True)
-
+ subprocess.run([python_command_in_venv, "manage.py", "loaddata", "data/vote.json"], check=True)
+
start_server = input("Do you want to start the Django server? (yes/no): ").strip().lower()
if start_server == "yes":
print("=================================================")
print("Django run in --insecure mode to load Static File")
print("==================================================")
subprocess.run([python_command_in_venv, "manage.py", "runserver", "--insecure"], check=True)
-
+
except subprocess.CalledProcessError as e:
print(f"Error: {e}")
sys.exit(1)
@@ -174,5 +176,6 @@ def main():
print("\nSetup process aborted.")
sys.exit(1)
+
if __name__ == "__main__":
main()