def create_or_update():
form = DiaryCreateForm()
if not form.validate_on_submit():
return jsonify({'status': 'error', 'form': form.errors}), 400
date = form.date.data
day = Diary.query.filter_by(user=current_user).filter_by(date=date).first()
if day:
day.save_title(form.title.data)
day.save_content(form.content.data)
db.session.commit()
return jsonify({'status': 'update'})
else:
diary = Diary(
user=current_user,
date=datetime.strptime(form.date.data, '%Y-%m-%d'),
)
diary.save_title(form.title.data)
diary.save_content(form.content.data)
db.session.add(diary)
db.session.commit()
return jsonify({'status': 'added'})
