예제 #1
0
def oauth_make_token(user, client, scope, user_session=None):
    # Look for an existing token
    if client.confidential:
        token = AuthToken.query.filter_by(user=user, client=client).first()
    elif user_session:
        token = AuthToken.query.filter_by(user_session=user_session,
                                          client=client).first()
    else:
        raise ValueError("user_session not provided")

    # If token exists, add to the existing scope
    if token:
        token.add_scope(scope)
    else:
        # If there's no existing token, create one
        if client.confidential:
            token = AuthToken(user=user,
                              client=client,
                              scope=scope,
                              token_type='bearer')
            token = failsafe_add(db.session, token, user=user, client=client)
        elif user_session:
            token = AuthToken(user_session=user_session,
                              client=client,
                              scope=scope,
                              token_type='bearer')
            token = failsafe_add(db.session,
                                 token,
                                 user_session=user_session,
                                 client=client)
    # TODO: Look up Resources for items in scope; look up their providing clients apps,
    # and notify each client app of this token
    return token
예제 #2
0
def oauth_make_token(user, auth_client, scope, user_session=None):
    # Look for an existing token
    token = auth_client.authtoken_for(user, user_session)

    # If token exists, add to the existing scope
    if token:
        token.add_scope(scope)
    else:
        # If there's no existing token, create one
        if auth_client.confidential:
            token = AuthToken(user=user,
                              auth_client=auth_client,
                              scope=scope,
                              token_type='bearer')
            token = failsafe_add(db.session,
                                 token,
                                 user=user,
                                 auth_client=auth_client)
        elif user_session:
            token = AuthToken(
                user_session=user_session,
                auth_client=auth_client,
                scope=scope,
                token_type='bearer',
            )
            token = failsafe_add(db.session,
                                 token,
                                 user_session=user_session,
                                 auth_client=auth_client)
        else:
            raise ValueError("user_session not provided")
    return token
예제 #3
0
    def test_failsafe_add(self):
        """
        failsafe_add gracefully handles IntegrityError from dupe entries
        """
        d1 = NamedDocument(name=u'add_and_commit_test', title=u"Test")
        d1a = failsafe_add(self.session, d1, name=u'add_and_commit_test')
        self.assertTrue(d1a is d1)  # We got back what we created, so the commit succeeded

        d2 = NamedDocument(name=u'add_and_commit_test', title=u"Test")
        d2a = failsafe_add(self.session, d2, name=u'add_and_commit_test')
        self.assertFalse(d2a is d2)  # This time we got back d1 instead of d2
        self.assertTrue(d2a is d1)
예제 #4
0
    def test_failsafe_add_existing(self):
        """
        failsafe_add doesn't fail if the item is already in the session
        """
        d1 = NamedDocument(name=u'add_and_commit_test', title=u"Test")
        d1a = failsafe_add(self.session, d1, name=u'add_and_commit_test')
        self.assertTrue(d1a is d1)  # We got back what we created, so the commit succeeded

        d2 = NamedDocument(name=u'add_and_commit_test', title=u"Test")
        self.session.add(d2)  # Add to session before going to failsafe_add
        d2a = failsafe_add(self.session, d2, name=u'add_and_commit_test')
        self.assertFalse(d2a is d2)  # This time we got back d1 instead of d2
        self.assertTrue(d2a is d1)
예제 #5
0
 def test_failsafe_add_silent_fail(self):
     """
     failsafe_add does not raise IntegrityError with bad data
     when no filters are provided
     """
     d1 = NamedDocument(name=u'missing_title')
     self.assertIsNone(failsafe_add(self.session, d1))
예제 #6
0
 def load_user(self, userid, uuid=None, create=False):
     # TODO: How do we cache this? Connect to a cache manager
     if uuid and hasattr(self.usermodel, '__uuid_primary_key__') and self.usermodel.__uuid_primary_key__:
         user = self.usermodel.query.get(uuid)  # This loads from SQLAlchemy session cache
     elif hasattr(self.usermodel, 'get'):
         user = self.usermodel.get(userid=userid, defercols=False)
     else:
         user = self.usermodel.query.filter_by(
             userid=userid).options(undefer('userinfo')).one_or_none()
     if user is None:
         if create:
             user = self.usermodel(userid=userid)
             if uuid and hasattr(self.usermodel, '__uuid_primary_key__') and self.usermodel.__uuid_primary_key__:
                 user.id = uuid
             failsafe_add(self.db.session, user, userid=userid)
     return user
