Sybren A. Stüvel
1e823a9dbe
Eve doesn't have any counting methods on `current_app.data`, so there is no one-to-one translation for `cursor.count()` in `file_storage/__init__.py`. Since the call was only used in a debug log entry, I just removed it altogether. I removed `pillar.cli.operations.index_users_rebuild()`, as it was importing `pillar.api.utils.algolia.algolia_index_user_save` which doesn't exist any more, so the code was dead anyway.
48 lines
1.5 KiB
Python
48 lines
1.5 KiB
Python
"""Code for merging projects."""
|
|
import logging
|
|
|
|
from bson import ObjectId
|
|
|
|
from pillar import current_app
|
|
from pillar.api.file_storage.moving import move_to_bucket
|
|
from pillar.api.utils import random_etag, utcnow
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
def merge_project(pid_from: ObjectId, pid_to: ObjectId):
|
|
"""Move nodes and files from one project to another.
|
|
|
|
Note that this may invalidate the nodes, as their node type definition
|
|
may differ between projects.
|
|
"""
|
|
log.info('Moving project contents from %s to %s', pid_from, pid_to)
|
|
assert isinstance(pid_from, ObjectId)
|
|
assert isinstance(pid_to, ObjectId)
|
|
|
|
files_coll = current_app.db('files')
|
|
nodes_coll = current_app.db('nodes')
|
|
|
|
# Move the files first. Since this requires API calls to an external
|
|
# service, this is more likely to go wrong than moving the nodes.
|
|
query = {'project': pid_from}
|
|
to_move = files_coll.find(query, projection={'_id': 1})
|
|
|
|
to_move_count = files_coll.count_documents(query)
|
|
log.info('Moving %d files to project %s', to_move_count, pid_to)
|
|
for file_doc in to_move:
|
|
fid = file_doc['_id']
|
|
log.debug('moving file %s to project %s', fid, pid_to)
|
|
move_to_bucket(fid, pid_to)
|
|
|
|
# Mass-move the nodes.
|
|
etag = random_etag()
|
|
result = nodes_coll.update_many(
|
|
query,
|
|
{'$set': {'project': pid_to,
|
|
'_etag': etag,
|
|
'_updated': utcnow(),
|
|
}}
|
|
)
|
|
log.info('Moved %d nodes to project %s', result.modified_count, pid_to)
|