re2o/cotisations/migrations/0026_auto_20171028_0126.py

# -*- coding: utf-8 -*-
# Generated by Django 1.10.7 on 2017-10-27 23:26
from __future__ import unicode_literals

from django.db import migrations, models


def create_type(apps, schema_editor):
    Cotisation = apps.get_model('cotisations', 'Cotisation')
    Vente = apps.get_model('cotisations', 'Vente')
    Article = apps.get_model('cotisations', 'Article')
    db_alias = schema_editor.connection.alias
    articles = Article.objects.using(db_alias).all()
    ventes = Vente.objects.using(db_alias).all()
    cotisations = Cotisation.objects.using(db_alias).all()
    for article in articles:
        if article.iscotisation:
            article.type_cotisation='All'
            article.save(using=db_alias)
    for vente in ventes:
        if vente.iscotisation:
            vente.type_cotisation='All'
            vente.save(using=db_alias)
    for cotisation in cotisations:
        cotisation.type_cotisation='All'
        cotisation.save(using=db_alias)

def delete_type(apps, schema_editor):
    Vente = apps.get_model('cotisations', 'Vente')
    Article = apps.get_model('cotisations', 'Article')
    db_alias = schema_editor.connection.alias
    articles = Articles.objects.using(db_alias).all()
    ventes = Vente.objects.using(db_alias).all()
    for article in articles:
        if article.type_cotisation:
            article.iscotisation=True
        else:
            article.iscotisation=False
        article.save(using=db_alias)
    for vente in ventes:
        if vente.iscotisation:
            vente.iscotisation=True
        else:
            vente.iscotisation=False
        vente.save(using=db_alias)

class Migration(migrations.Migration):

    dependencies = [
        ('cotisations', '0025_article_type_user'),
    ]

    operations = [
        migrations.AddField(
            model_name='article',
            name='type_cotisation',
            field=models.CharField(blank=True, choices=[('Connexion', 'Connexion'), ('Adhesion', 'Adhesion'), ('All', 'All')], default=None, max_length=255, null=True),
        ),
        migrations.AddField(
            model_name='cotisation',
            name='type_cotisation',
            field=models.CharField(choices=[('Connexion', 'Connexion'), ('Adhesion', 'Adhesion'), ('All', 'All')], max_length=255, default='All'),
        ),
        migrations.AddField(
            model_name='vente',
            name='type_cotisation',
            field=models.CharField(blank=True, choices=[('Connexion', 'Connexion'), ('Adhesion', 'Adhesion'), ('All', 'All')], max_length=255, null=True),
        ),
        migrations.RunPython(create_type, delete_type),
        migrations.RemoveField(
            model_name='article',
            name='iscotisation',
        ),
        migrations.RemoveField(
            model_name='vente',
            name='iscotisation',
        ),
    ]