from __future__ import unicode_literals

from django.db import migrations, models


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()
        user.save()

class Migration(migrations.Migration):

    dependencies = [
        ('main', '0020_user_display_name'),
    ]

    operations = [
        migrations.RunPython(populate_usernames)
    ]

from __future__ import unicode_literals

from django.db import migrations, models


def populate_usernames(apps, schema_editor):
    User = apps.get_model('main', 'User')
    for user in User.objects.all():
        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):

    dependencies = [
        ('main', '0020_user_display_name'),
    ]

    operations = [
        migrations.RunPython(populate_usernames)
    ]

Fix data migration for user display names
Fix data migration for user display names

  from __future__ import unicode_literals
  
  from django.db import migrations, models
  
  
  def populate_usernames(apps, schema_editor):
      User = apps.get_model('main', 'User')
      for user in User.objects.all():
+         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 = user._get_display_name()
+         user.display_name = display_name
          user.save()
  
  class Migration(migrations.Migration):
  
      dependencies = [
          ('main', '0020_user_display_name'),
      ]
  
      operations = [
          migrations.RunPython(populate_usernames)
      ]
  
Fix data migration for user display names
## Code Before:
from __future__ import unicode_literals

from django.db import migrations, models


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()
        user.save()

class Migration(migrations.Migration):

    dependencies = [
        ('main', '0020_user_display_name'),
    ]

    operations = [
        migrations.RunPython(populate_usernames)
    ]

## Instruction:
Fix data migration for user display names
## Code After:
from __future__ import unicode_literals

from django.db import migrations, models


def populate_usernames(apps, schema_editor):
    User = apps.get_model('main', 'User')
    for user in User.objects.all():
        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):

    dependencies = [
        ('main', '0020_user_display_name'),
    ]

    operations = [
        migrations.RunPython(populate_usernames)
    ]
