Fix data migration for user display names

editorsnotes · Mar 7, 2016 · f20e7ab · f20e7ab
1 parent 449a460
commit f20e7ab
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/editorsnotes/main/migrations/0021_populate_display_name.py b/editorsnotes/main/migrations/0021_populate_display_name.py
@@ -7,7 +7,13 @@
 def populate_usernames(apps, schema_editor):
     User = apps.get_model('main', 'User')
     for user in User.objects.all():
-        user.display_name = user._get_display_name()
+        if user.first_name or user.last_name:
+            display_name = user.first_name + ' ' + user.last_name
+            display_name = display_name.strip().rstrip()
+        else:
+            display_name = user.username
+
+        user.display_name = display_name
         user.save()
 
 class Migration(migrations.Migration):