def __init__(self, email): self.email = email user = gusers.User(email) user_role = db.GqlQuery("SELECT * FROM DbUserRole WHERE account = :1", user).get() if user_role is None: user_role = DbUserRole(role=DbUserRole._USER_ROLE, account=user) user_role.put() self.db_role = user_role
def get(self): user = users.GetCurrentUser() u = db.GqlQuery("SELECT * FROM DbUserRole WHERE account = :1", user).get() if u is not None: u.role = DbUserRole._ADMIN_ROLE u.put() else: u = DbUserRole(role=DbUserRole._ADMIN_ROLE, account=user) u.put() self.response.out.write("""Hello, %s. You're now defined as '%s'""" % (user.nickname(), u.role))