|
|
from django.shortcuts import render
|
|
|
|
|
|
# Create your views here.
|
|
|
import os
|
|
|
import datetime
|
|
|
from django.views.generic.list import ListView
|
|
|
from django.views.generic.detail import DetailView
|
|
|
from django.conf import settings
|
|
|
from django import forms
|
|
|
from django.http import HttpResponse
|
|
|
from django.http import HttpResponseRedirect
|
|
|
from django.views.decorators.csrf import csrf_exempt
|
|
|
from django.contrib.auth.decorators import login_required
|
|
|
from DjangoBlog.utils import cache, logger
|
|
|
from django.shortcuts import get_object_or_404
|
|
|
from blog.models import Article, Category, Tag
|
|
|
from comments.forms import CommentForm
|
|
|
|
|
|
|
|
|
class ArticleListView(ListView):
|
|
|
# template_name属性用于指定使用哪个模板进行渲染
|
|
|
template_name = 'blog/article_index.html'
|
|
|
|
|
|
# context_object_name属性用于给上下文变量取名（在模板中使用该名字）
|
|
|
context_object_name = 'article_list'
|
|
|
|
|
|
# 页面类型，分类目录或标签列表等
|
|
|
page_type = ''
|
|
|
paginate_by = settings.PAGINATE_BY
|
|
|
page_kwarg = 'page'
|
|
|
|
|
|
def get_view_cache_key(self):
|
|
|
return self.request.get['pages']
|
|
|
|
|
|
@property
|
|
|
def page_number(self):
|
|
|
page_kwarg = self.page_kwarg
|
|
|
page = self.kwargs.get(page_kwarg) or self.request.GET.get(page_kwarg) or 1
|
|
|
return page
|
|
|
|
|
|
def get_queryset_cache_key(self):
|
|
|
"""
|
|
|
子类重写.获得queryset的缓存key
|
|
|
"""
|
|
|
raise NotImplementedError()
|
|
|
|
|
|
def get_queryset_data(self):
|
|
|
"""
|
|
|
子类重写.获取queryset的数据
|
|
|
"""
|
|
|
raise NotImplementedError()
|
|
|
|
|
|
def get_queryset_from_cache(self, cache_key):
|
|
|
# raise NotImplementedError()
|
|
|
value = cache.get(cache_key)
|
|
|
if value:
|
|
|
logger.info('get view cache.key:{key}'.format(key=cache_key))
|
|
|
return value
|
|
|
else:
|
|
|
article_list = self.get_queryset_data()
|
|
|
cache.set(cache_key, article_list)
|
|
|
logger.info('set view cache.key:{key}'.format(key=cache_key))
|
|
|
return article_list
|
|
|
|
|
|
def get_queryset(self):
|
|
|
key = self.get_queryset_cache_key()
|
|
|
value = self.get_queryset_from_cache(key)
|
|
|
return value
|
|
|
|
|
|
|
|
|
class IndexView(ArticleListView):
|
|
|
def get_queryset_data(self):
|
|
|
article_list = Article.objects.filter(type='a', status='p')
|
|
|
return article_list
|
|
|
|
|
|
def get_queryset_cache_key(self):
|
|
|
cache_key = 'index_{page}'.format(page=self.page_number)
|
|
|
return cache_key
|
|
|
|
|
|
|
|
|
class ArticleDetailView(DetailView):
|
|
|
template_name = 'blog/article_detail.html'
|
|
|
model = Article
|
|
|
pk_url_kwarg = 'article_id'
|
|
|
context_object_name = "article"
|
|
|
|
|
|
def get_object(self, queryset=None):
|
|
|
obj = super(ArticleDetailView, self).get_object()
|
|
|
obj.viewed()
|
|
|
self.object = obj
|
|
|
return obj
|
|
|
|
|
|
def get_context_data(self, **kwargs):
|
|
|
articleid = int(self.kwargs[self.pk_url_kwarg])
|
|
|
comment_form = CommentForm()
|
|
|
user = self.request.user
|
|
|
|
|
|
if user.is_authenticated and not user.is_anonymous and user.email and user.username:
|
|
|
comment_form.fields.update({
|
|
|
'email': forms.CharField(widget=forms.HiddenInput()),
|
|
|
'name': forms.CharField(widget=forms.HiddenInput()),
|
|
|
})
|
|
|
comment_form.fields["email"].initial = user.email
|
|
|
comment_form.fields["name"].initial = user.username
|
|
|
|
|
|
article_comments = self.object.comment_list()
|
|
|
|
|
|
kwargs['form'] = comment_form
|
|
|
kwargs['article_comments'] = article_comments
|
|
|
kwargs['comment_count'] = len(article_comments) if article_comments else 0
|
|
|
|
|
|
kwargs['next_article'] = self.object.next_article
|
|
|
kwargs['prev_article'] = self.object.prev_article
|
|
|
|
|
|
return super(ArticleDetailView, self).get_context_data(**kwargs)
|
|
|
|
|
|
|
|
|
class CategoryDetailView(ArticleListView):
|
|
|
page_type = "分类目录归档"
|
|
|
|
|
|
def get_queryset_data(self):
|
|
|
slug = self.kwargs['category_name']
|
|
|
category = get_object_or_404(Category, slug=slug)
|
|
|
|
|
|
categoryname = category.name
|
|
|
self.categoryname = categoryname
|
|
|
categorynames = list(map(lambda c: c.name, category.get_sub_categorys()))
|
|
|
article_list = Article.objects.filter(category__name__in=categorynames, status='p')
|
|
|
return article_list
|
|
|
|
|
|
def get_queryset_cache_key(self):
|
|
|
slug = self.kwargs['category_name']
|
|
|
category = get_object_or_404(Category, slug=slug)
|
|
|
categoryname = category.name
|
|
|
self.categoryname = categoryname
|
|
|
cache_key = 'category_list_{categoryname}_{page}'.format(categoryname=categoryname, page=self.page_number)
|
|
|
return cache_key
|
|
|
|
|
|
def get_context_data(self, **kwargs):
|
|
|
|
|
|
categoryname = self.categoryname
|
|
|
try:
|
|
|
categoryname = categoryname.split('/')[-1]
|
|
|
except:
|
|
|
pass
|
|
|
kwargs['page_type'] = CategoryDetailView.page_type
|
|
|
kwargs['tag_name'] = categoryname
|
|
|
return super(CategoryDetailView, self).get_context_data(**kwargs)
|
|
|
|
|
|
|
|
|
class AuthorDetailView(ArticleListView):
|
|
|
page_type = '作者文章归档'
|
|
|
|
|
|
def get_queryset_cache_key(self):
|
|
|
author_name = self.kwargs['author_name']
|
|
|
cache_key = 'author_{author_name}_{page}'.format(author_name=author_name, page=self.page_number)
|
|
|
return cache_key
|
|
|
|
|
|
def get_queryset_data(self):
|
|
|
author_name = self.kwargs['author_name']
|
|
|
article_list = Article.objects.filter(author__username=author_name)
|
|
|
return article_list
|
|
|
|
|
|
def get_context_data(self, **kwargs):
|
|
|
author_name = self.kwargs['author_name']
|
|
|
kwargs['page_type'] = AuthorDetailView.page_type
|
|
|
kwargs['tag_name'] = author_name
|
|
|
return super(AuthorDetailView, self).get_context_data(**kwargs)
|
|
|
|
|
|
|
|
|
class TagListView(ListView):
|
|
|
template_name = ''
|
|
|
context_object_name = 'tag_list'
|
|
|
|
|
|
def get_queryset(self):
|
|
|
tags_list = []
|
|
|
tags = Tag.objects.all()
|
|
|
for t in tags:
|
|
|
t.article_set.count()
|
|
|
|
|
|
|
|
|
class TagDetailView(ArticleListView):
|
|
|
page_type = '分类标签归档'
|
|
|
|
|
|
def get_queryset_data(self):
|
|
|
slug = self.kwargs['tag_name']
|
|
|
tag = get_object_or_404(Tag, slug=slug)
|
|
|
tag_name = tag.name
|
|
|
self.name = tag_name
|
|
|
article_list = Article.objects.filter(tags__name=tag_name)
|
|
|
return article_list
|
|
|
|
|
|
def get_queryset_cache_key(self):
|
|
|
slug = self.kwargs['tag_name']
|
|
|
tag = get_object_or_404(Tag, slug=slug)
|
|
|
tag_name = tag.name
|
|
|
self.name = tag_name
|
|
|
cache_key = 'tag_{tag_name}_{page}'.format(tag_name=tag_name, page=self.page_number)
|
|
|
return cache_key
|
|
|
|
|
|
def get_context_data(self, **kwargs):
|
|
|
# tag_name = self.kwargs['tag_name']
|
|
|
tag_name = self.name
|
|
|
kwargs['page_type'] = TagDetailView.page_type
|
|
|
kwargs['tag_name'] = tag_name
|
|
|
return super(TagDetailView, self).get_context_data(**kwargs)
|
|
|
|
|
|
|
|
|
@csrf_exempt
|
|
|
def fileupload(request):
|
|
|
if request.method == 'POST':
|
|
|
response = []
|
|
|
for filename in request.FILES:
|
|
|
timestr = datetime.datetime.now().strftime('%Y/%m/%d')
|
|
|
imgextensions = ['jpg', 'png', 'jpeg', 'bmp']
|
|
|
fname = u''.join(str(filename))
|
|
|
|
|
|
isimage = len([i for i in imgextensions if fname.find(i) >= 0]) > 0
|
|
|
|
|
|
basepath = r'/var/www/resource/{type}/{timestr}'.format(
|
|
|
type='files' if not isimage else'image', timestr=timestr)
|
|
|
if settings.TESTING:
|
|
|
basepath = settings.BASE_DIR + '/uploads'
|
|
|
url = 'https://resource.lylinux.net/{type}/{timestr}/{filename}'.format(
|
|
|
type='files' if not isimage else'image', timestr=timestr, filename=filename)
|
|
|
if not os.path.exists(basepath):
|
|
|
os.makedirs(basepath)
|
|
|
savepath = os.path.join(basepath, filename)
|
|
|
with open(savepath, 'wb+') as wfile:
|
|
|
for chunk in request.FILES[filename].chunks():
|
|
|
wfile.write(chunk)
|
|
|
if isimage:
|
|
|
from PIL import Image
|
|
|
image = Image.open(savepath)
|
|
|
image.save(savepath, quality=20, optimize=True)
|
|
|
response.append(url)
|
|
|
return HttpResponse(response)
|
|
|
|
|
|
else:
|
|
|
return HttpResponse("only for post")
|
|
|
|
|
|
|
|
|
@login_required
|
|
|
def refresh_memcache(request):
|
|
|
try:
|
|
|
|
|
|
if request.user.is_superuser:
|
|
|
from DjangoBlog.utils import cache
|
|
|
if cache and cache is not None:
|
|
|
cache.clear()
|
|
|
return HttpResponse("ok")
|
|
|
else:
|
|
|
from django.http import HttpResponseForbidden
|
|
|
return HttpResponseForbidden()
|
|
|
except Exception as e:
|
|
|
return HttpResponse(e)
|
