def validationRoleID(role_id):
if role_id is None:
raise ValidationException("Role is required field")
elif not isinstance(role_id, int):
raise ValidationException("Role has invalid integer value")
elif not RoleDao.getRoleByID(role_id):
raise NotFoundException("Unable to find user role with given id")
def login_authenticate():
if UserDao.isUserExists(request.form['name'], request.form['password']):
user = UserDao.getUserByLogin(request.form['name'], request.form['password'])
session['login'] = user.login
session['user_id'] = user.id
session['role'] = RoleDao.getRoleByID(user.role_id).name
if session['role'] == 'Customer':
return render_template('my_orders.html',)
if session['role'] == 'Administrator':
return render_template('search_user.html',)
if session['role'] == 'Merchandiser':
return render_template('manage_orders.html',)
if session['role'] == 'Supervisor':
return render_template('product_grid.html',)
else:
error = 'Invalid username/password'
return render_template('login(2).html', error=error)
#!/usr/bin/env python
# ---Fill test data for user and related entities----
from models.region_dao import RegionDao
from models.role_dao import RoleDao
from models.user_dao import UserDao
RoleDao.createNewRole("Admin")
RegionDao.createNewRegion("Crimea")
UserDao.createNewUser('Anton', '111','Maxim','Sidorov', '*****@*****.**', 1,1)
UserDao.createNewUser('Vanya', '111','Ivan','Ivanov', '*****@*****.**', 1,1)
UserDao.createNewUser('Andrew', '111','Andrew','Petrov', '*****@*****.**', 1,1)
UserDao.createNewUser('Katya', '111','Ekaterina','Ivanova', '*****@*****.**', 1,1)
for instance in UserDao.getAllUsers():
print(instance.id,instance.login,instance.password,instance.first_name, instance.last_name, instance.email,
RoleDao.getRoleByID(instance.role_id).name, RegionDao.getRegionByID(instance.region_id).name)
def getRoleByID(role_id):
validationRoleID(role_id)
return RoleDao.getRoleByID(role_id)
