Blog/blog/controller/article.py

import json
from datetime import datetime, timedelta

from django.contrib.auth.decorators import login_required
from django.core.paginator import Paginator, PageNotAnInteger, EmptyPage
from django.http import JsonResponse, HttpResponseRedirect
from django.shortcuts import render, redirect
from django.urls import reverse
from django.utils import timezone

import blog
from blog import models
from system.error.ServerException import ServerException


@login_required(login_url='/login')
def article(request):
    page = request.GET.get('page')
    search_title = request.GET.get('search_title')
    search_content = request.GET.get('search_content')
    search_category = request.GET.get('search_category')
    if page is None:
        page = 0
    articles = blog.controller.article.get_article(is_paginator=True, page=int(page), search_title=search_title,
                                                   search_content=search_content, search_category=search_category,draft=False)
    return render(request, 'management/article/article.html', context={'articles': articles})


@login_required(login_url='/login')
def draft(request):
    page = request.GET.get('page')
    search_title = request.GET.get('search_title')
    search_content = request.GET.get('search_content')
    search_category = request.GET.get('search_category')
    if page is None:
        page = 0
    articles = blog.controller.article.get_article(is_paginator=True, page=int(page), search_title=search_title,
                                                   search_content=search_content, search_category=search_category,
                                                   draft=True)
    return render(request, 'management/article/draft.html', context={'articles': articles})

@login_required(login_url='/login')
def set_status(request,pk):

    article = models.Article.objects.filter(id=int(pk)).first()
    if article is not None:
        if article.status == 0:
            article.status = 1
        else:
            article.status = 0
        article.save()
        # 把所有 GET 参数重新编码（可能包含 page，也可能不包含）
        query = request.GET.urlencode()  # e.g. "page=2&foo=bar" 或者 ""

        # 构造基础列表页 URL
        base = '/management/article'  # 或者直接 '/management/article'

        # 如果有任何查询参数，就拼接 ?xxx，否则直接跳
        redirect_url = f'{base}?{query}' if query else base
        return redirect(redirect_url)


@login_required(login_url='/login')
def cancel_show_index(request, pk):
    article = models.Article.objects.get(id=int(pk))
    article.is_top = False
    article.save()
    return HttpResponseRedirect('/')  # 跳转到主界面

@login_required(login_url='/login')
def show_show_index(request, pk):
    article = models.Article.objects.get(id=int(pk))
    article.is_top = True
    article.save()
    return HttpResponseRedirect('/')  # 跳转到主界面


def new(request):
    return render(request, 'management/article/new_article.html')


def show_article(request, pk):
    article = models.Article.objects.filter(id=int(pk)).values('id', 'title', 'intro', 'category',
                                                               'user', 'created_time', 'type',
                                                               'cover__file_net_path',
                                                               'music__file_net_path', 'status',
                                                               'category__name',
                                                               'user__first_name', 'user_id',
                                                               'user__last_name', 'markdown_text', 'is_top',
                                                               'html_text').first()
    comments = models.Comment.objects.filter(article_id=article['id'])
    # request.session['msg'] = ''
    from blog.views import get_record_and_tags
    result = get_record_and_tags()



    return render(request, 'blog.html', context={'article': article, 'records': result['records'],
                                                 'tags': result['tags'], 'comments': comments})


@login_required(login_url='/')
def to_edit(request, pk):
    article = models.Article.objects.filter(id=int(pk)).values('id', 'title', 'intro', 'category',
                                                               'user', 'created_time', 'type',
                                                               'cover__file_net_path',
                                                               'music__file_net_path', 'status',
                                                               'category__name',
                                                               'user__first_name', 'user_id',
                                                               'user__last_name', 'markdown_text', 'html_text',
                                                               'is_top').first()
    tags = models.Tags.objects.raw(
        'select * from blog_tags left join blog_article_tags bat on blog_tags.id = bat.tags_id where  bat.article_id = {0}'.format(
            int(pk)))
    tags_str = []
    for tag in tags:
        tags_str.append(tag.name)
    return render(request, 'management/article/edit_article.html',
                  context={'article': article, 'tags': ','.join(tags_str)})


