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)
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=datetime.now() ,
is_top=is_top
)
now_utc = timezone.now()
plus_eight = now_utc + timedelta(hours=8)
article_obj.created_time = plus_eight
article_obj.save(update_fields=['created_time'])
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)
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() + timedelta(hours=8),
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("错误的请求")