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
