from django.db import migrations, models


def fill_blank_countries(apps, schema_editor):
    Zone = apps.get_model('core', 'Zone')
    Zone.objects.filter(country='').update(country='Unknown')


class Migration(migrations.Migration):

    dependencies = [
        ('core', '0013_zone_rate_per_km'),
    ]

    operations = [
        migrations.AddField(
            model_name='zone',
            name='allow_return_trips',
            field=models.BooleanField(default=False),
        ),
        migrations.RemoveField(model_name='zone', name='address'),
        migrations.RemoveField(model_name='zone', name='city'),
        migrations.RemoveField(model_name='zone', name='country_code'),
        migrations.RemoveField(model_name='zone', name='description'),
        migrations.RemoveField(model_name='zone', name='is_active'),
        migrations.RemoveField(model_name='zone', name='lat'),
        migrations.RemoveField(model_name='zone', name='lng'),
        migrations.RemoveField(model_name='zone', name='radius_km'),
        migrations.RemoveField(model_name='zone', name='rate_per_km'),
        migrations.RemoveField(model_name='zone', name='state'),
        migrations.RunPython(fill_blank_countries, migrations.RunPython.noop),
        migrations.AlterField(
            model_name='zone',
            name='country',
            field=models.CharField(max_length=120),
        ),
    ]