re2o/preferences/migrations/0063_mandate.py

# -*- coding: utf-8 -*-
# Generated by Django 1.11.23 on 2019-09-21 18:23
from __future__ import unicode_literals

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


def create_current_mandate(apps, schema_editor):
    AssoOption = apps.get_model("preferences", "AssoOption")
    Mandate = apps.get_model("preferences", "Mandate")
    Adherent = apps.get_model("users", "Adherent")
    assooption = AssoOption.objects.get_or_create()[0]
    pres_name = assooption.pres_name
    l = pres_name.split(" ")
    try:
        name, surname = l[0], l[1]
        president = Adherent.objects.get(
            name__icontains=name, surname__icontains=surname
        )
        Mandate.objects.create(president=president, start_date=timezone.now())
    except Exception as e:
        print(
            "Warning : I was unable to find an adherent corresponding to %s. You might want to edit your preferences afterward. I will disable the sending of vouchers by email."
            % pres_name
        )
        CotisationsOption = apps.get_model("preferences", "CotisationsOption")
        cotisoption = CotisationsOption.objects.get_or_create()[0]
        cotisoption.send_voucher_mail = False
        cotisoption.save()


class Migration(migrations.Migration):

    dependencies = [
        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
        ("preferences", "0062_auto_20190910_1909"),
    ]

    operations = [
        migrations.CreateModel(
            name="Mandate",
            fields=[
                (
                    "id",
                    models.AutoField(
                        auto_created=True,
                        primary_key=True,
                        serialize=False,
                        verbose_name="ID",
                    ),
                ),
                ("start_date", models.DateTimeField(verbose_name="start date")),
                (
                    "end_date",
                    models.DateTimeField(
                        blank=True, null=True, verbose_name="end date"
                    ),
                ),
                (
                    "president",
                    models.ForeignKey(
                        blank=True,
                        help_text="Displayed on subscription vouchers",
                        null=True,
                        on_delete=django.db.models.deletion.SET_NULL,
                        to=settings.AUTH_USER_MODEL,
                        verbose_name="President of the association",
                    ),
                ),
            ],
            options={
                "verbose_name": "Mandate",
                "verbose_name_plural": "Mandates",
                "permissions": (("view_mandate", "Can view a mandate"),),
            },
            bases=(re2o.mixins.RevMixin, re2o.mixins.AclMixin, models.Model),
        ),
        migrations.RunPython(create_current_mandate),
        migrations.AlterField(
            model_name="cotisationsoption",
            name="send_voucher_mail",
            field=models.BooleanField(
                default=False,
                help_text="Be carefull, if no mandate is defined on the preferences page, errors will be triggered when generating vouchers.",
                verbose_name="Send voucher by email when the invoice is controlled.",
            ),
        ),
        migrations.RemoveField(model_name="assooption", name="pres_name"),
    ]