def setup_class(cls):
# Example: Load dummy sqlite DB
print ("Runs before any methods in this class")
cls.app = AppFactory.create_app(TestConfig)
# db.init_app(cls.app)
with cls.app.app_context():
db.create_all()
roomPrice = RoomPrice("single", 100, 150)
room = Room(1, 101, 3, "Available", 1)
booking = Booking(1, 1, datetime.strptime('2017-01-02', '%Y-%m-%d').date()
,datetime.strptime('2017-01-10', '%Y-%m-%d').date(), 123123123, 2000)
RoomManager.set_availability_for_booking(datetime.strptime("2017-01-01", '%Y-%m-%d').date(), 1)
UserManager.create_user("*****@*****.**", "asdasd")
user = UserManager.get_user("*****@*****.**")
UserManager.update_details(user, "mr", "test", "05644654")
db.session.add(roomPrice)
db.session.add(room)
db.session.add(booking)
db.session.commit()
def edit_profile():
form = ProfileForm()
if form.validate_on_submit():
user = UserManager.get_user_by_id(form.user_id.data)
if user is None:
abort(404)
AccessManager.check_user_has_permission_on(user, P.VIEW_OTHER_USER,
P.VIEW_USER)
UserManager.update_details(user, form.first_name.data,
form.last_name.data,
form.contact_number.data)
return redirect(url_for('profile', id=user.id))
else:
user_id = request.args.get('id')
if user_id is not None:
user = UserManager.get_user_by_id(user_id)
else:
user = g.user
if user is None:
abort(404)
AccessManager.check_user_has_permission_on(user, P.VIEW_OTHER_USER,
P.VIEW_USER)
form.user_id.data = user.id
if not form.is_submitted() and user.details is not None:
form.first_name.data = user.details.first_name
form.last_name.data = user.details.last_name
form.contact_number.data = user.details.contact_number
return render_template('accounts/edit-profile.html', form=form)