@login_required(login_url='/')
def edit_article(request):
    if request.method == 'POST':
        title = request.POST['title']
        intro = request.POST['intro']
        category = request.POST['category']
        html_text = request.POST['html']
        markdown_text = request.POST['markdown']
        type_ = request.POST['type']
        status = request.POST['status']
        tags = request.POST['tags']
        pk = request.POST['pk']
        is_top = request.POST['is_top']
        category_obj = models.Category.objects.get(id=category)

        article_obj = models.Article.objects.get(id=int(pk))
        tags_id = []
        for tag in str(tags).split(','):
            if tag == '':
                break
            else:
                tag_obj = models.Tags.objects.filter(name=tag).first()
                if tag_obj is None:
                    obj = models.Tags(name=tag)
                    obj.save()
                    tags_id.append(obj.pk)
                else:
                    tags_id.append(tag_obj.pk)
        if len(tags_id) > 0:
            article_obj.tags.set(tags_id)
        # models.Article.objects.filter(id=int(pk)).update(title=title, intro=intro, category=category_obj,
        #                                                  html_text=html_text,
        #                                                  markdown_text=markdown_text,
        #                                                  type=type_, status=status,
        #                                                  tags=str(tags).split(',')
        #                                                  )
        article_obj.title = title
        article_obj.intro = intro
        article_obj.category = category_obj
        article_obj.html_text = html_text
        article_obj.markdown_text = markdown_text
        article_obj.type = type_
        article_obj.status = status
        article_obj.is_top = is_top
        article_obj.save()
        return JsonResponse({"success": True, "message": "修改成功"})
    else:
        raise ServerException("错误的请求")


@login_required(login_url='/')
def add_article(request):
    if request.method == 'POST':
        title = request.POST['title']
        intro = request.POST['intro']
        category = request.POST['category']
        html_text = request.POST['html']
        markdown_text = request.POST['markdown']
        type_ = request.POST['type']
        status = request.POST['status']
        tags = request.POST['tags']
        is_top = request.POST['is_top']

        category_obj = models.Category.objects.get(id=category)
        now_utc = timezone.now()  # aware UTC 时间
        # ① 如果你想要 UTC+8：
        beijing_time = now_utc + timedelta(hours=8)
        article_obj = models.Article.objects.create(title=title, intro=intro, category=category_obj,
                                                    html_text=html_text,
                                                    markdown_text=markdown_text,
                                                    type=type_, status=status, user=request.user,
                                                    created_time=beijing_time,
                                                    is_top=is_top
                                                    )
        tags_id = []
        for tag in str(tags).split(','):
            if tag == '':
                break
            else:
                tag_obj = models.Tags.objects.filter(name=tag).first()
                if tag_obj is None:
                    obj = models.Tags(name=tag)
                    obj.save()
                    tags_id.append(obj.pk)
                else:
                    tags_id.append(tag_obj.pk)
        if len(tags_id) > 0:
            article_obj.tags.set(tags_id)
        return JsonResponse({"success": True, "message": "添加成功"})
    else:
        raise ServerException("错误的请求")


@login_required(login_url='/')
def add_media(request):
    if request.method == 'POST':
        title = request.POST['title']
        category = request.POST['category']
        type_ = request.POST['type']
        status = request.POST['status']
        tags = request.POST['tags']
        cover = request.POST['cover_id']
        music_id = request.POST['music_id']
        cover_obj = models.FileRecord.objects.get(id=cover)
        music_obj = models.FileRecord.objects.get(id=music_id)
        now_utc = timezone.now()  # aware UTC 时间
        # ① 如果你想要 UTC+8：
        beijing_time = now_utc + timedelta(hours=8)
        category_obj = models.Category.objects.get(id=category)
        article_obj = models.Article.objects.create(title=title, category=category_obj,
                                                    type=type_, status=status, user=request.user,
                                                    created_time=beijing_time,
                                                    cover=cover_obj, music=music_obj
                                                    )
        if tags != 'null':
            article_obj.tags.set(str(tags).split(','))
        return JsonResponse({"success": True, "message": "添加成功"})
    else:
        raise ServerException("错误的请求")


@login_required(login_url='/')
def add_music(request):
    if request.method == 'POST':
        title = request.POST['title']
        category = request.POST['category']
        type_ = request.POST['type']
        status = request.POST['status']
        tags = request.POST['tags']
        cover = request.POST['cover_id']
        music_id = request.POST['music_id']
        cover_obj = models.FileRecord.objects.get(id=cover)
        music_obj = models.FileRecord.objects.get(id=music_id)

        category_obj = models.Category.objects.get(id=category)
        article_obj = models.Article.objects.create(title=title, category=category_obj,
                                                    type=type_, status=status, user=request.user,
                                                    created_time=datetime.now(),
                                                    cover=cover_obj, music=music_obj
                                                    )
        if tags != 'null':
            article_obj.tags.set(str(tags).split(','))
        return JsonResponse({"success": True, "message": "添加成功"})
    else:
        raise ServerException("错误的请求")


def set_tag(articles):
    for i in range(len(articles)):
        tags = models.Tags.objects.raw(
            'select * from blog_tags left join blog_article_tags bat on blog_tags.id = bat.tags_id where  bat.article_id = {0}'.format(
                articles[i]['id']))
        tags_field = []
        for item in tags:
            tags_field.append(item.name)
        articles[i]['tags'] = tags_field


