ClearML/clearml-server
1
0
Fork 0
mirror of https://github.com/clearml/clearml-server synced 2025-01-31 10:56:48 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Move new migrations to 1_0_2

Browse Source
This commit is contained in:
allegroai 2021-05-27 16:56:47 +03:00
parent ad72a435f1
commit 3f2304549d
2 changed files with 22 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
apiserver/mongo/migrations/1_0_0.py
View File

@ -97,22 +97,6 @@ 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"]
@ -141,5 +125,4 @@ 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*"])

22
apiserver/mongo/migrations/1_0_2.py Normal file
View File

@ -0,0 +1,22 @@
from pymongo.collection import Collection
from pymongo.database import Database
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_backend(db: Database):
_migrate_project_description(db)