helloworld_api.py

import endpoints
from protorpc import messages
from protorpc import message_types
from protorpc import remote


from auth.endpoints import get_current_user

package = 'Hello'


class Greeting(messages.Message):
    """Greeting that stores a message."""
    message = messages.StringField(1)


class GreetingCollection(messages.Message):
    """Collection of Greetings."""
    items = messages.MessageField(Greeting, 1, repeated=True)


STORED_GREETINGS = GreetingCollection(items=[
    Greeting(message='hello world!'),
    Greeting(message='goodbye world!'),
    ])


@endpoints.api(name='helloworld', version='v1')
class HelloWorldApi(remote.Service):
    """Helloworld API v1."""

    @endpoints.method(message_types.VoidMessage, GreetingCollection,
                      path='hellogreeting', http_method='GET',
                      name='greetings.listGreeting')
    def greetings_list(self, *args):
        user = get_current_user()
        if not user:
            raise endpoints.UnauthorizedException()

        return STORED_GREETINGS


APPLICATION = endpoints.api_server([HelloWorldApi])