re2o/users/migrations/0073_auto_20180629_1614.py

# -*- coding: utf-8 -*-
# Generated by Django 1.10.7 on 2018-06-29 14:14
from __future__ import unicode_literals

from django.conf import settings
from django.db import migrations, models
import django.db.models.deletion
import re2o.mixins


class Migration(migrations.Migration):

    def create_initial_email_address(apps, schema_editor):
        db_alias = schema_editor.connection.alias
        User = apps.get_model("users", "User")
        EMailAddress = apps.get_model("users", "EMailAddress")
        users = User.objects.using(db_alias).all()
        for user in users:
            EMailAddress.objects.using(db_alias).create(
                local_part=user.pseudo,
                user=user
            )

    def delete_all_email_address(apps, schema_editor):
        db_alias = schema_editor.connection.alias
        EMailAddress = apps.get_model("users", "EMailAddress")
        EMailAddress.objects.using(db_alias).delete()

    dependencies = [
        ('users', '0072_auto_20180426_2021'),
    ]

    operations = [
        migrations.CreateModel(
            name='EMailAddress',
            fields=[
                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
                ('local_part', models.CharField(help_text="Local part of the email address", max_length=128, unique=True)),
                ('user', models.ForeignKey(help_text='User of the local email', on_delete=django.db.models.deletion.CASCADE, to=settings.AUTH_USER_MODEL)),
            ],
            bases=(re2o.mixins.RevMixin, re2o.mixins.AclMixin, models.Model),
            options={'permissions': (('view_emailaddress', 'Can see a local email account object'),), 'verbose_name': 'Local email account', 'verbose_name_plural': 'Local email accounts'},
        ),
        migrations.AddField(
            model_name='user',
            name='local_email_enabled',
            field=models.BooleanField(default=False, help_text="Wether or not to enable the local email account."),
        ),
        migrations.AddField(
            model_name='user',
            name='local_email_redirect',
            field=models.BooleanField(default=False, help_text='Whether or not to redirect the local email messages to the main email.'),
        ),
        migrations.RunPython(create_initial_email_address,
                             delete_all_email_address),
    ]