Migration from old database

counter/migrations/0013_customer_recorded_products.py

@@ -1,7 +1,29 @@
 # -*- coding: utf-8 -*-
 from __future__ import unicode_literals
 
+from django.utils.translation import ugettext_lazy as _
 from django.db import migrations, models
+from django.conf import settings
+
+from core.models import User
+from counter.models import Customer, Product, Selling, Counter
+
+
+def balance_ecocups(apps, schema_editor):
+    for customer in Customer.objects.all():
+        for selling in customer.buyings.filter(product__id__in=[settings.SITH_RECORD_PRODUCT, settings.SITH_UNRECORD_PRODUCT]).all():
+            if selling.product.id == settings.SITH_RECORD_PRODUCT:
+                customer.recorded_products -= selling.quantity
+            elif selling.product.id == settings.SITH_UNRECORD_PRODUCT:
+                customer.recorded_products += selling.quantity
+        if customer.recorded_products > settings.SITH_RECORD_LIMIT:
+            qt = customer.recorded_products - settings.SITH_RECORD_LIMIT
+            cons = Product.objects.get(id=settings.SITH_RECORD_PRODUCT)
+            Selling(label=_("Record regularization"), product=cons, unit_price=cons.selling_price,
+                    club=cons.club, counter=Counter.objects.filter(name='Foyer').first(),
+                    quantity=qt, seller=User.objects.get(id=0), customer=customer).save()
+            customer.recorded_products -= qt
+            customer.save()
 
 
 class Migration(migrations.Migration):
@@ -16,4 +38,5 @@ class Migration(migrations.Migration):
             name='recorded_products',
             field=models.IntegerField(verbose_name='recorded items', default=0),
         ),
+        migrations.RunPython(balance_ecocups),
     ]