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)