防止 ndb 数据存储中的重复属性

Question

我正在构建一个简单的时事通讯应用程序来收集电子邮件地址。我有一个通过 ajax 将电子邮件值发送到我的 python 应用程序的表单，但是我一辈子都无法弄清楚如何查看电子邮件是否已经存在。下面的代码目前可以正常工作，我只是不知道在哪里/如何添加“检查预先存在的实体”的东西。

import webapp2
import json

from google.appengine.ext import ndb

class Email(ndb.Model):
    email = ndb.StringProperty()
    subscribed = ndb.BooleanProperty()

    @staticmethod
    def create(email):
        ekey = ndb.Key("Email", email)
        entity = Email.get_or_insert(ekey)
        if entity.email:  ###
            # This email already exists
            return None
        entity.email = email
        entity.subscribed = True
        entity.put()
        return entity

class Subscribe(webapp2.RequestHandler):
    def post(self):
        add = Email.create(self.request.get('email'))
        success = add is not None 
        self.response.headers['Content-Type'] = 'application/json'   
        obj = {
            'success': success
        } 
        self.response.out.write(json.dumps(obj))


app = webapp2.WSGIApplication([
    webapp2.Route(r'/newsletter/new', Subscribe),
], debug=True)

Answer 1

根据 Dan 的评论更新了答案。丹，谢谢你纠正我。进一步更新，以回应更新后的问题。

您可以将电子邮件地址设置为电子邮件实体的 id 并使用get_or_insert。在这里，电子邮件存储是多余的，因为它是 id 也是一个属性。您可能可以通过检查下面### 处的另一个属性来摆脱这种冗余。

class Email(ndb.Model):
    email = ndb.StringProperty()
    subscribed = ndb.BooleanProperty()

    @staticmethod
    def create(email):
        ekey = ndb.Key("Email", email)
        entity = Email.get_or_insert(ekey)
        if entity.email:  ###
            # This email already exists
            return None
        entity.email = email
        entity.subscribed = True
        entity.put()
        return entity

class Subscribe(webapp2.RequestHandler):
    def post(self):
        add = Email.create(self.request.get('email'))
        success = add is not None 
        self.response.headers['Content-Type'] = 'application/json'   
        obj = {
            'success': success
        } 
        self.response.out.write(json.dumps(obj))

我没有测试上面的代码，所以可能会有错误，但它应该会引导你走上正确的道路。

Answer 2

始终创建单独的模型类是一种好方法。我创建了单独的模型类并更新了您的订阅者类发布方法，如果电子邮件存在，它将返回 false，否则将返回 true。希望能解决你的问题。

class EmailModel(ndb.Model):
       email = ndb.StringProperty()
       subscribed = ndb.BooleanProperty()

class Subscribe(webapp2.RequestHandler):
      def post(self):
         email = self.request.get('email')
         entity = EmailModel.query(EmailModel.email == email).get()
         if entity:
             # Duplicate
             check = False
         else:
             add = Email()
             add.email = self.request.get('email')
             add.subscribed = True
             add.put()
             check = True
         if check:
              self.response.headers['Content-Type'] = 'application/json'   
              obj = {
                'success': True
              } 
              self.response.out.write(json.dumps(obj))
         else:
              self.response.headers['Content-Type'] = 'application/json'   
              obj = {
                'success': False
              } 
              self.response.out.write(json.dumps(obj))