def add_book():
""" Добавление книги """
# Ошибка, если пользователь не администратор или не зарегистрирован
if 'username' not in session:
return render_template('error.html', loged=False, title='Ошибка')
if not UsersModel(db.get_connection()).is_admin(session['username']):
return render_template('error.html', loged=False, title='Ошибка')
form = AddBookForm()
if form.validate_on_submit():
# создать книгу
books = BooksModel(db.get_connection())
f = None
all_books = BooksModel(db.get_connection()).get_all()
for i in range(len(all_books)):
if all_books[i][2] == form.author.data:
f = all_books[i][0]
if f:
# добавить книгу в список произведений автора
author = AuthorsModel(db.get_connection()).get_by_name(
form.author.data)
book_list = author[4] + '//' + form.name.data
AuthorsModel(db.get_connection()).add_book(book_list, author[0])
if form.pictures.data is not None:
filename = secure_filename(form.pictures.data.filename)
server_file = 'static/img/' + filename
form.pictures.data.save(server_file)
books.insert(name=form.name.data,
author=form.author.data,
year=form.year.data,
pages=form.pages.data,
pictures=filename,
bio=form.bio.data,
stock=form.stock.data,
price=form.price.data)
# редирект на главную страницу
return redirect('index')
return render_template("add_book.html",
title='Добавить книгу',
form=form,
loged=True)
def add_book():
if 'username' not in session:
return redirect('login')
if session['username'] != 'admin':
return redirect('index')
form = AddbookForm()
available_library = [(i[0], i[1])
for i in LibraryModel(db.get_connection()).get_all()]
form.library_id.choices = available_library
if form.validate_on_submit():
cars = BooksModel(db.get_connection())
cars.insert(name=form.name.data,
price=form.price.data,
info=form.info.data,
library=form.library_id.data)
return redirect(url_for('book_admin'))
return render_template("add_book.html",
title='Добавление книги',
form=form)
def add_book():
"""
Добавление автомобиля
"""
# если пользователь не авторизован, кидаем его на страницу входа
if 'username' not in session:
return redirect('login')
# если админ, то его на свою страницу
if session['username'] != 'admin':
return redirect('index')
form = AddBookForm()
available_dealers = [(i[0], i[1]) for i in DealersModel(db.get_connection()).get_all()]
form.dealer_id.choices = available_dealers
if form.validate_on_submit():
# создать автомобиль
books = BooksModel(db.get_connection())
books.insert(model=form.model.data,
price=form.price.data,
power=form.power.data,
color=form.color.data,
dealer=form.dealer_id.data)
# редирект на главную страницу
return redirect(url_for('book_admin'))
return render_template("add_book.html", title='Добавление книги', form=form)