def reservar(): form = ReserveForm(request.form) choices = [] for habitacion in Habitacion.select(): valor = u'Habitacion: {}, Hotel: {}, Precioxdia: {}'.format(habitacion.numeroHabitacion,habitacion.hotel.nombre,habitacion.tipoHabitacion.precio) choices.insert(len(choices),(habitacion.id,valor)) form.habitacion.choices = choices if form.validate_on_submit(): try: reserva = Reserva.get(Reserva.fechaPedido == form.fechaPedido.data,Reserva.habitacion == form.habitacion.data) if reserva: return render_template('reserva.html', form=form, error=u'reserva ya existente en sistema') except: fechaTermino = form.fechaPedido.data + timedelta(days=int(form.numeroDias.data)) Reserva.create( fechaPedido=form.fechaPedido.data, fechaTermino=fechaTermino, cliente=current_user.id, habitacion=form.habitacion.data ) return redirect(url_for('index')) try: return redirect(url_for('index')) except: return render_template('reserva.html', form=form, error=u'Error') return render_template('reserva.html', form=form)
def admin_habitaciones(): if not current_user.is_admin(): return redirect(url_for('index')) form = CreateHabitacionForm(request.form) tipos_habitacion_choices = [] for tipo_habitacion in TipoHabitacion.select(): valor = u'Tipo: {},Numero Personas: {},Precio: {}'.format(tipo_habitacion.nombre,tipo_habitacion.numeroPersonas,tipo_habitacion.precio) tipos_habitacion_choices.insert(len(tipos_habitacion_choices),(tipo_habitacion.id,valor)) form.tipoHabitacion.choices = tipos_habitacion_choices hoteles_choices = [] for hotel in Hotel.select(): valor = u'Hotel: {}'.format(hotel.nombre) hoteles_choices.insert(len(hoteles_choices),(hotel.id,valor)) form.hotel.choices = hoteles_choices habitaciones_p = [] for habitacion in Habitacion.select(): numeroHabitacion = habitacion.numeroHabitacion piso = habitacion.piso tipoHabitacion = habitacion.tipoHabitacion.nombre hotel = habitacion.hotel.nombre formd = DeleteHabitacionForm(request.form) habitaciones_p.insert(len(habitaciones_p),{'numeroHabitacion':numeroHabitacion, 'piso':piso,'tipoHabitacion':tipoHabitacion,'hotel':hotel,'form':formd }) if form.validate_on_submit(): try: tipoHabitacion = TipoHabitacion.get(TipoHabitacion.numeroHabitacion == form.numeroHabitacion.data) if tipoHabitacion: return render_template('admin_habitaciones.html',habitaciones=habitaciones_p, form=form, error=u'habitacion ya existente en sistema') except: Habitacion.create( numeroHabitacion=form.numeroHabitacion.data, piso=form.piso.data, tipoHabitacion=form.tipoHabitacion.data, hotel=form.hotel.data ) return redirect(url_for('admin_habitaciones')) return render_template('admin_habitaciones.html',habitaciones=habitaciones_p,form=form)