MeChat/chat/views.py

from django.shortcuts import render
from django.http import HttpResponse
from django.views.decorators.csrf import csrf_exempt
from django.contrib.auth.models import User
from django.contrib.auth.decorators import login_required
import json
from friends.models import Friends

from .form import StoreDataForm
from .models import Message
from django.http import JsonResponse
from django.db.models import Q

DEBUG = True


# 批处理流程
# def more_clean(request, form):
#     if request.method == 'POST':
#         user_login_form = form(request.POST)
#         if user_login_form.is_valid():
#             # 清洗出合法的数据
#             data = user_login_form.cleaned_data
#             return data
#     else:
#         response = {'result': 'illegal input'}
#         return HttpResponse(json.dumps(response))


#
# 存储消息到表中
@csrf_exempt
@login_required(login_url='/account/login/')
def send_message(request):
    if request.method == 'POST':
        store_data_form = StoreDataForm(request.POST)
        if store_data_form.is_valid():
            data = store_data_form.cleaned_data
            user = User.objects.filter(username=data['toUsername'])
            if user:
                temp_friends_1 = Friends.objects.filter(
                    user=User.objects.filter(username=data["toUsername"]).get(), friend=request.user)
                temp_friends_2 = Friends.objects.filter(
                    friend=User.objects.filter(username=data["toUsername"]).get(), user=request.user)
                if temp_friends_1 or temp_friends_2:
                    Message.objects.create(
                        fromUser=request.user,
                        toUser=user.get(),
                        plaintext=data['plaintext'],
                    )
                    return HttpResponse(status=200)
                else:
                    return HttpResponse(status=423)
            else:
                return HttpResponse(status=422)
        else:
            return HttpResponse(status=400)
    elif request.method == 'GET':
        if DEBUG:
            return render(request, 'send_message.html')
    else:
        return HttpResponse(status=400)
    # try:
    #     Message.objects.create(
    #         date=post_data["date"],
    #         belongUserid=postData["belongUserid"],
    #         fromUserid=postData["fromUserid"],
    #         toUserid=postData["toUserid"],
    #         EphemeralPub=postData["EphemeralPub"],
    #         plaintext=postData["plaintext"],
    #         EphemeralPri=postData["EphemeralPri"])
    #     response = {"result": "添加成功"}
    #     return JsonResponse(response)
    # except:
    #     response = {"result": "添加失败！"}
    #     return JsonResponse(response)


# 适用于设备的漫游消息
@csrf_exempt
@login_required(login_url='/account/login/')
def filter_messages(request):
    if request.method == 'POST':
        data = request.POST
        user = User.objects.filter(username=data['username'])
        if user:
            temp_friends_1 = Friends.objects.filter(
                user=User.objects.filter(username=data["toUsername"]).get(), friend=request.user)
            temp_friends_2 = Friends.objects.filter(
                friend=User.objects.filter(username=data["toUsername"]).get(), user=request.user)
            if temp_friends_1 or temp_friends_2:
                # to_user = User.objects.filter(username=data['username'])
                # messages = Message.objects.filter(
                #     (Q(fromUser=to_user) & Q(toUser=request.user)) |
                #     (Q(fromUser=request.user) & Q(toUser=to_user))
                # ).order_by("date")
                response = []
                messages = Message.objects.filter(fromUser__username=data['username'],
                                                  toUser__username=request.user.username)
                for message in messages:
                    message_dict = {'from_user_name': message.fromUser.username, 'plaintext': message.plaintext,
                                    'to_user_name': request.user.username}
                    response.append(message_dict)
                messages = Message.objects.filter(toUser__username=data['username'],
                                                  fromUser__username=request.user.username)
                for message in messages:
                    message_dict = {'from_user_name': message.fromUser.username, 'plaintext': message.plaintext,
                                    'to_user_name': request.user.username}
                    response.append(message_dict)
                return HttpResponse(json.dumps(response), status=200)
            else:
                return HttpResponse(status=423)
        else:
            return HttpResponse(status=422)
    elif request.method == 'GET':
        if DEBUG:
            return render(request, 'filter_message.html')
    else:
        return HttpResponse(status=400)
    # try:
    #     loginUserid = data["fromUsername"]
    #     messagesTemp = list(Message.objects.filter(
    #
    #         (Q(fromUserid=pk) & Q(toUserid=loginUserid) & Q(belongUserid=loginUserid)) |
    #         (Q(fromUserid=loginUserid) & Q(toUserid=pk) & Q(belongUserid=loginUserid))
    #     ).order_by("date"))
    #     response = {"result": "已发送", "data": messagesTemp}
    #     return JsonResponse(response)
    # except Message.DoesNotExist:
    #     response = {"result": "该用户不存在或者没有消息记录"}
    #     return JsonResponse(response)


# 发送信息
@login_required(login_url='/account/login/')
def request_for_message(request):
    messages = Message.objects.filter(toUser__username=request.user.username, is_received=False)
    response = []
    for message in messages:
        message_dict = {'from_user_name': message.fromUser.username, 'plaintext': message.plaintext,
                        'to_user_name': request.user.username, 'message_id': message.message_id}
        response.append(message_dict)
    return HttpResponse(json.dumps(response), status=200)


# 成功接收消息
@login_required(login_url='/account/login/')
def receive_successfully(request):
    if request.method == 'POST':
        data = request.POST
        try:
            message = Message.objects.filter(message_id=data['message_id']).get()
            message.is_received = True
            message.save()
            return HttpResponse(status=200)
        except Message.DoesNotExist:
            return HttpResponse(status=423)
    elif request.method == 'GET':
        if DEBUG:
            return render(request, 'receive_successfully.html')
    else:
        return HttpResponse(status=400)