Skip to content

Commit 5547715

Browse files
committed
Merge remote-tracking branch 'origin/978-migrate-users' into 978-migrate-users
# Conflicts: # scripts/migration/migrate.py
2 parents c936760 + 8050460 commit 5547715

File tree

1 file changed

+8
-10
lines changed

1 file changed

+8
-10
lines changed

scripts/migration/migrate.py

Lines changed: 8 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -48,10 +48,10 @@ def email_user_new_login(user_email):
4848
print(f"Login to the new Clowder instance: {user_email}")
4949

5050

51-
def generate_user_api_key(user, password):
51+
def generate_user_api_key(user, password=DEFAULT_PASSWORD):
5252
"""Generate an API key for a user."""
5353
login_endpoint = f"{CLOWDER_V2}/api/v2/login"
54-
user['password'] = DEFAULT_PASSWORD
54+
user.update({"password": password})
5555
response = requests.post(login_endpoint, json=user)
5656
token = response.json().get("token")
5757
current_headers = {"Authorization": f"Bearer {token}"}
@@ -105,7 +105,7 @@ def add_v1_space_members_to_v2_group(space, group_id, headers):
105105
for member in space_members:
106106
member_email = member["email"]
107107
endpoint = f"{CLOWDER_V2}/api/v2/groups/{group_id}/add/{member_email}"
108-
response = requests.post(
108+
requests.post(
109109
endpoint,
110110
headers=headers,
111111
)
@@ -128,7 +128,7 @@ def create_local_user(user_v1):
128128
if existing_user.get("email") == user_v1["email"]:
129129
print(f"User {user_v1['email']} already exists in Clowder v2.")
130130
return generate_user_api_key(
131-
user_v1, DEFAULT_PASSWORD
131+
existing_user, DEFAULT_PASSWORD
132132
) # Return the existing user's API key
133133

134134
# User does not exist, proceed to create a new user
@@ -286,10 +286,7 @@ def download_and_upload_file(file, all_dataset_folders, dataset_v2_id, headers_v
286286
def process_user_and_resources(user_v1, USER_MAP, DATASET_MAP):
287287
"""Process user resources from Clowder v1 to Clowder v2."""
288288
user_v1_datasets = get_clowder_v1_user_datasets(user_id=user_v1["id"])
289-
try:
290-
user_v2_api_key = create_local_user(user_v1)
291-
except Exception as e:
292-
print(e)
289+
user_v2_api_key = create_local_user(user_v1)
293290
USER_MAP[user_v1["id"]] = user_v2_api_key
294291
user_headers_v2 = {
295292
"x-api-key": user_v2_api_key,
@@ -344,7 +341,7 @@ def process_user_and_resources(user_v1, USER_MAP, DATASET_MAP):
344341
"identityProvider": "Chen Wang ([email protected]) [Local Account]",
345342
}
346343
]
347-
users_v1 = get_clowder_v1_users()
344+
# users_v1 = get_clowder_v1_users()
348345
for user_v1 in users_v1:
349346
if (
350347
"[Local Account]" in user_v1["identityProvider"]
@@ -356,7 +353,8 @@ def process_user_and_resources(user_v1, USER_MAP, DATASET_MAP):
356353
print(f"Migrated user {user_v1['email']} and associated resources.")
357354
else:
358355
print(f"Skipping user {user_v1['email']} as it is not a local account.")
359-
print(f"Now migrating spaces")
356+
357+
print("Now migrating spaces.")
360358
for user_v1 in users_v1:
361359
print(f"Migrating spaces of user {user_v1['email']}")
362360
user_v1_spaces = get_clowder_v1_user_spaces(user_v1)

0 commit comments

Comments
 (0)