mirror of
https://github.com/clearml/clearml-server
synced 2025-03-04 02:59:05 +00:00
Update migration script
This commit is contained in:
parent
027edd86bb
commit
babfcbb707
@ -97,6 +97,22 @@ def _migrate_model_labels(db: Database):
|
|||||||
tasks.update_one({"_id": doc["_id"]}, {"$set": set_commands})
|
tasks.update_one({"_id": doc["_id"]}, {"$set": set_commands})
|
||||||
|
|
||||||
|
|
||||||
|
def _migrate_project_description(db: Database):
|
||||||
|
projects: Collection = db["project"]
|
||||||
|
filter = {
|
||||||
|
"$or": [
|
||||||
|
{
|
||||||
|
"$expr": {"$lt": [{"$strLenCP": "$description"}, 100]},
|
||||||
|
"description": {"$regex": "^Auto-generated at ", "$options": "i"},
|
||||||
|
},
|
||||||
|
{"description": {"$regex": "^Auto-generated during move$", "$options": "i"}},
|
||||||
|
{"description": {"$regex": "^Auto-generated while cloning$", "$options": "i"}},
|
||||||
|
]
|
||||||
|
}
|
||||||
|
for doc in projects.find(filter=filter):
|
||||||
|
projects.update_one({"_id": doc["_id"]}, {"$unset": {"description": 1}})
|
||||||
|
|
||||||
|
|
||||||
def _migrate_project_names(db: Database):
|
def _migrate_project_names(db: Database):
|
||||||
projects: Collection = db["project"]
|
projects: Collection = db["project"]
|
||||||
|
|
||||||
@ -111,20 +127,13 @@ def _migrate_project_names(db: Database):
|
|||||||
for iteration in range(max_tries):
|
for iteration in range(max_tries):
|
||||||
new_name = name.replace("/", "_" * (iteration + 1))
|
new_name = name.replace("/", "_" * (iteration + 1))
|
||||||
try:
|
try:
|
||||||
projects.update_one(
|
projects.update_one({"_id": doc["_id"]}, {"$set": {"name": new_name}})
|
||||||
{"_id": doc["_id"]},
|
|
||||||
{
|
|
||||||
"$set": {"name": new_name}
|
|
||||||
}
|
|
||||||
)
|
|
||||||
break
|
break
|
||||||
except DuplicateKeyError:
|
except DuplicateKeyError:
|
||||||
pass
|
pass
|
||||||
|
|
||||||
if iteration >= max_tries - 1:
|
if iteration >= max_tries - 1:
|
||||||
print(
|
print(f"Could not upgrade the name {name} of the project {doc.get('_id')}")
|
||||||
f"Could not upgrade the name {name} of the project {doc.get('_id')}"
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def migrate_backend(db: Database):
|
def migrate_backend(db: Database):
|
||||||
@ -132,4 +141,5 @@ def migrate_backend(db: Database):
|
|||||||
_migrate_docker_cmd(db)
|
_migrate_docker_cmd(db)
|
||||||
_migrate_model_labels(db)
|
_migrate_model_labels(db)
|
||||||
_migrate_project_names(db)
|
_migrate_project_names(db)
|
||||||
|
_migrate_project_description(db)
|
||||||
_drop_all_indices_from_collections(db, ["task*"])
|
_drop_all_indices_from_collections(db, ["task*"])
|
Loading…
Reference in New Issue
Block a user