예제 #7
0
파일: helper.py 프로젝트: saurabh17a/hasjob
def save_jobview(event_session_id, jobpost_id, bgroup, viewed_time):
    """
    Save a jobpost view as a background job to ensure this doesn't fire *before* the
    associated save_impressions job. Queue order is important for the cointoss flag
    in A/B testing.
    """
    jvs = JobViewSession.get_by_ids(event_session_id=event_session_id,
                                    jobpost_id=jobpost_id)
    if jvs is None:
        jvs = JobViewSession(
            event_session_id=event_session_id,
            jobpost_id=jobpost_id,
            datetime=viewed_time,
            bgroup=bgroup,
        )
        jvs = failsafe_add(db.session,
                           jvs,
                           event_session_id=event_session_id,
                           jobpost_id=jobpost_id)

        # Since this is a new view, is there an existing job impression in the same session
        # which has a bgroup defined? If yes, this view has an associated coin toss.
        ji = JobImpression.get_by_ids(jobpost_id=jobpost_id,
                                      event_session_id=event_session_id)
        if ji:
            jvs.cointoss = True
            if ji.bgroup != jvs.bgroup and jvs.bgroup is not None:
                jvs.crosstoss = True
        else:
            jvs.cointoss = False

        db.session.commit()
예제 #8
0
파일: tag.py 프로젝트: Praseetha-KR/hasjob
 def get(cls, tag, create=False):
     title = tag[:cls.__name_length__]
     name = make_name(title)
     ob = cls.query.filter_by(name=name).one_or_none()
     if create and not ob:
         ob = cls(name=name, title=title)
         ob = failsafe_add(db.session, ob, name=name)
     return ob
예제 #9
0
파일: tag.py 프로젝트: thuannvn/hasjob
 def get(cls, tag, create=False):
     title = tag[:cls.__name_length__]
     name = make_name(title)
     ob = cls.query.filter_by(name=name).one_or_none()
     if create and not ob:
         ob = cls(name=name, title=title)
         ob = failsafe_add(db.session, ob, name=name)
     return ob
예제 #10
0
 def add_email(self, email, primary=False, type=None, private=False):
     useremail = UserEmail(user=self,
                           email=email,
                           type=type,
                           private=private)
     useremail = failsafe_add(db.session, useremail, user=self, email=email)
     if primary:
         self.primary_email = useremail
     return useremail
예제 #11
0
파일: oauth.py 프로젝트: gonrin/lastuser
def oauth_make_token(user, client, scope, user_session=None):
    if client.confidential:
        token = AuthToken.query.filter_by(user=user, client=client).first()
    elif user_session:
        token = AuthToken.query.filter_by(user_session=user_session, client=client).first()
    if token:
        token.add_scope(scope)
    else:
        if client.confidential:
            token = AuthToken(user=user, client=client, scope=scope, token_type='bearer')
            token = failsafe_add(db.session, token, user=user, client=client)
        elif user_session:
            token = AuthToken(user_session=user_session, client=client, scope=scope, token_type='bearer')
            token = failsafe_add(db.session, token, user_session=user_session, client=client)
        else:
            raise ValueError("user_session not provided")
    # TODO: Look up Resources for items in scope; look up their providing clients apps,
    # and notify each client app of this token
    return token
예제 #12
0
def register_internal(username, fullname, password):
    user = User(username=username, fullname=fullname, password=password)
    if not username:
        user.username = None
    if user.username:
        # We can only use failsafe_add when a unique identifier like username is present
        user = failsafe_add(db.session, user, username=user.username)
    else:
        db.session.add(user)
    user_registered.send(user)
    return user
예제 #13
0
파일: helpers.py 프로젝트: sindhus/lastuser
def register_internal(username, fullname, password):
    user = User(username=username, fullname=fullname, password=password)
    if not username:
        user.username = None
    if user.username:
        # We can only use failsafe_add when a unique identifier like username is present
        user = failsafe_add(db.session, user, username=user.username)
    else:
        db.session.add(user)
    user_registered.send(user)
    return user
예제 #14
0
파일: user.py 프로젝트: sindhus/lastuser
 def add_email(self, email, primary=False, type=None, private=False):
     if primary:
         for emailob in self.emails:
             if emailob.primary:
                 emailob.primary = False
     useremail = UserEmail(user=self, email=email, primary=primary, type=type, private=private)
     useremail = failsafe_add(db.session, useremail, user=self, email=email)
     with db.session.no_autoflush:
         for team in Team.query.filter_by(domain=useremail.domain):
             if self not in team.users:
                 team.users.append(self)
     return useremail
예제 #15
0
def session_form(space, proposal=None, session=None):
    if session:
        form = SessionForm(obj=session, model=Session)
    else:
        form = SessionForm()
    form.venue_room_id.choices = rooms_list(space)
    if request.method == 'GET':
        if not (session or proposal):
            form.is_break.data = True
        if proposal:
            form.description.data = proposal.description
            form.speaker_bio.data = proposal.bio
            form.speaker.data = proposal.owner.fullname
            form.title.data = proposal.title
        return render_template('session_form.html',
                               form=form,
                               formid='session_form')
    if form.validate_on_submit():
        new = False
        if not session:
            new = True
            session = Session()
        if proposal:
            session.proposal = proposal
        form.populate_obj(session)
        if new:
            session.parent = space
            session.make_id()  # FIXME: This should not be required
            session.make_name()
            session = failsafe_add(db.session,
                                   session,
                                   proposal_space_id=space.id,
                                   url_id=session.url_id)
        db.session.commit()
        data = dict(id=session.url_id,
                    title=session.title,
                    room_scoped_name=session.venue_room.scoped_name
                    if session.venue_room else None,
                    is_break=session.is_break,
                    modal_url=session.url_for('edit'),
                    delete_url=session.url_for('delete'),
                    proposal_id=session.proposal_id)
        return jsonify(status=True, data=data)
    return jsonify(status=False,
                   form=render_template('session_form.html',
                                        form=form,
                                        formid='session_new'))
