mirror of
https://github.com/clearml/clearml-server
synced 2025-02-01 11:26:43 +00:00
44 lines
1.1 KiB
Python
44 lines
1.1 KiB
Python
|
from collections import Collection
|
||
|
|
from typing import Sequence
|
||
|
|
|
||
|
|
from pymongo.database import Database, Collection
|
||
|
|
|
||
|
|
|
||
|
|
def _drop_all_indices_from_collections(db: Database, names: Sequence[str]):
|
||
|
|
for collection_name in db.list_collection_names():
|
||
|
|
if collection_name not in names:
|
||
|
|
continue
|
||
|
|
collection: Collection = db[collection_name]
|
||
|
|
collection.drop_indexes()
|
||
|
|
|
||
|
|
|
||
|
|
def migrate_auth(db: Database):
|
||
|
|
"""
|
||
|
|
Remove the old indices from the collections since
|
||
|
|
they may come out of sync with the latest changes
|
||
|
|
in the code and mongo libraries update
|
||
|
|
"""
|
||
|
|
_drop_all_indices_from_collections(db, ["user"])
|
||
|
|
|
||
|
|
|
||
|
|
def migrate_backend(db: Database):
|
||
|
|
"""
|
||
|
|
Remove the old indices from the collections since
|
||
|
|
they may come out of sync with the latest changes
|
||
|
|
in the code and mongo libraries update
|
||
|
|
"""
|
||
|
|
_drop_all_indices_from_collections(
|
||
|
|
db,
|
||
|
|
[
|
||
|
|
"company",
|
||
|
|
"model",
|
||
|
|
"project",
|
||
|
|
"queue",
|
||
|
|
"settings",
|
||
|
|
"task",
|
||
|
|
"task__trash",
|
||
|
|
"user",
|
||
|
|
"versions",
|
||
|
|
],
|
||
|
|
)
|