コード例 #1
0
    def GetHostRequest(self, request, context):
        with session_scope() as session:
            host_request = (session.query(HostRequest).filter(
                HostRequest.conversation_id == request.host_request_id).filter(
                    or_(HostRequest.from_user_id == context.user_id,
                        HostRequest.to_user_id ==
                        context.user_id)).one_or_none())

            if not host_request:
                context.abort(grpc.StatusCode.NOT_FOUND,
                              errors.HOST_REQUEST_NOT_FOUND)

            initial_message = (session.query(
                Message.time).filter(Message.conversation_id ==
                                     host_request.conversation_id).order_by(
                                         Message.id.asc()).limit(1).one())

            latest_message = (session.query(Message).filter(
                Message.conversation_id ==
                host_request.conversation_id).order_by(
                    Message.id.desc()).limit(1).one())

            return requests_pb2.HostRequest(
                host_request_id=host_request.conversation_id,
                from_user_id=host_request.from_user_id,
                to_user_id=host_request.to_user_id,
                status=hostrequeststatus2api[host_request.status],
                created=Timestamp_from_datetime(initial_message.time),
                from_date=date_to_api(host_request.from_date),
                to_date=date_to_api(host_request.to_date),
                last_seen_message_id=host_request.from_last_seen_message_id
                if context.user_id == host_request.from_user_id else
                host_request.to_last_seen_message_id,
                latest_message=message_to_pb(latest_message),
            )
コード例 #2
0
ファイル: admin.py プロジェクト: Couchers-org/couchers
def _user_to_details(user):
    return admin_pb2.UserDetails(
        user_id=user.id,
        username=user.username,
        email=user.email,
        gender=user.gender,
        birthdate=date_to_api(user.birthdate),
        banned=user.is_banned,
        deleted=user.is_deleted,
    )
コード例 #3
0
ファイル: requests.py プロジェクト: Couchers-org/couchers
    def GetHostRequest(self, request, context):
        with session_scope() as session:
            host_request = session.execute(
                select(HostRequest).where_users_column_visible(
                    context,
                    HostRequest.surfer_user_id).where_users_column_visible(
                        context, HostRequest.host_user_id).
                where(HostRequest.conversation_id ==
                      request.host_request_id).where(
                          or_(HostRequest.surfer_user_id == context.user_id,
                              HostRequest.host_user_id ==
                              context.user_id))).scalar_one_or_none()

            if not host_request:
                context.abort(grpc.StatusCode.NOT_FOUND,
                              errors.HOST_REQUEST_NOT_FOUND)

            initial_message = session.execute(
                select(Message).where(
                    Message.conversation_id ==
                    host_request.conversation_id).order_by(
                        Message.id.asc()).limit(1)).scalar_one()

            latest_message = session.execute(
                select(Message).where(
                    Message.conversation_id ==
                    host_request.conversation_id).order_by(
                        Message.id.desc()).limit(1)).scalar_one()

            return requests_pb2.HostRequest(
                host_request_id=host_request.conversation_id,
                surfer_user_id=host_request.surfer_user_id,
                host_user_id=host_request.host_user_id,
                status=hostrequeststatus2api[host_request.status],
                created=Timestamp_from_datetime(initial_message.time),
                from_date=date_to_api(host_request.from_date),
                to_date=date_to_api(host_request.to_date),
                last_seen_message_id=host_request.surfer_last_seen_message_id
                if context.user_id == host_request.surfer_user_id else
                host_request.host_last_seen_message_id,
                latest_message=message_to_pb(latest_message),
            )
コード例 #4
0
    def ListHostRequests(self, request, context):
        if request.only_sent and request.only_received:
            context.abort(grpc.StatusCode.INVALID_ARGUMENT,
                          errors.HOST_REQUEST_SENT_OR_RECEIVED)

        with session_scope() as session:
            pagination = request.number if request.number > 0 else DEFAULT_PAGINATION_LENGTH
            pagination = min(pagination, MAX_PAGE_SIZE)

            # By outer joining messages on itself where the second id is bigger, only the highest IDs will have
            # none as message_2.id. So just filter for these ones to get highest messages only.
            # See https://stackoverflow.com/a/27802817/6115336
            message_2 = aliased(Message)
            query = (session.query(
                Message, HostRequest, Conversation).outerjoin(
                    message_2,
                    and_(Message.conversation_id == message_2.conversation_id,
                         Message.id < message_2.id)).join(
                             HostRequest, HostRequest.conversation_id ==
                             Message.conversation_id).join(
                                 Conversation, Conversation.id ==
                                 HostRequest.conversation_id).filter(
                                     message_2.id == None).filter(
                                         or_(
                                             HostRequest.conversation_id <
                                             request.last_request_id,
                                             request.last_request_id == 0)))

            if request.only_sent:
                query = query.filter(
                    HostRequest.from_user_id == context.user_id)
            elif request.only_received:
                query = query.filter(HostRequest.to_user_id == context.user_id)
            else:
                query = query.filter(
                    or_(HostRequest.to_user_id == context.user_id,
                        HostRequest.from_user_id == context.user_id))

            # TODO: I considered having the latest control message be the single source of truth for
            # the HostRequest.status, but decided agains it because of this filter.
            # Another possibility is to filter in the python instead of SQL, but that's slower
            if request.only_active:
                query = query.filter(
                    or_(
                        HostRequest.status == HostRequestStatus.pending,
                        HostRequest.status == HostRequestStatus.accepted,
                        HostRequest.status == HostRequestStatus.confirmed,
                    ))
                query = query.filter(HostRequest.end_time <= func.now())

            query = query.order_by(Message.id.desc()).limit(pagination + 1)

            results = query.all()

            host_requests = [
                requests_pb2.HostRequest(
                    host_request_id=result.HostRequest.conversation_id,
                    from_user_id=result.HostRequest.from_user_id,
                    to_user_id=result.HostRequest.to_user_id,
                    status=hostrequeststatus2api[result.HostRequest.status],
                    created=Timestamp_from_datetime(
                        result.Conversation.created),
                    from_date=date_to_api(result.HostRequest.from_date),
                    to_date=date_to_api(result.HostRequest.to_date),
                    last_seen_message_id=result.HostRequest.
                    from_last_seen_message_id
                    if context.user_id == result.HostRequest.from_user_id else
                    result.HostRequest.to_last_seen_message_id,
                    latest_message=message_to_pb(result.Message),
                ) for result in results[:pagination]
            ]
            last_request_id = (min(
                map(
                    lambda g: g.HostRequest.conversation_id
                    if g.HostRequest else 1, results[:pagination]))
                               if len(results) > 0 else 0)  # TODO
            no_more = len(results) <= pagination

            return requests_pb2.ListHostRequestsRes(
                last_request_id=last_request_id,
                no_more=no_more,
                host_requests=host_requests)