Flask SQLAlchemy以DRY代码更新用户信息

问题描述

当前使用Flask和SQLAlchemy，但对Python来说相对较新。假设我有一个将用户信息作为对象的函数，然后检查是否在数据库中保存了该用户。

user_information = { "first_name": "John","last_name": "Doe","phone_number": "123-456-7890","address": "555 Pine street","favorite_food": "" }
user_record = flask.g.db.session.query(User).filter_by(first_name=user_information["first_name"]).first()

如果说user_record回来为

user_record = User("first_name": "John","phone_number": "098-765-4321","address": "","favorite_food": "pizza")

请注意，电话号码已更改，我们的记录中没有记录地址，新的输入数据中没有喜欢的食物。

我的问题是，我该如何干净地确保我们更新用户数据库以获取最新的电话号码，如果我们以前没有任何数据，请更新地址，忽略新的喜欢的食物，因为我们希望拥有旧的数据。我知道我可以简单地创建一个带有大量if / else的新对象，以便为每个键/值进行选择并将其保存到数据库中，但是不会变干，尤其是当我们有30个用户列时。 db.merge是否已经处理了这个问题，或者Python中有一些漂亮的速记技巧来创建更新的用户对象？

解决方法

在您的问题陈述中，听起来好像您不知道任何ID，但是有一些列（也许是名字和姓氏）在一起都是唯一的。然后，您将具有以下内容：

class User(db.Model):
  @classmethod
  def create_or_update(cls,fields,identifiers):
    # Remove all empty fields - you could potentially also early return here
    # if there are no filled fields at all
    filled_fields = {k: v for k,v in identifiers.items() if v != "" and v is not None}

    match = cls.objects.filter(**identifiers).one_or_none()
    if match:
      # Object exists,update its values
      for key,value in filled_fields.items():
        setattr(match,key,value)
    else:
      # Object does not exist,create it
      new_object = cls(**identifiers,**filled_fields)
      db.session.add(new_object)
    db.session.commit()
    return match

user_information = { "phone_number": "123-456-7890","address": "555 Pine street","favorite_food": "" }
identifiers = { "first_name": "John","last_name": "Doe" }
User.create_or_update(user_information,identifiers)

一些指针：

如果多个记录匹配，
• one_or_none会给出错误消息
• setattr可用于设置名称为变量的变量，例如some_dict[some_key]
• 记住commit所做的更改
• 如果它们的值相同，则覆盖字段没有危害

但是，如果您使用的是PostgreSQL，并且想更进一步，请查看INSERT ON CONFLICT UPDATE（也称为UPSERT）。这意味着，如果输入的记录与某些唯一性约束（例如名字+姓氏）冲突，则将使用现有行并在其中分配新值。基本上，这一切都为您完成！