Esempio n. 1
0
    def get_attending(self, request):
        """
        Get Conferences the calling user is attending (i.e. registered for)
        :param request:
        :return: ConferenceForms
        """
        if not isinstance(request, message_types.VoidMessage):
            raise endpoints.BadRequestException()

        prof = ProfileApi.profile_from_user()  # get user Profile
        conf_keys = prof.conferencesToAttend  # Changed from original code

        if len(conf_keys) == 0:
            # user hasn't registered for anything, so bail out of this method
            return ConferenceForms()

        conferences = ndb.get_multi(conf_keys)

        # get organizers
        organisers = [ndb.Key(Profile, conf.organizerUserId) for conf in
                      conferences]
        profiles = ndb.get_multi(organisers)

        # put display names in a dict for easier fetching
        names = {}
        for profile in profiles:
            names[profile.key.id()] = profile.displayName

        # return set of ConferenceForm objects per Conference
        return ConferenceForms(items=[conf.to_form(names[conf.organizerUserId])
                                      for conf in conferences]
                               )
Esempio n. 2
0
    def _wishlist(self, request, add=True):
        """
        Transaction to add or remove Session from a ConferenceWishlist given by
         a WishlistRequest
        :param request: Wishlist RPC Request [VoidMessage, session key in query
         string]
        :param add: whether to add (True) to the wishlist or remove from a
        wishlist (False)
        :return: BooleanMessage - True if successful, False if failure
        """
        prof = ProfileApi.profile_from_user()  # get user Profile
        session = get_from_webkey(request.websafeSessionKey)  # get session

        if not session or not isinstance(session, Session):
            raise endpoints.NotFoundException('Not a valid session')

        # see if the wishlist exists
        wishlist = ConferenceWishlist().query(ancestor=prof.key) \
            .filter(ConferenceWishlist.conferenceKey == session.conferenceKey) \
            .get()

        # User requested to add to the wishlist, so create if needed
        if not wishlist:
            if add:
                # need to create the wishlist first
                conf_key = session.key.parent()
                wishlist = ConferenceWishlist(conferenceKey=conf_key,
                                              parent=prof.key)
            else:
                # remove request, but no wishlist!
                raise endpoints.NotFoundException(
                    'Nothing wishlisted for Conference')

        # update wishlist by adding/removing session key
        s_key = session.key
        if s_key not in wishlist.sessionKeys:
            if add:
                # add the key
                wishlist.sessionKeys.append(s_key)
                wishlist.put()
            else:
                # can't remove a nonexistant key
                raise endpoints.NotFoundException(
                    'Session not in wishlist for Conference')
        else:
            # key already exists
            if add:
                # session already wishlisted, so return error
                return BooleanMessage(data=False)
            else:
                # remove key from wishlist
                wishlist.sessionKeys.remove(s_key)

                # if wishlist is empty, remove the wishlist. else just update.
                if len(wishlist.sessionKeys) == 0:
                    wishlist.key.delete()
                else:
                    wishlist.put()

        return BooleanMessage(data=True)
Esempio n. 3
0
    def _wishlist(self, request, add=True):
        """
        Transaction to add or remove Session from a ConferenceWishlist given by
         a WishlistRequest
        :param request: Wishlist RPC Request [VoidMessage, session key in query
         string]
        :param add: whether to add (True) to the wishlist or remove from a
        wishlist (False)
        :return: BooleanMessage - True if successful, False if failure
        """
        prof = ProfileApi.profile_from_user()  # get user Profile
        session = get_from_webkey(request.websafeSessionKey)  # get session

        if not session or not isinstance(session, Session):
            raise endpoints.NotFoundException('Not a valid session')

        # see if the wishlist exists
        wishlist = ConferenceWishlist().query(ancestor=prof.key) \
            .filter(ConferenceWishlist.conferenceKey == session.conferenceKey) \
            .get()

        # User requested to add to the wishlist, so create if needed
        if not wishlist:
            if add:
                # need to create the wishlist first
                conf_key = session.key.parent()
                wishlist = ConferenceWishlist(conferenceKey=conf_key,
                                              parent=prof.key)
            else:
                # remove request, but no wishlist!
                raise endpoints.NotFoundException(
                    'Nothing wishlisted for Conference')

        # update wishlist by adding/removing session key
        s_key = session.key
        if s_key not in wishlist.sessionKeys:
            if add:
                # add the key
                wishlist.sessionKeys.append(s_key)
                wishlist.put()
            else:
                # can't remove a nonexistant key
                raise endpoints.NotFoundException(
                    'Session not in wishlist for Conference')
        else:
            # key already exists
            if add:
                # session already wishlisted, so return error
                return BooleanMessage(data=False)
            else:
                # remove key from wishlist
                wishlist.sessionKeys.remove(s_key)

                # if wishlist is empty, remove the wishlist. else just update.
                if len(wishlist.sessionKeys) == 0:
                    wishlist.key.delete()
                else:
                    wishlist.put()

        return BooleanMessage(data=True)
