def add_staff():
form = RegisterFormStaff()
if form.validate_on_submit():
role = int(form.role.data)
role = list(RoleEnum)[role]
UserManager.create_user(form.email.data, form.password.data, role)
flash(form.email.data + ' Has been added to users with role: ' +
form.role.data)
return render_template('staff/add_staff.html', form=form)
def register():
form = RegisterForm()
if form.validate_on_submit():
if not UserManager.create_user(form.email.data, form.password.data):
flash('This email is already in use.', 'danger')
else:
LoginManager.login(form.email.data, form.password.data)
return redirect(url_for('edit_profile'))
return render_template('accounts/register.html', form=form)
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 post(self):
'''
Create a new user
'''
data = request.get_json()
data['password_confirmation'] = data['password']
form = RegisterForm(data=data, csrf_enabled=False)
if form.validate():
if not UserManager.create_user(form.email.data, form.password.data):
return {'form_errors': 'This email is already in use.'}, 400
else:
return {'email': form.email.data}, 201 # TODO: return a dictionary pointing to the newly created resource
# {'Location': url_for('get_user', id = user.id, _external = True)} # Suggest location of next request
else:
return {'form_field_errors': form.errors}, 400