kpsul/bda/migrations/0006_add_tirage_switch.py

# -*- coding: utf-8 -*-
from __future__ import unicode_literals

from django.db import migrations, models
from django.utils import timezone


def forwards_func(apps, schema_editor):
    Tirage = apps.get_model("bda", "Tirage")
    db_alias = schema_editor.connection.alias
    for tirage in Tirage.objects.using(db_alias).all():
        if tirage.tokens:
            tirage.tokens = 'Before %s\n"""%s"""\n' % (
                timezone.now().strftime("%y-%m-%d %H:%M:%S"),
                tirage.tokens,
            )
            tirage.save()


class Migration(migrations.Migration):
    dependencies = [("bda", "0005_encoding")]

    operations = [
        migrations.RenameField("tirage", "token", "tokens"),
        migrations.AddField(
            model_name="tirage",
            name="enable_do_tirage",
            field=models.BooleanField(
                default=False, verbose_name=b"Le tirage peut \xc3\xaatre lanc\xc3\xa9"
            ),
        ),
        migrations.RunPython(forwards_func, migrations.RunPython.noop),
    ]