2019-09-24 18:34:35 +00:00
|
|
|
from pymongo.database import Database, Collection
|
|
|
|
|
|
2021-01-05 14:28:49 +00:00
|
|
|
from apiserver.database.utils import partition_tags
|
2019-09-24 18:34:35 +00:00
|
|
|
|
|
|
|
|
|
|
|
|
|
def migrate_backend(db: Database):
|
|
|
|
|
for name in ("project", "task", "model"):
|
|
|
|
|
collection: Collection = db[name]
|
|
|
|
|
for doc in collection.find(projection=["tags", "system_tags"]):
|
|
|
|
|
tags = doc.get("tags")
|
|
|
|
|
if tags is not None:
|
|
|
|
|
user_tags, system_tags = partition_tags(
|
|
|
|
|
name, tags, doc.get("system_tags", [])
|
|
|
|
|
)
|
|
|
|
|
collection.update_one(
|
|
|
|
|
{"_id": doc["_id"]},
|
|
|
|
|
{"$set": {"system_tags": system_tags, "tags": user_tags}}
|
|
|
|
|
)
|
