mirror of
https://github.com/ae-utbm/sith3.git
synced 2024-07-01 02:48:03 +00:00
43 lines
1.7 KiB
Python
43 lines
1.7 KiB
Python
# -*- 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):
|
|
|
|
dependencies = [
|
|
('counter', '0012_auto_20170515_2202'),
|
|
]
|
|
|
|
operations = [
|
|
migrations.AddField(
|
|
model_name='customer',
|
|
name='recorded_products',
|
|
field=models.IntegerField(verbose_name='recorded items', default=0),
|
|
),
|
|
migrations.RunPython(balance_ecocups),
|
|
]
|