def edit(self, bookmark_id):
if not users.get_current_user():
webapp2.abort(401)
bookmark = db.get(db.Key.from_path('Bookmark', int(bookmark_id)))
form = BookmarkForm(self.get_locale(), self.request.POST, bookmark)
if self.request.POST and form.validate():
form.populate_obj(bookmark)
bookmark.date = datetime.now()
bookmark.put()
return self.redirect('/bookmark/list')
self.render_template('form.html', {'title': _('Bookmark'), 'form': form, 'name': 'bookmark', 'id': bookmark.key().id()})
def add_bookmark():
if not session.get('logged_in'):
abort(401)
form = BookmarkForm(request.form)
if request.method == 'POST' and form.validate():
if form.url.data and form.description.data:
author = User.query.filter_by(username=session.get('username')).first()
bookmark = Bookmark(form.url.data, form.description.data, datetime.utcnow(), author)
db.session.add(bookmark)
db.session.commit()
flash('New bookmark saved successfully')
else:
flash('URL is required.')
return redirect(url_for('show_bookmarks'))
def create(self, parent_id=None):
if not users.get_current_user():
webapp2.abort(401)
form = BookmarkForm(self.get_locale(), self.request.POST, None)
if self.request.POST and form.validate():
parent = None
if form.reference_id.data:
parent = db.get(db.Key.from_path('Bookmark', int(form.reference_id.data)))
bookmark = Bookmark(author=users.get_current_user(), reference=parent, title=form.title.data, url=form.url.data)
bookmark.date = datetime.now()
bookmark.put()
return self.redirect('/bookmark/list')
else:
if parent_id:
form.reference_id.data = parent_id
self.render_template('form.html', {'title': _('Bookmark'), 'form': form, 'name': 'bookmark'})
