feat: 添加文章点赞功能。

Author: ShineYull

core/urls.py

@@ -64,6 +64,9 @@
     path('api/del_talk/', del_talk, name='del_talk'),
     path('api/run_online_script/', run_online_script, name='run_online_script'),
     path('api/send_email/', send_email, name="send_email"),
+    path('api/save_postlike/', save_postlike, name="save_postlike"),
+    path('api/delete_postlike/', delete_postlike, name="delete_postlike"),
+    path('api/change_postlike/', change_postlike, name="change_postlike"),
 
     path('pub/save/', pub.save, name='pub_save'),
     path('pub/save_post/', pub.save_post, name='pub_save_post'),
@@ -99,6 +102,8 @@
     path('pub/cancelSubscribe/', pub.cancelSubscribe, name='pub_cancelSubscribe'),
     path('pub/getSubscribeSum/', pub.getSubscribeSum, name='pub_getSubscribeSum'),
     path('pub/getVerificationCode/', pub.getVerificationCode, name='pub_getVerificationCode'),
+    path('pub/set_postlike/', pub.set_postlike, name='pub_set_postlike'),
+    path('pub/get_postlike/', pub.get_postlike, name='pub_get_postlike'),
 
     re_path(r'^(?!api)^(?!pub).*$\.*', pages, name='pages'),
 ]

hexoweb/api.py

@@ -9,6 +9,7 @@
 from django.views.decorators.csrf import csrf_exempt
 
 from core import settings
+from hexoweb.models import PostLikeModel
 from .functions import *
 
 
@@ -936,4 +937,60 @@ def send_email(request):
     html_content = getSubscribeHtml()
     for to_email in MailModel.objects.all():
         send_custom_email(to_email.mail, to_email.name, from_email, email_passd, html_content, 'html')
-    return JsonResponse(safe=False, data={"msg":"OK，邮件已经发送完成。", "status":"true"})
+    return JsonResponse(safe=False, data={"msg":"OK，邮件已经发送完成。", "status":"true"})
+
+# 保存点赞位
+@login_required(login_url="/login/")
+def save_postlike(request):
+    if request.method != "POST":
+        return
+
+    postname = request.POST.get("postname")
+    postlike = PostLikeModel.objects.filter(postName=postname).first()  # 查看是否已经订阅过
+    if (not postlike):
+        try:
+            PostLikeModel.objects.create(postName=postname)    
+            return JsonResponse(safe=False, data={"msg":"OK，创建点赞位成功。", "status":True})
+        except Exception as error:
+            return JsonResponse(safe=False, data={"msg":"数据库插入记录失败。", "status":False})
+    return JsonResponse(safe=False, data={"msg":"点赞位已经存在", "status":False})
+
+# 删除点赞位
+@login_required(login_url="/login/")
+def delete_postlike(request):
+    if request.method != "POST":
+        return
+
+    postname = request.POST.get("postname")
+    postlike = PostLikeModel.objects.filter(postName=postname).first()  # 查看是否已经订阅过
+    if (postlike):
+        try:
+            PostLikeModel.objects.filter(postName=postname).delete()
+            return JsonResponse(safe=False, data={"msg":"OK，删除点赞位成功。", "status":True})
+        except Exception as error:
+            return JsonResponse(safe=False, data={"msg":"数据库删除记录失败。", "status":False})
+    return JsonResponse(safe=False, data={"msg":"点赞位不存在！", "status":False})
+
+# 修改点赞位
+@login_required(login_url="/login/")
+def change_postlike(request):
+    if request.method != "POST":
+        return
+
+    print("进入")
+    postname = request.POST.get("postname")
+    content = request.POST.get("content")
+    print("正在查询")
+    postlike = PostLikeModel.objects.filter(postName=postname).first()  # 查看是否已经订阅过
+    if (postlike):
+        try:
+            print("修改值：", content)
+            print("文章名称：", postname)
+            PostLikeModel.objects.filter(postName=postname).update(postName=content)
+            print("修改")
+            return JsonResponse(safe=False, data={"msg":"OK，修改点赞位成功。", "status":True})
+        except Exception as error:
+            print("异常")
+            return JsonResponse(safe=False, data={"msg":"数据库修改记录失败。", "status":False})
+    print("查询不到")
+    return JsonResponse(safe=False, data={"msg":"点赞位不存在！", "status":False})

hexoweb/models.py

@@ -81,4 +81,8 @@ class VerificationCodeModel(models.Model):
 
 class MailModel(models.Model):
     mail=models.CharField(primary_key=True, max_length=50, verbose_name='邮箱')
-    name=models.CharField(unique=False, max_length=30, verbose_name='名称')
+    name=models.CharField(unique=False, max_length=30, verbose_name='名称')
+
+class PostLikeModel(models.Model):
+    postName=models.CharField(primary_key=True, max_length=50, verbose_name='文章名称')
+    like=models.IntegerField(default=0)

hexoweb/pub.py

@@ -10,7 +10,7 @@
 from core import settings
 
 from .functions import *
-from .models import EssayModel, ImageModel, MailModel, SettingModel, VerificationCodeModel
+from .models import EssayModel, ImageModel, MailModel, PostLikeModel, SettingModel, VerificationCodeModel
 
 
 # 保存内容 pub/save
@@ -824,4 +824,38 @@ def cancelSubscribe(request):
 # 订阅总数
 @csrf_exempt
 def getSubscribeSum(request):
-    return JsonResponse(safe=False, data={"msg": subscribe_sum(), "status": True})
+    return JsonResponse(safe=False, data={"msg": subscribe_sum(), "status": True})
+
+# 设置文章点赞
+@csrf_exempt
+def set_postlike(request):
+    if request.method != "POST":
+        return
+
+    try:
+        postname = json.loads(request.body).get('postname')
+        postlikemodel = PostLikeModel.objects.filter(postName=postname).first()
+        if (not postlikemodel):
+            return JsonResponse(safe=False, data={"msg": "文章未创建点赞功能！", "status": False})
+        postlikemodel.like += 1
+        count = postlikemodel.like
+        postlikemodel.save()
+    except Exception as error:
+        return JsonResponse(safe=False, data={"msg": "点赞发生错误，请检查数据库或网络！", "status": False})
+    return JsonResponse(safe=False, data={"msg": count, "status": True})
+
+# 获取文章点赞数
+@csrf_exempt
+def get_postlike(request):
+    if request.method != "POST":
+        return
+
+    try:
+        postname = json.loads(request.body).get('postname')
+        postlikemodel = PostLikeModel.objects.filter(postName=postname).first()
+        if (not postlikemodel):
+            return JsonResponse(safe=False, data={"msg": "文章未创建点赞功能！", "status": False})
+        count = postlikemodel.like
+    except Exception as error:
+        return JsonResponse(safe=False, data={"msg": "获取点赞数失败，请检查数据库或网络！", "status": True})
+    return JsonResponse(safe=False, data={"msg": count, "status": True})

hexoweb/views.py

@@ -601,6 +601,18 @@ def pages(request):
             context["post_number"] = len(posts)
             context["page_number"] = ceil(context["post_number"] / 15)
             context["search"] = search
+        elif "post_like" in load_template:
+            posts = []
+            postlike = PostLikeModel.objects.all()
+
+            for i in postlike:
+                posts.append({"postName": i.postName,
+                                "like": i.like,
+                            })
+
+            context["posts"] = posts
+            context["post_number"] = len(posts)
+            context["page_number"] = ceil(context["post_number"] / 15)
         elif "talks" in load_template:
             search = request.GET.get("s")
             posts = []