diff --git a/opentech/apply/users/management/commands/migrate_users.py b/opentech/apply/users/management/commands/migrate_users.py index eafd1d0863c294bae10621d6c9d39113ff00b957..95edb3885b26f1a3b408e1c91549534db5bef832 100644 --- a/opentech/apply/users/management/commands/migrate_users.py +++ b/opentech/apply/users/management/commands/migrate_users.py @@ -24,15 +24,13 @@ class Command(BaseCommand): User = get_user_model() users = json.load(json_data) - from pprint import pprint for uid in users: user = users[uid] full_name = user.get('field_otf_real_name', None) - if isinstance(name, dict) and 'und' in name: - full_name = name['und'][0]['safe_value'] - - if not full_name: + try: + full_name = full_name['und'][0]['safe_value'] + except (KeyError, TypeError): full_name = user.get('name') _, created = User.objects.get_or_create(