def getUserEntity(user): entity = User.all().filter('user =',user).get() if not entity: entity = User(user=user) entity.put() return entity
def test_legal_moves(self): userX = User() userX.put() myGame = Game(userX, moveX = True, last_cell = 1, all_mini_wins = [' ']*9, metaboard = [' ']*9) legal_moves = getLegalMoves(myGame) self.assert_(legal_moves == [(1, 0), (1, 1), (1, 2), (1, 3), (1, 4), (1, 5), (1, 6), (1, 7), (1, 8)], msg=str(legal_moves))
def test_ai(self): userX = User() userX.put() foo = 6 myGame = Game(userX, moveX = True, last_cell = 4, all_mini_wins = [' ']*9, metaboard = [' ']*9) result = negamax(myGame, 7, float('-inf'), float('inf')) self.assertEqual(result, (0, (4, 0)))
class UserFromSession(): """Take the session, get the user key, and return the user object. If no user if set yet, create a new user, store in datastore, set cookie, and return that user""" user = None; def __init__(self, session): user_key = session.get('user_key') if user_key: self.user = db.get(db.Key(user_key)) if not self.user: # No user key stored - make a new one self.user = User(); self.user.put() session['user_key'] = str(self.user.key()) def get_user(self): return self.user
def current_user(self): if self.session.get("user"): # User is logged in return self.session.get("user") else: # Either used just logged in or just saw the first page # We'll see here cookie = facebook.get_user_from_cookie(self.request.cookies, FACEBOOK_APP_ID, FACEBOOK_APP_SECRET) if cookie: # Okay so user logged in. # Now, check to see if existing user user = User.get_by_key_name(cookie["uid"]) if not user: # Not an existing user so get user info graph = facebook.GraphAPI(cookie["access_token"]) profile = graph.get_object("me") loc = profile["location"] loc_id = loc["id"] city = graph.get_object(loc_id) user = User( key_name=str(profile["id"]), id=str(profile["id"]), name=profile["name"], profile_url=profile["link"], access_token=cookie["access_token"], city = city["name"] ) print user.access_token user.put() elif user.access_token != cookie["access_token"]: user.access_token = cookie["access_token"] user.put() # User is now logged in self.session["user"] = dict( name=user.name, profile_url=user.profile_url, id=user.id, access_token=user.access_token ) return self.session.get("user") return None