예제 #16
0
파일: session.py 프로젝트: hasgeek/funnel
def session_form(project, proposal=None, session=None):
    # Look for any existing unscheduled session
    if proposal and not session:
        session = Session.for_proposal(proposal)

    if session:
        form = SessionForm(obj=session, model=Session)
    else:
        form = SessionForm()
        if proposal:
            form.description.data = proposal.description
            form.speaker_bio.data = proposal.bio
            form.speaker.data = proposal.owner.fullname
            form.title.data = proposal.title

    form.venue_room_id.choices = rooms_list(project)
    if request.method == 'GET':
        if not (session or proposal):
            form.is_break.data = True
        return render_template('session_form.html.jinja2', form=form, formid='session_form')
    if form.validate_on_submit():
        new = False
        if not session:
            new = True
            session = Session()
        if proposal:
            session.proposal = proposal
        form.populate_obj(session)
        if new:
            session.parent = project
            session = failsafe_add(db.session, session, project_id=project.id, url_id=session.url_id)
        db.session.commit()
        data = dict(
            id=session.url_id, title=session.title, room_scoped_name=session.venue_room.scoped_name if session.venue_room else None,
            is_break=session.is_break, modal_url=session.url_for('edit'), delete_url=session.url_for('delete'),
            proposal_id=session.proposal_id)
        return jsonify(status=True, data=data)
    return jsonify(
        status=False,
        form=render_template('session_form.html.jinja2', form=form, formid='session_new'))
예제 #17
0
def save_jobview(event_session_id, jobpost_id, bgroup, viewed_time):
    """
    Save a jobpost view as a background job to ensure this doesn't fire *before* the
    associated save_impressions job. Queue order is important for the cointoss flag
    in A/B testing.
    """
    jvs = JobViewSession.get_by_ids(event_session_id=event_session_id, jobpost_id=jobpost_id)
    if jvs is None:
        jvs = JobViewSession(event_session_id=event_session_id, jobpost_id=jobpost_id, datetime=viewed_time,
            bgroup=bgroup)
        jvs = failsafe_add(db.session, jvs, event_session_id=event_session_id, jobpost_id=jobpost_id)

        # Since this is a new view, is there an existing job impression in the same session
        # which has a bgroup defined? If yes, this view has an associated coin toss.
        ji = JobImpression.get_by_ids(jobpost_id=jobpost_id, event_session_id=event_session_id)
        if ji:
            jvs.cointoss = True
            if ji.bgroup != jvs.bgroup and jvs.bgroup is not None:
                jvs.crosstoss = True
        else:
            jvs.cointoss = False

        db.session.commit()
예제 #18
0
파일: user.py 프로젝트: hasgeek/lastuser
 def add_email(self, email, primary=False, type=None, private=False):
     useremail = UserEmail(user=self, email=email, type=type, private=private)
     useremail = failsafe_add(db.session, useremail, user=self, email=email)
     if primary:
         self.primary_email = useremail
     return useremail
예제 #19
0
def session_form(project, proposal=None, session=None):
    # Look for any existing unscheduled session
    if proposal and not session:
        session = Session.for_proposal(proposal)

    if session:
        form = SessionForm(obj=session, model=Session)
    else:
        form = SessionForm()
        if proposal:
            form.description.data = proposal.outline
            form.speaker_bio.data = proposal.bio
            form.speaker.data = proposal.owner.fullname
            form.title.data = proposal.title

    form.venue_room_id.choices = rooms_list(project)
    if not form.venue_room_id.choices:
        del form.venue_room_id
    if request.method == 'GET':
        return render_template(
            'session_form.html.jinja2',
            form=form,
            formid='session_form',
            title=_("Edit session"),
        )
    if form.validate_on_submit():
        new = False
        if not session:
            new = True
            session = Session()
        if proposal:
            session.proposal = proposal
        form.populate_obj(session)
        if new:
            session.parent = project
            if session.proposal:
                session = failsafe_add(
                    db.session,
                    session,
                    project_id=project.id,
                    proposal_id=session.proposal_id,
                )
            else:
                db.session.add(session)
        db.session.commit()
        if request_is_xhr():
            data = {
                'id':
                session.url_id,
                'title':
                session.title,
                'room_scoped_name': (session.venue_room.scoped_name
                                     if session.venue_room else None),
                'is_break':
                session.is_break,
                'modal_url':
                session.url_for('edit'),
                'delete_url':
                session.url_for('delete'),
                'proposal_id':
                session.proposal_id,  # FIXME: Switch to UUID
            }
            return jsonify(status=True, data=data)
        else:
            return redirect(session.url_for('view'))
    return jsonify(
        status=False,
        form=render_template(
            'session_form.html.jinja2',
            form=form,
            formid='session_new',
            title=_("Edit session"),
        ),
    )