def edit_date_status(photographer_id):
if photographer_id == current_user.id:
user = User.query.filter_by(id=photographer_id).first()
form = DateStatusForm()
if request.method == 'POST':
if request.form.get('action') == 'create':
if form.validate_on_submit():
start_date = form.from_day.data
end_date = form.end_day.data
date_status = DateStatus()
date_status.start_date = start_date
date_status.end_date = end_date
date_status.user_id = photographer_id
date_status.status = EnumValues.find_one_by_code(DATE_STATUS_NOT_AVAILABLE)
save_obj_commit(date_status)
flash('工作日历修改成功')
else:
flash('输入错误，请重新选择起始日期')
elif request.form.get('action') == 'delete':
id_to_del = int(request.form.get('id_to_delete'))
delete_by_id(DateStatus, id_to_del)
flash('工作日历中不可用时间段删除成功')
date_statuses = user.date_statuses
return rt('edit_date_status.html', form=form, date_statuses=date_statuses)
else:
flash('您没有权限编辑该用户的工作日历')
return redirect(url_for('index'))
def edit_collection(collection_id):
categories = EnumValues.type_filter(const.PHOTO_CATEGORY_KEY).all()
styles = EnumValues.type_filter(const.PHOTO_STYLE_KEY).all()
photo_collection = PhotoCollection.query.get(collection_id)
if photo_collection.photographer != current_user:
flash("您没有权限编辑本作品集，将返回网站主页")
return redirect(url_for("index"))
form = PhotoCollectionForm(categories, styles)
if request.method == "POST":
if (
request.form.get("photo-collection-to-delete") is not None
and len(request.form.get("photo-collection-to-delete")) != 0
):
id_to_delete = int(request.form.get("photo-collection-to-delete"))
delete_by_id(PhotoCollection, id_to_delete)
flash("作品集删除成功")
return redirect(url_for("my_photos"))
elif request.form.get("photo-works-to-delete") is not None:
works_to_delete = request.form.get("photo-works-to-delete").split(",")
delete_photo_works(works_to_delete)
files = request.files.getlist("photos[]")
if not (len(files) == 1 and files[0].filename == u""):
add_photo_works(files, photo_collection)
else:
save_photo_collection(form, photo_collection)
AppInfo.get_db().session.commit()
return rt(
"edit_collection.html", photo_collection=photo_collection, categories=categories, form=form, styles=styles
)
