286 lines
8.7 KiB
Python
286 lines
8.7 KiB
Python
from django.shortcuts import render, redirect
|
|
from django.contrib.auth.decorators import login_required, permission_required
|
|
from django.db.models import CharField, Value
|
|
from authentication.models import User
|
|
from reviews.models import Ticket, Review, UserFollows
|
|
from reviews.forms import TicketForm, ReviewForm, ToFollowForm
|
|
from itertools import chain
|
|
|
|
|
|
@login_required
|
|
def flux(request):
|
|
"""
|
|
Display the user's home page
|
|
Get tickets & reviews of followed users, sort latest first
|
|
:param: request
|
|
:return: queryset of sorted tickets and reviews
|
|
"""
|
|
followed = UserFollows.objects.filter(user=request.user)
|
|
users_followed = []
|
|
for userf in followed:
|
|
users_followed.append(userf.followed_user)
|
|
tickets = Ticket.objects.filter(user__in=users_followed)
|
|
tickets = tickets.annotate(content_type=Value('TICKET', CharField()))
|
|
reviews = Review.objects.filter(user__in=users_followed)
|
|
reviews = reviews.annotate(content_type=Value('REVIEW', CharField()))
|
|
posts = sorted(
|
|
chain(reviews, tickets),
|
|
key=lambda post: post.time_created,
|
|
reverse=True)
|
|
return render(request,
|
|
'reviews/flux.html',
|
|
{'posts': posts})
|
|
|
|
|
|
@login_required
|
|
def posts(request):
|
|
"""
|
|
Get list of tickets and reviews of current user
|
|
:param: request
|
|
:return: queryset of tickets and reviews
|
|
"""
|
|
|
|
tickets = Ticket.objects.filter(user=request.user)
|
|
reviews = Review.objects.filter(user=request.user)
|
|
return render(request,
|
|
'reviews/posts.html',
|
|
{'tickets': tickets, 'reviews': reviews})
|
|
|
|
|
|
@login_required
|
|
def ticket(request, ticket_id):
|
|
"""
|
|
Get detail of a ticket
|
|
:param: request, Ticket id
|
|
:return: object Ticket
|
|
"""
|
|
ticket = Ticket.objects.get(id=ticket_id)
|
|
return render(request,
|
|
'reviews/ticket.html',
|
|
{'ticket': ticket})
|
|
|
|
|
|
@login_required
|
|
def create_ticket(request):
|
|
"""
|
|
Send form, retrieve user's data, create and write object Ticket
|
|
:param: request (POST)
|
|
:return: form, write Ticket in DB
|
|
"""
|
|
tickets = Ticket.objects.all()
|
|
ticket_form = TicketForm()
|
|
if request.method == 'POST':
|
|
ticket_form = TicketForm(request.POST, request.FILES)
|
|
if ticket_form.is_valid():
|
|
ticket = ticket_form.save(commit=False)
|
|
ticket.user = request.user
|
|
ticket.save()
|
|
return redirect('flux')
|
|
return render(request,
|
|
'reviews/ticket_create.html',
|
|
context={'ticket_form': ticket_form, 'tickets': tickets})
|
|
|
|
|
|
@login_required
|
|
@permission_required('review.change_ticket', raise_exception=True)
|
|
def update_ticket(request, ticket_id):
|
|
"""
|
|
Get new data for a given Ticket id and save changes
|
|
:param: request, Ticket id
|
|
:return: form, write Ticket in DB
|
|
"""
|
|
ticket = Ticket.objects.get(id=ticket_id)
|
|
if request.method == 'POST':
|
|
ticket_form = TicketForm(request.POST, instance=ticket)
|
|
if ticket_form.is_valid():
|
|
ticket = ticket_form.save()
|
|
return redirect('flux')
|
|
else:
|
|
ticket_form = TicketForm(instance=ticket)
|
|
|
|
return render(request,
|
|
'reviews/ticket_update.html',
|
|
{'ticket_form': ticket_form, 'ticket': ticket})
|
|
|
|
|
|
@login_required
|
|
@permission_required('review.delete_ticket', raise_exception=True)
|
|
def delete_ticket(request, ticket_id):
|
|
"""
|
|
Delete object from DB for a given Ticket
|
|
:param: request, Ticket id
|
|
:return: delete Ticket from DB
|
|
"""
|
|
ticket = Ticket.objects.get(id=ticket_id)
|
|
if request.method == 'POST':
|
|
ticket.delete()
|
|
return redirect('flux')
|
|
|
|
return render(request,
|
|
'reviews/ticket_delete.html',
|
|
{'ticket': ticket})
|
|
|
|
|
|
@login_required
|
|
def review(request, review_id):
|
|
"""
|
|
Get detail of a Review
|
|
:param: request, Review id
|
|
:return: object Review
|
|
"""
|
|
review = Review.objects.get(id=review_id)
|
|
return render(request,
|
|
'reviews/review.html',
|
|
{'review': review})
|
|
|
|
|
|
@login_required
|
|
def create_review(request):
|
|
"""
|
|
Create Ticket and related Review in one go
|
|
:param: request (POST)
|
|
:return: form, write Ticket and Review in DB
|
|
"""
|
|
ticket_form = TicketForm()
|
|
review_form = ReviewForm()
|
|
if request.method == 'POST':
|
|
ticket_form = TicketForm(request.POST, request.FILES)
|
|
review_form = ReviewForm(request.POST)
|
|
print(request.POST)
|
|
|
|
if all([ticket_form.is_valid(), review_form.is_valid()]):
|
|
ticket = ticket_form.save(commit=False)
|
|
review = review_form.save(commit=False)
|
|
review.user = ticket.user = request.user
|
|
review.save()
|
|
ticket.save()
|
|
review.ticket = ticket
|
|
ticket.review = review
|
|
ticket.save()
|
|
review.save()
|
|
return redirect('posts')
|
|
|
|
context = {
|
|
'ticket_form': ticket_form,
|
|
'review_form': review_form,
|
|
}
|
|
return render(request,
|
|
'reviews/review_create.html', context)
|
|
|
|
|
|
@login_required
|
|
def ticket_review(request, ticket_id):
|
|
"""
|
|
Create a Review for a given Ticket
|
|
:param: request (POST), Ticket id
|
|
:return: form, write Review in DB
|
|
"""
|
|
ticket = Ticket.objects.get(id=ticket_id)
|
|
review_form = ReviewForm()
|
|
if request.method == 'POST':
|
|
review_form = ReviewForm(request.POST)
|
|
print(request.POST)
|
|
if review_form.is_valid():
|
|
review = review_form.save(commit=False)
|
|
review.user = request.user
|
|
review.ticket = ticket
|
|
ticket.review = review
|
|
review.save()
|
|
ticket.review = review
|
|
ticket.save()
|
|
return redirect('posts')
|
|
context = {
|
|
'ticket': ticket,
|
|
'review_form': review_form,
|
|
}
|
|
return render(request,
|
|
'reviews/review_ticket.html', context)
|
|
|
|
|
|
@login_required
|
|
def update_review(request, review_id):
|
|
"""
|
|
Get new data for a given Review and save changes
|
|
:param: request, Review id
|
|
:return: form, write Review in DB
|
|
"""
|
|
review = Review.objects.get(id=review_id)
|
|
if request.method == 'POST':
|
|
print(request.POST)
|
|
review_form = ReviewForm(request.POST, instance=review)
|
|
print(review_form.is_valid())
|
|
if review_form.is_valid():
|
|
review = review_form.save()
|
|
return redirect('posts')
|
|
else:
|
|
review_form = ReviewForm(instance=review)
|
|
return render(request,
|
|
'reviews/review_update.html',
|
|
{'review_form': review_form, 'review': review})
|
|
|
|
|
|
@login_required
|
|
def delete_review(request, review_id):
|
|
"""
|
|
Delete object from DB for a given Review
|
|
:param: request, Review id
|
|
:return: delete Review from DB
|
|
"""
|
|
review = Review.objects.get(id=review_id)
|
|
if request.method == 'POST':
|
|
review.delete()
|
|
return redirect('posts')
|
|
return render(request,
|
|
'reviews/review_delete.html',
|
|
{'review': review})
|
|
|
|
|
|
@login_required
|
|
def subscribed(request):
|
|
"""
|
|
Create UserFollows object for a given username in form
|
|
Create a list of followed_users for current user
|
|
:param: request (POST with given username)
|
|
:return: queryset of UserFollows following current user,
|
|
and a queryset of UserFollows followed by current user
|
|
"""
|
|
follows = UserFollows()
|
|
user_form = ToFollowForm()
|
|
# list of UserFollows following current user
|
|
following = UserFollows.objects.filter(followed_user=request.user)
|
|
if request.method == 'POST':
|
|
user_form = ToFollowForm(request.POST)
|
|
if user_form.is_valid():
|
|
user = user_form.cleaned_data["user"]
|
|
follows.followed_user = User.objects.get(username=user)
|
|
follows.user = request.user
|
|
follows.save()
|
|
return redirect('subscribed')
|
|
# list of UserFollows followed by current user
|
|
followed = UserFollows.objects.filter(user=request.user)
|
|
context = {
|
|
'user_form': user_form,
|
|
'followed': followed,
|
|
'following': following
|
|
}
|
|
return render(request,
|
|
'reviews/subscribed.html', context)
|
|
|
|
|
|
@login_required
|
|
def unsubscribe(request, followed_user_id):
|
|
"""
|
|
Delete a UserFollows for a given followed_user
|
|
:param: request, User.id
|
|
:return: delete UserFollows from DB
|
|
"""
|
|
followed = UserFollows.objects.get(
|
|
user=request.user,
|
|
followed_user=followed_user_id)
|
|
if request.method == 'POST':
|
|
followed.delete()
|
|
return redirect('subscribed')
|
|
return render(request,
|
|
'reviews/unsubscribe.html',
|
|
{'followed': followed})
|