def add_room(cinema_id, user_id):
form = AddRoomForm()
if form.validate_on_submit():
title = form.roomname.data
content = form.roomcount.data
nm = RoomModel(db.get_connection())
nm.insert(title, content, session['user_id'], cinema_id)
return redirect("/rooms/{}/{}".format(str(cinema_id), str(user_id)))
return render_template('add_room.html',
title='Добавление комнаты',
form=form,
username=session['username'],
cinema_id=cinema_id)
def add_room(self):
if self.type == -1:
nlevel = self.nlevel.text()
nroom = self.nroom.text()
square = self.country.text()
lodger = RoomModel(db.get_connection())
if nlevel in lodger.get_ids(self.adm[7]) or int(nlevel) == 0:
self.change_status(
"Комната с таким номером существует или номер не является натуральным числом"
)
elif nroom == '':
self.change_status("Вы не указали колличество комнат")
elif square == '':
self.change_status("Вы не указали площадь")
else:
lodger.insert(nlevel, nroom, square, self.adm[7])
self.close()
else:
nroom = self.nroom.text()
square = self.country.text()
lodger = RoomModel(db.get_connection())
lodger.change(self.type, nroom, square)
self.close()