def add_book():
form = BookForm()
if form.validate_on_submit():
db_sess = db_session.create_session()
book = Books()
book.title = form.title.data
book.pic_url = form.pic_url.data
book.author = form.author.data
book.content = form.content.data
book.file_name = form.file.data.filename
book.marked_file_name = f"marked_{form.file.data.filename}"
form.file.data.save(os.path.join(
app.auto_find_instance_path(), form.file.data.filename))
current_user.books.append(book)
db_sess.merge(current_user)
db_sess.commit()
book_load(os.path.join(
app.auto_find_instance_path(), form.file.data.filename), f'{book.file_name}')
path = os.path.join(
app.auto_find_instance_path(), form.file.data.filename)
path2 = os.path.join(
app.auto_find_instance_path(), f'{book.marked_file_name}.csv')
if marking(book.file_name, path, path2):
book_load(os.path.join(
app.auto_find_instance_path(), f'{book.marked_file_name}.csv'), f'{book.marked_file_name}.csv')
os.remove(os.path.join(
app.auto_find_instance_path(), book.file_name))
os.remove(os.path.join(
app.auto_find_instance_path(), f'{book.marked_file_name}.csv'))
disk.publish(f"/book/{book.file_name}")
disk.publish(f"/book/{book.marked_file_name}.csv")
return redirect('/')
return render_template('books.html', file_label='Файл книги (.txt)', title='Добавление книги',
form=form)
def add_book():
form = BookForm()
if form.validate_on_submit():
db_sess = db_session.create_session()
book = Book()
book.title = form.title.data
book.author = form.author.data
genre = db_sess.query(Genre).filter(Genre.title == form.genre.data).first()
if not genre:
new_genre = Genre(title=form.genre.data)
db_sess.add(new_genre)
db_sess.commit()
genre = db_sess.query(Genre).filter(Genre.title == form.genre.data).first()
book.genre = genre
book.created_date = form.created_date.data
book.annotation = form.annotation.data
book.img_file = form.img_file.data.filename
print(form.img_file.data)
print(type(form.img_file.data))
form.img_file.data.save(f'static/img/{form.img_file.data.filename}')
book.text_file = form.text_file.data.filename
form.text_file.data.save(f'static/text/{form.text_file.data.filename}')
genre.book.append(book)
db_sess.merge(genre)
db_sess.commit()
return redirect('/')
return render_template('book.html', title='Добавление книги', form=form)
def add_book():
# Метод для добавления новой книги.
# Информация о коллекция выбирается автоматически из коллекций, принадлежащих пользователю.
# Из-за того, что в RadioField находится только индекс выбранного элемента, выбор коллекции производится вручную
form = BookForm()
db_sess = db_session.create_session()
if current_user.is_authenticated:
collections = db_sess.query(Collection).filter((Collection.user == current_user))
else:
collections = []
index = form.collection_id.data
if request.method == "POST" and form.validate_on_submit():
db_sess = db_session.create_session()
book = Book()
book.title = form.title.data
book.author = form.author.data
book.opinion = form.opinion.data
book.year = form.year.data
book.is_private = form.is_private.data
if index is not None:
book.collection_id = collections[int(index) - 1].id
current_user.books.append(book)
db_sess.merge(current_user)
db_sess.commit()
return redirect('/books')
choices = []
for item in collections:
choices.append((item.id, item.name))
form.collection_id.choices = choices
return render_template('add_book.html', title='Добавление книги', form=form, collections=collections.count())
def edit_book(id):
form = BookForm()
if request.method == "GET":
db_sess = db_session.create_session()
book = db_sess.query(Books).filter(Books.id == id,
Books.user == current_user
).first()
if book:
form.title.data = book.title
form.author.data = book.author
form.pic_url.data = book.pic_url
form.content.data = book.content
else:
abort(404)
if form.validate_on_submit():
db_sess = db_session.create_session()
book = db_sess.query(Books).filter(Books.id == id,
Books.user == current_user
).first()
if book:
book.title = form.title.data
book.author = form.author.data
book.pic_url = form.pic_url.data
book.content = form.content.data
if book.file_name != form.file.data.filename:
disk.remove(f"/book/{book.file_name}")
disk.remove(f"/book/{book.marked_file_name}.csv")
book.file_name = form.file.data.filename
book.marked_file_name = f"marked_{form.file.data.filename}"
form.file.data.save(os.path.join(
app.auto_find_instance_path(), form.file.data.filename))
book_load(os.path.join(
app.auto_find_instance_path(), form.file.data.filename), f'{book.file_name}')
path = os.path.join(
app.auto_find_instance_path(), form.file.data.filename)
path2 = os.path.join(
app.auto_find_instance_path(), f'{book.marked_file_name}.csv')
if marking(book.file_name, path, path2):
book_load(os.path.join(
app.auto_find_instance_path(), f'{book.marked_file_name}.csv'), f'{book.marked_file_name}.csv')
os.remove(os.path.join(
app.auto_find_instance_path(), book.file_name))
os.remove(os.path.join(
app.auto_find_instance_path(), f'{book.marked_file_name}.csv'))
disk.publish(f"/book/{book.file_name}")
disk.publish(f"/book/{book.marked_file_name}.csv")
db_sess.commit()
return redirect('/')
else:
abort(404)
return render_template('books.html',
title='Редактирование новости',
form=form
)
def edit_books(id):
# Метод для редактирования книги.
# Информация о коллекция выбирается автоматически из коллекций, принадлежащих пользователю.
# Из-за того, что в RadioField находится только индекс выбранного элемента, выбор коллекции производится вручную
db_sess = db_session.create_session()
if current_user.is_authenticated:
collections = db_sess.query(Collection).filter((Collection.user == current_user))
else:
collections = []
form = BookForm()
if request.method == "GET":
db_sess = db_session.create_session()
book = db_sess.query(Book).filter(Book.id == id,
Book.user == current_user).first()
choices = []
for item in collections:
choices.append((item.id, item.name))
form.collection_id.choices = choices
if book:
form.title.data = book.title
form.author.data = book.author
form.opinion.data = book.opinion
form.year.data = book.year
if book.collection_id is not None:
index = 0
for col in collections:
index += 1
if col.id == book.collection_id:
form.collection_id.data = str(index)
form.is_private.data = book.is_private
else:
abort(404)
if request.method == "POST" and form.validate_on_submit():
db_sess = db_session.create_session()
book = db_sess.query(Book).filter(Book.id == id,
Book.user == current_user).first()
index = form.collection_id.data
if book:
book.title = form.title.data
book.author = form.author.data
book.opinion = form.opinion.data
book.year = form.year.data
book.is_private = form.is_private.data
if index is not None:
book.collection_id = collections[int(index) - 1].id
db_sess.commit()
return redirect('/books')
else:
abort(404)
return render_template('edit_book.html', title='Редактирование книги', form=form,
collections=collections.count())
def edit_book(book_id):
form = BookForm()
if request.method == "GET":
db_sess = db_session.create_session()
book = db_sess.query(Book).filter(Book.id == book_id).first()
if book:
form.title.data = book.title
form.author.data = book.author
genre = db_sess.query(Genre).filter(Genre.id == book.genre_id).first()
form.genre.data = genre.title
form.created_date.data = book.created_date
form.annotation.data = book.annotation
else:
abort(404)
if form.validate_on_submit():
db_sess = db_session.create_session()
book = db_sess.query(Book).filter(Book.id == book_id).first()
if book:
book.title = form.title.data
book.author = form.author.data
genre = db_sess.query(Genre).filter(Genre.title == form.genre.data).first()
if not genre:
new_genre = Genre(title=form.genre.data)
db_sess.add(new_genre)
db_sess.commit()
genre = db_sess.query(Genre).filter(Genre.title == form.genre.data).first()
book.genre_id = genre.id
book.created_date = form.created_date.data
book.annotation = form.annotation.data
book.img_file = form.img_file.data.filename
form.img_file.data.save(f'static/img/{form.img_file.data.filename}')
book.text_file = form.text_file.data.filename
form.text_file.data.save(f'static/text/{form.text_file.data.filename}')
db_sess.commit()
return redirect('/')
else:
abort(404)
return render_template('book.html',
title='Редактирование книги',
form=form
)
def add_book():
form = BookForm()
session = db_session.create_session()
form.categories.choices = [
(category.id, category.name)
for category in session.query(Category).order_by(Category.name).all()
]
if form.validate_on_submit():
user = session.query(User).get(current_user.id)
# noinspection PyArgumentList
book = Book(title=form.title.data,
description=form.description.data,
price=form.price.data,
stock=form.stock.data,
release_date=form.release_date.data)
if form.is_user_author:
author = session.query(Author).get(user.author_id)
if not author:
# noinspection PyArgumentList
author = Author(first_name=current_user.first_name,
last_name=current_user.last_name)
user.author = author
book.author = author
# Save image and/or book file to server if provided
if form.image.data:
image_path = resize_and_save_image(form.image.data.filename)
book.image_path = image_path
if form.content.data:
file_path = save_file(form.content.data.filename)
book.file_path = file_path
session.merge(book)
session.commit()
return redirect('/')
return render_template('add_book.html',
title='Add book',
form=form,
header='Add book')
def add_book():
if request.method == 'GET':
if 'logged_in' not in session:
message_body = 'You are not logged in.'
message_title = 'Error!'
return render_template('message.html',
message_title=message_title,
message_body=message_body)
else:
form = BookForm()
return render_template('add_book.html',
form=form,
error=form.errors)
else:
form = BookForm()
if form.radio.data == 'book':
del form.issue
del form.title_of_magazine
if form.validate_on_submit():
tmp_authors = [[form.first_name.data, form.surname.data],
[form.first_name_1.data, form.surname_1.data],
[form.first_name_2.data, form.surname_2.data],
]
new_authors = []
for first_name, surname in tmp_authors:
if first_name is not '' and surname is not '':
author = Author.query.filter_by(first_name=first_name,
last_name=surname
).first()
if not author:
new_author = Author(
first_name=first_name,
last_name=surname
)
new_authors.append(new_author)
db.session.add(new_author)
db.session.commit()
else:
new_authors.append(author)
tmp_tag = Tag.query.filter_by(name=form.tag.data).first()
if not tmp_tag:
new_tag = Tag(
name=form.tag.data
)
db.session.add(new_tag)
db.session.commit()
else:
new_tag = tmp_tag
new_book = Book(
title=form.title.data,
table_of_contents=form.table_of_contents.data,
language=form.language.data,
category=form.category.data,
tags=[new_tag],
description=form.description.data,
isbn=form.isbn.data,
authors=new_authors,
original_title=form.original_title.data,
publisher=form.publisher.data,
pub_date=datetime(year=int(form.pub_date.data),
month=1,
day=1))
if book_exists(new_book):
message_body = 'This book already exists.'
message_title = 'Oops!'
return render_template('message.html',
message_title=message_title,
message_body=message_body)
db.session.add(new_book)
db.session.commit()
message_body = 'The book has been added.'
message_title = 'Success!'
return render_template('message.html',
message_title=message_title,
message_body=message_body)
return render_template('add_book.html',
form=form,
error=form.errors)
if form.radio.data == 'magazine':
del form.publisher
del form.original_title
del form.isbn
del form.title
if form.validate_on_submit():
tmp_tag = Tag.query.filter_by(name=form.tag.data).first()
if not tmp_tag:
new_tag = Tag(
name=form.tag.data
)
db.session.add(new_tag)
db.session.commit()
else:
new_tag = tmp_tag
new_magazine = Magazine(
title=form.title_of_magazine.data,
table_of_contents=form.table_of_contents.data,
language=form.language.data,
category=form.category.data,
tags=[new_tag],
description=form.description.data,
year=datetime(year=int(form.pub_date.data),
month=1,
day=1),
issue=form.issue.data)
db.session.add(new_magazine)
db.session.commit()
message_body = 'The magazine has been added.'
message_title = 'Success!'
return render_template('message.html',
message_title=message_title,
message_body=message_body)
return render_template('add_book.html',
form=form,
error=form.errors)