Esempio n. 4
0
    def _register(request, reg=True):
        """
        Register or unregister user for selected conference.
        :param request: RPC Message Request with a urlsafe Conference Key
        :param reg: whether to register (True) or unregister (False) the
        requesting User
        :return: BooleanMessage - True if successful, False if failure
        """
        prof = ProfileApi.profile_from_user()  # get user Profile

        # check if conf exists given websafeConfKey
        # get conference; check that it exists
        c_key = ndb.Key(urlsafe=request.websafeConferenceKey)
        conf = c_key.get()
        if not conf:
            raise endpoints.NotFoundException('No conference found for key')

        # register
        if reg:
            # check if user already registered otherwise add
            if c_key in prof.conferencesToAttend:
                raise ConflictException(
                    'Already registered for this conference')

            # check if seats avail
            if conf.seatsAvailable <= 0:
                raise ConflictException('There are no seats available.')

            # register user, take away one seat
            prof.conferencesToAttend.append(c_key)
            conf.seatsAvailable -= 1

            # update datastore
            prof.put()
            conf.put()

        # un-register
        else:
            # check if user already registered
            if c_key in prof.conferencesToAttend:
                # unregister user, add back one seat
                prof.conferencesToAttend.remove(c_key)
                conf.seatsAvailable += 1

                # update datastore
                prof.put()
                conf.put()
            else:
                return BooleanMessage(data=False)

        return BooleanMessage(data=True)
Esempio n. 5
0
    def get_wishlists(self, request):
        """
        Endpoint for retrieving all wishlists for a requesting user
        :param request:
        :return:
        """
        if not isinstance(request, VoidMessage):
            raise endpoints.BadRequestException()

        prof = ProfileApi.profile_from_user()  # get user Profile
        wishlists = ConferenceWishlist.query(ancestor=prof.key).fetch()

        return WishlistForms(
            items=[wishlist.to_form() for wishlist in wishlists])
Esempio n. 6
0
    def get_wishlists(self, request):
        """
        Endpoint for retrieving all wishlists for a requesting user
        :param request:
        :return:
        """
        if not isinstance(request, VoidMessage):
            raise endpoints.BadRequestException()

        prof = ProfileApi.profile_from_user()  # get user Profile
        wishlists = ConferenceWishlist.query(ancestor=prof.key).fetch()

        return WishlistForms(
            items=[wishlist.to_form() for wishlist in wishlists]
        )
Esempio n. 7
0
    def get_wishlist(self, request):
        """
        Gets the list of sessions wishlisted by a User given a Conference.
        :param request: Conference GET Request [VoidMessage, conference key in
        query string]
        :return: SessionForms
        """
        prof = ProfileApi.profile_from_user()  # get user Profile
        conf_key = ndb.Key(urlsafe=request.websafeConferenceKey)

        wishlist = ConferenceWishlist().query(ancestor=prof.key) \
            .filter(ConferenceWishlist.conferenceKey == conf_key) \
            .get()

        if not wishlist:
            error = 'No wishlist for conference with key %s' % \
                    request.websafeConferenceKey
            raise endpoints.NotFoundException(error)

        sessions = ndb.get_multi(wishlist.sessionKeys)

        return SessionForms(items=SessionApi.populate_forms(sessions))