def get_article(top: int = -1, page: int = -1, is_paginator: bool = False, category: models.Category = None,
                tag: models.Tags = None, search_category: str = None, search_title: str = None,
                search_content: str = None, date_record: str = None,
                draft: bool = False):
    if category is None and tag is None and date_record is None and draft == False:
        search_dict = dict()
        search_dict['status'] = 1
        if search_title:
            search_dict['title__contains'] = search_title
        if search_content:
            search_dict['markdown_text__contains'] = search_content
        if search_category:
            search_dict['category'] = search_category
        articles = models.Article.objects.filter(**search_dict).order_by('-created_time').values(
            'id', 'title', 'intro', 'category',
            'user', 'created_time', 'type',
            'cover__file_net_path',
            'music__file_net_path', 'status',
            'category__name',
            'user__first_name', 'user_id',
            'user__last_name',
            'is_top', 'html_text')
    elif date_record is not None:
        date_format = "%Y-%m"
        date = datetime.strptime(date_record, date_format)
        articles = models.Article.objects.filter(created_time__year=date.year, created_time__month=date.month).order_by(
            '-created_time').values('id', 'title',
                                    'intro',
                                    'category',
                                    'user',
                                    'created_time',
                                    'type',
                                    'cover__file_net_path',
                                    'music__file_net_path',
                                    'status',
                                    'category__name',
                                    'user__first_name',
                                    'user_id',
                                    'user__last_name',
                                    'is_top',
                                    'html_text').distinct()
    elif tag is not None:
        articles = models.Article.objects.filter(tags__name=tag.name).order_by('-created_time').values('id', 'title',
                                                                                                       'intro',
                                                                                                       'category',
                                                                                                       'user',
                                                                                                       'created_time',
                                                                                                       'type',
                                                                                                       'cover__file_net_path',
                                                                                                       'music__file_net_path',
                                                                                                       'status',
                                                                                                       'category__name',
                                                                                                       'user__first_name',
                                                                                                       'user_id',
                                                                                                       'user__last_name',
                                                                                                       'tags__name',
                                                                                                       'is_top',
                                                                                                       'html_text')
    elif draft:
        search_dict = dict()
        search_dict['status'] = 0

        if search_title:
            search_dict['title__contains'] = search_title
        if search_content:
            search_dict['markdown_text__contains'] = search_content
        if search_category:
            search_dict['category'] = search_category
        articles = models.Article.objects.filter(**search_dict).order_by('-created_time').values('id', 'title',
                                                                                                 'intro',
                                                                                                 'category',
                                                                                                 'user',
                                                                                                 'created_time',
                                                                                                 'type',
                                                                                                 'cover__file_net_path',
                                                                                                 'music__file_net_path',
                                                                                                 'status',
                                                                                                 'category__name',
                                                                                                 'user__first_name',
                                                                                                 'user_id',
                                                                                                 'user__last_name',
                                                                                                 'is_top',
                                                                                                 'html_text').distinct()
    else:
        articles = models.Article.objects.filter(category=category).order_by('-created_time').values('id', 'title',
                                                                                                     'intro',
                                                                                                     'category',
                                                                                                     'user',
                                                                                                     'created_time',
                                                                                                     'type',
                                                                                                     'cover__file_net_path',
                                                                                                     'music__file_net_path',
                                                                                                     'status',
                                                                                                     'category__name',
                                                                                                     'user__first_name',
                                                                                                     'user_id',
                                                                                                     'user__last_name',
                                                                                                     'is_top',
                                                                                                     'html_text')

    if is_paginator:
        if page <= 0:
            page = 1
        paginator = Paginator(articles, 15)
        try:
            articles = paginator.page(page)
        except PageNotAnInteger:
            articles = paginator.page(1)
        except EmptyPage:
            articles = paginator.page(paginator.num_pages)
        set_tag(articles=articles)

        return articles
    else:
        articles = list(articles)
    if top > -1:
        articles = articles[:top]
    set_tag(articles=articles)
    return articles
    # return JsonResponse({'all_article': json.loads(json.dumps(articles, default=str))})


@login_required(login_url='/')
def get_all_article(request):
    articles = models.Article.objects.order_by('-created_time').values('id', 'title',
                                                                       'intro',
                                                                       'category',
                                                                       'user',
                                                                       'created_time',
                                                                       'type',
                                                                       'cover__file_net_path',
                                                                       'music__file_net_path',
                                                                       'status',
                                                                       'category__name',
                                                                       'user__first_name', 'user_id',
                                                                       'user__last_name')
    return JsonResponse(json.loads(json.dumps(list(articles), default=str)), safe=False)


@login_required(login_url='/')
def delete_article(request):
    if request.method == 'GET':
        id = request.GET.get("id")
        article = models.Article.objects.get(id=int(id))
        article.tags.clear()
        article.delete()

        return HttpResponseRedirect('/management/article')  # 跳转到主界面
    else:
        raise ServerException("错误的请求")