mirror of
https://github.com/clearml/clearml-server
synced 2025-01-31 10:56:48 +00:00
19 lines
668 B
Python
19 lines
668 B
Python
|
from pymongo.database import Database, Collection
|
||
|
|
|
||
|
|
from database.utils import partition_tags
|
||
|
|
|
||
|
|
|
||
|
|
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}}
|
||
|
|
)
|