def test_get_role_name(self): """Test for getting the role name of a role number.""" role = 1 role_name = get_role_name(role) self.assertEqual(role_name, 'Trainer') role = 4 with self.assertRaises(BaseException): get_role_name(role)
def download_view(): """View to download users. Handle post request via ajax and return file to browser :returns: A csv file containing all users :rtype: HttpResponse """ csv_users = "ID|NAME|WEBSITE|ROLE|LONGITUDE|LATITUDE" all_user_role = (0, 1, 2) i = 0 for user_role in all_user_role: users = get_all_users(user_role) for user in users: i += 1 csv_users += '\n%i|%s|%s|%s|%s|%s' % ( i, user['name'], user['website'], get_role_name(user['role']), user['longitude'], user['latitude']) filename = '%s-users.csv' % APP.config['PROJECT_NAME'] content = "attachment;filename='%s'" % filename return Response( csv_users, mimetype='text/csv', headers={'Content-Disposition': content})
def download_view(): """View to download users. Handle post request via ajax and return file to browser :returns: A csv file containing all users :rtype: HttpResponse """ csv_users = "ID|NAME|WEBSITE|ROLE|LONGITUDE|LATITUDE" all_user_role = (0, 1, 2) i = 0 for user_role in all_user_role: users = get_all_users(user_role) for user in users: i += 1 csv_users += '\n%i|%s|%s|%s|%s|%s' % ( i, user['name'], user['website'], get_role_name( user['role']), user['longitude'], user['latitude']) filename = '%s-users.csv' % APP.config['PROJECT_NAME'] content = "attachment;filename='%s'" % filename return Response(csv_users, mimetype='text/csv', headers={'Content-Disposition': content})