def show_categories(page=1):
per_page = request.args.get('per_page', 10, type=int)
# 获取当前官网
current_site = Site.current_site(master_uid=Master.master_uid())
builder = Category.query.filter_by(site_id=current_site.id)
total_count = builder.count()
categories = Category.always_category(
site_id=current_site.id,
language_id=current_site.default_language,
path=0,
page=1,
per_page=per_page)
paginated_categories = []
for cate in categories:
category = {
'id': cate.id,
'name': cate.name,
'icon': Asset.query.get(cate.icon_id) if cate.icon_id else None,
'sort_order': cate.sort_order,
'status': cate.status
}
paginated_categories.append(category)
pagination = Pagination(query=None,
page=1,
per_page=per_page,
total=total_count,
items=None)
return render_template('admin/categories/show_list.html',
paginated_categories=paginated_categories,
pagination=pagination)
def edit_category(id):
# 获取当前官网
current_site = Site.current_site(master_uid=Master.master_uid())
site_languages = current_site.support_languages
category = Category.query.get_or_404(id)
form = CategoryForm()
if form.validate_on_submit():
category.top = form.top.data
category.parent_id = form.parent_id.data
category.icon_id = form.icon.data
category.cover_id = form.cover.data
category.sort_order = form.sort_order.data
category.status = form.status.data
# update category description
for cate_desc in category.category_descriptions:
lang_id = cate_desc.language_id
cate_desc.name = request.form.get(
'category_description[%d][name]' % lang_id)
cate_desc.description = request.form.get(
'category_description[%d][description]' % lang_id)
db.session.commit()
# rebuild category path
Category.repair_categories(category.parent_id)
flash(gettext('Edit category is ok!'), 'success')
return redirect(url_for('.show_categories'))
mode = 'edit'
form.top.data = category.top
form.parent_id.data = category.parent_id
form.icon.data = category.icon_id
form.cover.data = category.cover_id
form.sort_order.data = category.sort_order
form.status.data = category.status
paginated_categories = Category.always_category(
site_id=current_site.id,
language_id=current_site.default_language,
path=0,
page=1,
per_page=100)
return render_template('admin/categories/create_and_edit.html',
form=form,
category=category,
mode=mode,
languages=site_languages,
all_descriptions=category.all_descriptions(),
paginated_categories=paginated_categories)
def create_category():
# 获取当前官网
current_site = Site.current_site(master_uid=Master.master_uid())
site_languages = current_site.support_languages
form = CategoryForm()
if form.validate_on_submit():
# if parent_id有值，则不是一级分类
if form.parent_id.data:
form.top.data = False
# category description ImmutableMultiDict([('csrf_token', 'IjQ2OGZhZjg5NjM0Jk'),
# ('category_description[1][name]', '数码'), ('category_description[1][description]', '电子'),
# ('category_description[2][name]', 'shuma'), ('category_description[2][description]', 'dianzi'),
# ('category_description[3][name]', ''), ('category_description[3][description]', ''),
# ('parent_id', '0'), ('top', ''), ('icon', '0'), ('cover', '0'), ('sort_order', '1')])
category_description_list = []
for lang in site_languages:
cate_name = request.form.get('category_description[%d][name]' %
lang.id)
if cate_name is None or cate_name == '':
flash(gettext("Category name can't empty!"), 'danger')
return redirect(url_for('.create_category'))
cate = {}
cate['language_id'] = lang.id
cate['name'] = cate_name
cate['description'] = request.form.get(
'category_description[%d][description]' % lang.id)
category_description_list.append(cate)
# save category
category = Category(master_uid=Master.master_uid(),
site_id=current_site.id,
top=form.top.data,
parent_id=form.parent_id.data,
icon_id=form.icon.data,
cover_id=form.cover.data,
sort_order=form.sort_order.data,
status=form.status.data)
db.session.add(category)
# save category description
for cate_dict in category_description_list:
cate_desc = CategoryDescription(category=category, **cate_dict)
db.session.add(cate_desc)
db.session.commit()
# rebuild category path
Category.repair_categories(category.parent_id)
flash(gettext('Add category is ok!'), 'success')
return redirect(url_for('.show_categories'))
mode = 'create'
category = None
paginated_categories = Category.always_category(
site_id=current_site.id,
language_id=current_site.default_language,
path=0,
page=1,
per_page=100)
return render_template('admin/categories/create_and_edit.html',
form=form,
category=category,
mode=mode,
languages=site_languages,
all_descriptions=None,
paginated_categories=paginated_categories)
