如何解决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)。这意味着,如果输入的记录与某些唯一性约束(例如名字+姓氏)冲突,则将使用现有行并在其中分配新值。基本上,这一切都为您完成!
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。