6

所以我正在为 jinja2 模板平台上的 google appengine 用 python 编写一个应用程序。我已经让 OpenID 在我的网站上正常工作，它允许用户登录，我可以在右上角显示他们的电子邮件/ID。

现在我希望用户拥有自己的用户名，并能够存储一些关于他们的额外数据，例如站点访问和站点区域的某些帖子。

我知道我需要在数据库中创建一个包含用户和所有内容的表，但使用 OpenID 执行此操作的正确方法是什么。我想在有人第一次登录时自动创建用户，但我不太确定如何有效地做到这一点。我想我可以通过在登录后将它们重定向到一个页面来管理它，该页面检查他们在数据库中的 federated_identity，如果它已经存在，它只是将它们重定向到主页，但如果不存在，它会创建一个新用户。

有没有更有效的方法，这样我就不会在每次有人登录时都查询数据库，或者这是一种相当不错的方法？

4

1 回答 1

6

您是对的，您必须创建自己的用户模型并存储额外的信息以及您在应用程序中拥有的任何业务逻辑。您所要做的就是存储一些独特的东西（federated_id例如）以及额外username的 ,name等。

使用 OpenID 登录后，您将检查该特定信息federated_id是否在您的数据存储中，您将返回该特定信息user_db，或者使用默认值创建一个新的并返回新创建的user_db.

重要的是，在您的请求中，您将始终处理您自己的user_db实体，而不是users.get_current_user()因为这基本上是只读的和有限的。

这是一个完整的示例，它还演示了一个带有函数的基本处理程序类，该get_user()函数将user_db在用户登录时返回您自己的实体，None否则。如果用户是第一次登录，则会使用一些默认值（name, username, email, admin, federated_id）创建一个新的 user_db 实体：

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

class User(ndb.Model):
  name = ndb.StringProperty(required=True)
  username = ndb.StringProperty(required=True)
  email = ndb.StringProperty()
  federated_id = ndb.StringProperty()
  admin = ndb.BooleanProperty()

class BaseHandler(webapp2.RequestHandler):
  def get_user_db(self):
    federated_user = users.get_current_user()
    if not federated_user:
      return None
    user_db_qry = User.query(User.federated_id == federated_user.user_id())
    user_db_list = user_db_qry.fetch(1)
    #If that returns non empty list, then return the first element which is the user_db
    if user_db_list:
      return user_db_list[0]

    #Otherwise create a new user_db entity, with default values and return
    user_db = User(
        name=federated_user.nickname().title(),
        username=federated_user.nickname(),
        email=federated_user.email(),
        federated_id=federated_user.user_id(),
        admin=users.is_current_user_admin(),
      )
    user_db.put()
    return user_db

class MainHandler(BaseHandler):
  def get(self):
    self.response.headers['Content-Type'] = 'text/html'
    if self.get_user_db():
      self.response.out.write('Hello, %s!</br>' % self.get_user_db().name)
      self.response.out.write('<a href="%s">Logout</a>' % (users.create_logout_url(self.request.uri)))
    else:
      self.response.out.write('Hello, stranger!<br>')
      self.response.out.write('<a href="%s">Login</a>' % (users.create_login_url(self.request.uri)))

app = webapp2.WSGIApplication([('/', MainHandler)],
                              debug=True)

这个问题有很多不同的方法，但我认为这是一个很好的开始。

于 2012-06-25T10:28:43.687 回答