apis.py

from google.appengine.api import channel
from google.appengine.api import users
from google.appengine.ext import ndb

import data
import datetime
import json
import webapp2


class GetUser(webapp2.RequestHandler):
    def post(self):
        user = users.get_current_user()
        result = data.User.query(data.User.user == user).fetch()

        if len(result) == 0:
            response = {'result': 'not_found' }
            self.response.write(json.dumps(response))
            return

        currentUser = result[0]

        response = {
            'result': 'ok',
            'user_id': currentUser.user.user_id(),
            'name': currentUser.name,
            'avatar': currentUser.avatar,
            'codes': currentUser.codes }
        self.response.write(json.dumps(response));


class SetUser(webapp2.RequestHandler):
    def post(self):
        user = users.get_current_user()
        result = data.User.query(data.User.user == user).fetch()
        currentUser = None
        if len(result) > 0:
            currentUser = result[0]
        if currentUser is None:
            currentUser = data.User()
            currentUser.user = user

        currentUser.name = self.request.get('name')
        currentUser.avatar = self.request.get('avatar')
        key = currentUser.put()

        response = {'result': 'ok', 'key': key.id() }
        self.response.write(json.dumps(response))


class SaveCodes(webapp2.RequestHandler):
    def post(self):
        user = users.get_current_user()
        result = data.User.query(data.User.user == user).fetch()
        currentUser = None
        if len(result) > 0:
            currentUser = result[0]
        if currentUser is None:
            currentUser = data.User()
            currentUser.user = user

        currentUser.codes = self.request.get_all('codes[]')
        key = currentUser.put()

        response = {'result': 'ok', 'key': key.id() }
        self.response.write(json.dumps(response))


class SaveMap(webapp2.RequestHandler):
    def post(self):
        user = users.get_current_user()
        result = data.Map.query(data.Map.owner == user).fetch()
        blocks = None
        if len(result) > 0:
            blocks = result[0]
        if blocks is None:
            blocks = data.Map()
            blocks.created = datetime.datetime.now()
            blocks.owner = user

        blocks.data = self.request.get('data')
        blocks.updated = datetime.datetime.now()
        key = blocks.put()

        response = {'result': 'ok', 'key': key.id() }
        self.response.write(json.dumps(response))


class GetMaps(webapp2.RequestHandler):
    def post(self):
        user = users.get_current_user()
        maps = data.Map.query(data.Map.owner == user).fetch()
        maps = maps if maps is not None else []
        maps = [m.serialize() for m in maps]
        response = {
            'result': 'ok',
            'maps': maps
            }
        self.response.write(json.dumps(response));


class LoadMapById(webapp2.RequestHandler):
    def post(self):
        key = self.request.get('key')
        result = data.Map.get_by_id(long(key))
        if result is None:
            self.error(404)
            self.response.out.write('not_found.')
            return

        response = {
            'result': 'ok',
            'key': result.key.id(),
            'data': result.data
            }
        self.response.write(json.dumps(response));


class SetMapName(webapp2.RequestHandler):
    def post(self):
        key = self.request.get('key')
        result = data.Map.get_by_id(long(key))
        if result is None:
            self.error(404)
            self.response.out.write('not_found.')
            return

        result.name = self.request.get('name')
        key = result.put();
        response = {
            'result': 'ok',
            'key': result.key.id()
            }
        self.response.write(json.dumps(response));


class CreateMap(webapp2.RequestHandler):
    def post(self):
        user = users.get_current_user()
        name = self.request.get('name')

        blocks = data.Map()
        blocks.name = name;
        blocks.created = datetime.datetime.now()
        blocks.updated= datetime.datetime.now()
        blocks.owner = user
        blocks.put()

        response = {
            'result': 'ok',
            'map': blocks.serialize()
            }
        self.response.write(json.dumps(response));


class DeleteMap(webapp2.RequestHandler):
    def post(self):
        key = self.request.get('key')
        result = data.Map.get_by_id(long(key))
        if result is None:
            self.error(404)
            self.response.out.write('not_found.')
            return

        ndb.Key(data.Map, long(key)).delete()
        response = {
            'result': 'ok',
            }
        self.response.write(json.dumps(response));


class SendMessage(webapp2.RequestHandler):
    def post(self):
        user = users.get_current_user()
        token = self.request.get('token')
        data = {
            'message': self.request.get('message'),
            'user_id': user.user_id()
            }            
        channel.send_message(token, json.dumps(data))


application = webapp2.WSGIApplication([
        ('/_/get_user/', GetUser),
        ('/_/set_user/', SetUser),
        ('/_/save_codes/', SaveCodes),
        ('/_/save_map/', SaveMap),
        ('/_/get_maps/', GetMaps),
        ('/_/load_map_by_id/', LoadMapById),
        ('/_/set_map_name/', SetMapName),
        ('/_/create_map/', CreateMap),
        ('/_/delete_map/', DeleteMap),
        ('/_/send_message/', SendMessage),
    ], debug=True)