def retrieve(self, request, pk=None):
message_list = Dynamo.get_messages_by_recipient(pk)
serializer = MessageSerializer(message_list, many=True)
if len(serializer.data):
return Response({"Status": "Success", "Response": serializer.data})
return Response({"Status": "Fail", "Response": serializer.data})
def message_view(request):
if request.user.is_authenticated():
user = get_object_or_404(User, username=request.user.username)
messages = Dynamo.get_messages_by_recipient(user.username)
if request.user.is_staff:
return render(request,'createMessage.html', {"message_list": messages})
else:
return render(request, 'userMessages.html', {"message_list": messages})
return redirect("../")
def send_message(request):
try:
user = User.objects.get(username=str(request.POST['username'])).username
to_send = Message(
sender='test',
recipient=user,
urgency=int(request.POST['urgency']),
content=str(request.POST['message']),
timestamp=int(time.time()),
read=False
)
Dynamo.initialize().send_message(MessageSerializer(to_send).data)
Dynamo.get_messages_by_recipient(user)
return redirect(message_view)
except Exception as e:
print("\tERROR\tFailed to create message: " + str(e))
return redirect(error_message)
