You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

68 satır
2.2KB

  1. # @name: book.py
  2. # @version: 0.1
  3. # @creation_date: 2021-11-03
  4. # @license: The MIT License <https://opensource.org/licenses/MIT>
  5. # @author: Simon Bowie <ad7588@coventry.ac.uk>
  6. # @purpose: book route for book-related functions and pages
  7. # @acknowledgements:
  8. # https://www.digitalocean.com/community/tutorials/how-to-make-a-web-application-using-flask-in-python-3
  9. from flask import Blueprint, render_template, request, flash, redirect, url_for
  10. from flask_login import login_required, current_user
  11. from .models import Book
  12. from werkzeug.exceptions import abort
  13. from . import db
  14. book = Blueprint('book', __name__)
  15. # function to retrieve data about a single book from the database
  16. def get_book(book_id):
  17. book = Book.query.filter_by(id=book_id).first()
  18. if book is None:
  19. abort(404)
  20. return book
  21. # route for displaying all books in database
  22. @book.route('/books')
  23. def get_books():
  24. books = Book.query
  25. return render_template('books.html', books=books)
  26. # route for displaying a single book based on the ID in the database
  27. @book.route('/books/<int:book_id>')
  28. def show_book(book_id):
  29. book = get_book(book_id)
  30. return render_template('book.html', book=book)
  31. # route for editing a single book based on the ID in the database
  32. @book.route('/books/<int:book_id>/edit', methods=('GET', 'POST'))
  33. @login_required
  34. def edit_book(book_id):
  35. book = get_book(book_id)
  36. if request.method == 'POST':
  37. name = request.form['name']
  38. description = request.form['description']
  39. if not name:
  40. flash('Name is required!')
  41. else:
  42. book = Book.query.get(book_id)
  43. book.name = name
  44. book.description = description
  45. db.session.commit()
  46. return redirect(url_for('book.get_books'))
  47. return render_template('edit.html', book=book)
  48. # route for function to delete a single book from the edit page
  49. @book.route('/books/<int:book_id>/delete', methods=('POST',))
  50. @login_required
  51. def delete_book(book_id):
  52. book = get_book(book_id)
  53. deletion = Book.query.get(book_id)
  54. db.session.delete(deletion)
  55. db.session.commit()
  56. flash('Successfully deleted!')
  57. return redirect(url_for('book.get_books'))