TinyDB - 更新数据 (upsert)

我们讨论了更新和插入查询，但有时我们需要同时进行更新和插入操作。在这种情况下，TinyDB 提供了独特的upsert查询。此查询帮助我们根据条件同时插入和更新数据。它的工作方式如下：

• 如果数据存在，则选择更新操作。

• 如果数据不存在，则选择插入操作。

语法

upsert查询的语法如下：

db.upsert({ 'key' : 'value', 'logged - in' : True}, regular expression)

让我们通过几个例子来演示如何在 TinyDB 中使用upsert查询。我们将使用前面章节中使用过的相同的学生数据库。

示例 1

让我们看看如何使用upsert查询将学号为 2 的学生的地址更改为“Surat”。在这种情况下，我们有一个匹配的用户，因此它将更新地址并将登录状态设置为 True：

from tinydb import TinyDB, Query
db = TinyDB('student.json')
db.upsert({'address':'Surat'}, Query().roll_number==2)

这将产生以下输出，这意味着记录编号“2”已更新。

[2]

使用以下代码验证记录编号“2”是否已更新：

db.get(doc_id=2)

它将显示更新后的数据：

{
   'roll_number': 2,
   'st_name': 'Ram',
   'mark': [250, 280],
   'subject': ['TinyDB', 'MySQL'],
   'address': 'Surat'
}

示例 2

让我们看看如何在表中对不存在的数据使用upsert查询：

from tinydb import TinyDB, Query
db = TinyDB('student.json')
db.upsert({'E-mail':'ram@gmail.com','logged-in': True},
Query().roll_number==2)

它将显示以下输出，这意味着 ID 为“2”的文档已更新。

[2]

使用以下代码验证更新后的值：

db.get(doc_id=2)

它将产生以下输出：

{
   "roll_number":2,
   "st_name":"Ram",
   "mark":[
      250,
      280
   ],
   "subject":[
      "TinyDB",
      "MySQL"
   ],
   "address":"Surat",
   "logged-in":true,
   "E-mail":"ram@gmail.com"
}

请注意，我们使用upsert查询创建了一个新的键（E-mail），它之前不存在，并为其提供了值“ram@gmail.com”。