from django.db import models
from django.contrib.contenttypes.fields import GenericForeignKey
from django.contrib.contenttypes.models import ContentType
from django.contrib.auth.models import User

# Create your models here.


# 用于記錄點贊數量的模型
class LikeCount(models.Model):
 content_type = models.ForeignKey(ContentType, on_delete=models.DO_NOTHING)
 object_id = models.PositiveIntegerField()
 content_object = GenericForeignKey('content_type', 'object_id')
 # 用于記錄點贊數量的字段
 like_num = models.IntegerField(default=0)


# 用于記錄點贊狀態的模型
class LikeRecord(models.Model):
 content_type=models.ForeignKey(ContentType, on_delete=models.DO_NOTHING)
 object_id=models.PositiveIntegerField()
 content_object=GenericForeignKey('content_type', 'object_id')
 # 記錄點贊的用戶
 like_user = models.ForeignKey(User, on_delete=models.DO_NOTHING)
 # 記錄點贊的時間
 like_time = models.DateTimeField(auto_now_add=True)

from django.shortcuts import render, HttpResponseRedirect
from django.contrib.contenttypes.models import ContentType
from django.http import JsonResponse
from .models import LikeCount, LikeRecord

# Create your views here.


# 數據操作成功返回數據方法
def success_response(like_num):
 data = {}
 data['status'] = 'SUCCESS'
 data['like_num'] = like_num
 return JsonResponse(data)


# 數據操作失敗返回信息的方法
def error_response(message):
 data = {}
 data['status'] = 'ERROR'
 data['message'] = message
 return JsonResponse(data)


def like_up(request):
 # 得到GET中的數據以及當前用戶
 user = request.user
 # 判斷用戶是否登錄
 if not user.is_authenticated:
  return error_response('未登錄，不能進行點贊操作')
 content_type = request.GET.get('content_type')
 content_type = ContentType.objects.get(model=content_type)
 object_id = request.GET.get('object_id')
 is_like = request.GET.get('is_like')

 # 創建一個點贊記錄
 if is_like == 'true':
  # 進行點贊，即實例化一個點贊記錄
  like_record, created = LikeRecord.objects.get_or_create(content_type=content_type, object_id=object_id, like_user=user)
  # 通過created來判斷點贊記錄是否存在，如果存在則不進行點贊，如果不存在則進行點贊數量加一
  if created:
   # 不存在點贊記錄并且已經創建點贊記錄，需要將點贊數量加一
   like_count, created = LikeCount.objects.get_or_create(content_type=content_type, object_id=object_id)
   like_count.like_num += 1
   like_count.save()
   return success_response(like_count.like_num)
  else:
   # 已經進行過點贊
   return error_response('已經點贊過')
 else:
  # 取消點贊
  # 先查詢數據是否存在，存在則進行取消點贊
  if LikeRecord.objects.filter(content_type=content_type, object_id=object_id, like_user=user).exists():
   # 數據存在，取消點贊
   # 刪除點贊記錄
   LikeRecord.objects.get(content_type=content_type, object_id=object_id, like_user=user).delete()
   # 判斷對應的點贊數量數據是否存在，如果存在則對點贊數量進行減一
   like_count, create = LikeCount.objects.get_or_create(content_type=content_type, object_id=object_id)
   if create:
    # 數據不存在，返回錯誤信息
    return error_response('數據不存在，不能取消點贊')
   else:
    # 數據存在，對數量進行減一
    like_count.like_num -= 1
    like_count.save()
    return success_response(like_count.like_num)
  else:
   # 數據不存在，不能取消點贊
   return error_response('數據不存在，不能取消點贊')

<script type="text/javascript">
 function change_like(obj, content_type, object_id) {
  // 判斷obj中是否包含active的元素，用于判斷當前狀態是否為激活狀態
  var is_like = obj.getElementsByClassName('active').length == 0
  $.ajax({
   url: '/like_up/',
   // 為了避免加入csrf_token令牌，所以使用GET請求
   type: 'GET',
   // 返回的數據用于創建一個點贊記錄
   data: {
    content_type: content_type,
    object_id: object_id,
    is_like: is_like,
   },
   cache: false,
   success: function (data) {
    console.log(data);
    if (data['status'] == 'SUCCESS'){
     // 更新點贊狀態
     // 通過class找到對應的標簽
     var record = $(obj.getElementsByClassName('glyphicon'))
     if (is_like){
      record.addClass('active')
     }
     else {
      record.removeClass('active')
     }
     // 更新點贊數量
     var like_num = $(obj.getElementsByClassName('like_num'))
     like_num.text('（' + data['like_num'] + '）')

    }
    else {
     // 以彈窗的形式顯示錯誤信息
     alert(data['message'])
    }
   },
   error: function (xhr) {
    console.log(xhr)
   }
  });
  return false;
 };
</script>