def diary_del(diary_id):
"""Admin Diary Delete Action
Used for delete Diary.Also del reference Tag and Category.
Methods:
GET
Args:
diary_id: diary ObjectID
Returns:
none
"""
diary = Diary.objects(pk=diary_id)
Category.objects(name=diary[0].category).update_one(pull__diaries=diary[0])
diary.delete()
return redirect(url_for("admin.diary_list"))
def category_paging(category_id, page_num, category_name=None):
"""Category list page.
show 5 diaries in each page.
Args:
category_id: categoryObjectID
category_name: only for SEO
page_num: page_num
Return:
next_page: bool True or False
page_num: now page_num
category: category_name used for title
diaries: listed 5 diaries in each page
categories: used in sidebar
pages: used for top-nav
profile: user object
"""
next_page = False
diary_num = len(Category.objects(pk=category_id)[0].diaries)
if diary_num > (int(page_num) - 1) * 5 + 5:
next_page = True
profile = User.objects.first()
categories = Category.objects.order_by('-publish_time')
pages = StaticPage.objects.all()
diaries = sorted(Category.objects(pk=category_id)[0].diaries,
key=attrgetter('publish_time'),
reverse=True)[(int(page_num) - 1) * 5 :int(page_num) * 5]
return render_template('frontend/category/list.html',
category=category_name, diaries=diaries,
categories=categories, next_page=next_page,
page_num=page_num, category_id=category_id,
pages=pages, profile=profile)
def diary_edit(diary_id=None):
""" Edit diary from admin
receives title, content(html), tags and cagetory
save title, content(html), pure content(further use), tags and cagetory
also auto save author as current_user.
this method will auto save new Category or Tag if not exist otherwise save
in existed none with push only diary_object
Args:
diary_id: diary_id
title: string
html: string
cagetory: string
tags: list
Save:
title: string
html: string
content: string without html tags
category: string
tags: list
summary: first 80 characters in content with 3 dots in the end
author: current_user_object
"""
if request.method == 'POST' and 'title' and 'content' in request.form:
re_helper = ReHelper()
title = re_helper.r_slash(request.form["title"])
html = request.form["content"]
category = re_helper.r_slash(request.form["category"])
tags = request.form["tags"]
''' save simple data for further use'''
parser = MyHTMLParser()
parser.feed(html)
content = parser.html # the pure content without html tags
splited_tags = tags.split(',')
author = UserModel.objects.first()
try:
diary = Diary.objects(pk=diary_id).first()
except:
diary = Diary(title=title)
old_cat = diary.category
old_tags = diary.tags
diary.title = title
diary.content = content
diary.category = category
diary.summary = content[0:80] + '...'
diary.html = html
diary.author = author
diary.tags = splited_tags
diary.save()
a, cat = Category.objects.get_or_create(name=category,
defaults={'diaries': [diary]})
if not cat:
Category.objects(name=category).update_one(push__diaries=diary)
if old_cat is not None:
Category.objects(name=old_cat).update_one(pull__diaries=diary)
for t in old_tags:
Tag.objects(name=t).update_one(pull__diaries=diary)
for i in splited_tags:
b, tag = Tag.objects.get_or_create(name=i,
defaults={'diaries': [diary]})
if not tag:
Tag.objects(name=i).update_one(push__diaries=diary)
return redirect(url_for("admin.diary_list"))
else:
try:
diary = Diary.objects(pk=diary_id).first()
except:
diary = None
categories = Category.objects.all()
return render_template('admin/diary/edit.html', diary=diary,
categories=categories)
