def setUp(self): self.config = testing.setUp() from sqlalchemy import create_engine engine = create_engine('sqlite://') from consensus.models import ( Base, User, ) DBSession.configure(bind=engine) Base.metadata.create_all(engine) with transaction.manager: role = Role('ROLE_USER', 'The default role for all users.') DBSession.add(role)
def signup(request): try: username = request.POST.getone('username') password = request.POST.getone('password') except KeyError: return HTTPBadRequest() new_user = User(username, str(bkrypt.Password.create(password))) role_user = DBSession.query(Role).filter_by(alias='ROLE_USER').first() try: with transaction.manager: DBSession.add(new_user) user = DBSession.query(User).filter_by(username=username).first() user.roles.append(role_user) except IntegrityError: return HTTPBadRequest() return HTTPFound(location=request.route_url('login'))
def setUp(self): self.config = testing.setUp() self.auth_strategy = AuthenticationStrategy() from sqlalchemy import create_engine engine = create_engine('sqlite://') from consensus.models import ( Base, User, ) DBSession.configure(bind=engine) Base.metadata.create_all(engine) with transaction.manager: model = User('TestUser',str(Password.create('TestPass'))) role = Role('ROLE_USER', 'The default role for all users.') DBSession.add(role) model.roles.append(DBSession.query(Role).filter_by(alias='ROLE_USER').first()) DBSession.add(model)
def create_election(request): if (not is_authenticated(request)): return HTTPUnauthorized() try: election_name = request.POST.getone('name') election_desc = request.POST.getone('body') method = request.POST.getone('method') except KeyError: return HTTPBadRequest() read_method = DBSession.query(Method).filter_by(python_name=method).first() if (read_method is None): return HTTPBadRequest() election = Election(election_name, election_desc, read_method) with transaction.manager: new_id = DBSession.add(election) return HTTPFound(location=request.route_url('view_election', election_id=new_id))