klmp200/Sith
1
0
Fork 0
mirror of https://github.com/ae-utbm/sith.git synced 2024-11-22 14:13:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Kinda fix the refilling migration: there were some traps!

Browse Source
This commit is contained in:
Skia 2016-08-19 14:39:28 +02:00
parent da96e9da84
commit bfb2dc5f82
1 changed files with 21 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
migrate.py
View File

@ -2,6 +2,7 @@ import MySQLdb
import os import os
import django import django
import random import random
import datetime
from io import StringIO from io import StringIO
from pytz import timezone from pytz import timezone
@ -361,17 +362,27 @@ def migrate_refillings():
100: "OTHER", 100: "OTHER",
None: "OTHER", None: "OTHER",
} }
PAYMENT = {
2: "CARD",
1: "CASH",
0: "CHECK",
}
cur = db.cursor(MySQLdb.cursors.SSDictCursor) cur = db.cursor(MySQLdb.cursors.SSDictCursor)
cur.execute(""" cur.execute("""
SELECT * SELECT *
FROM cpt_rechargements FROM cpt_rechargements
""") """)
Refilling.objects.filter(payment_method="SITH_ACCOUNT").delete()
print("Sith account refillings deleted")
fail = 100
root_cust = Customer.objects.filter(user__id=0).first() root_cust = Customer.objects.filter(user__id=0).first()
mde = Counter.objects.filter(id=1).first() mde = Counter.objects.filter(id=1).first()
threshold_date = datetime.datetime(year=2006, month=3, day=21, hour=22, minute=17)
Refilling.objects.filter(payment_method="CASH").delete()
Refilling.objects.filter(payment_method="CHECK").delete()
Refilling.objects.filter(date__lte=threshold_date.replace(tzinfo=timezone('Europe/Paris'))).delete()
print("Sith account refillings deleted")
fail = 100
for r in cur: for r in cur:
if r['type_paiement_rech'] == 2 and r['date_rech'] > threshold_date:
continue # There are no invoices before threshold_date when paid with credit card
try: try:
cust = Customer.objects.filter(user__id=r['id_utilisateur']).first() cust = Customer.objects.filter(user__id=r['id_utilisateur']).first()
user = User.objects.filter(id=r['id_utilisateur']).first() user = User.objects.filter(id=r['id_utilisateur']).first()
@ -390,6 +401,7 @@ def migrate_refillings():
customer=cust, customer=cust,
operator=op or root_cust.user, operator=op or root_cust.user,
amount=r['montant_rech']/100, amount=r['montant_rech']/100,
payment_method=PAYMENT[r['type_paiement_rech']],
bank=BANK[r['banque_rech']], bank=BANK[r['banque_rech']],
date=r['date_rech'].replace(tzinfo=timezone('Europe/Paris')), date=r['date_rech'].replace(tzinfo=timezone('Europe/Paris')),
) )
@ -513,6 +525,10 @@ def migrate_sellings():
""") """)
Selling.objects.filter(payment_method="SITH_ACCOUNT").delete() Selling.objects.filter(payment_method="SITH_ACCOUNT").delete()
print("Sith account selling deleted") print("Sith account selling deleted")
for c in Customer.objects.all():
c.amount = sum([r.amount for r in c.refillings.all()])
c.save()
print("Customer amount reset to sum of refillings")
ae = Club.objects.filter(unix_name="ae").first() ae = Club.objects.filter(unix_name="ae").first()
mde = Counter.objects.filter(id=1).first() mde = Counter.objects.filter(id=1).first()
root = User.objects.filter(id=0).first() root = User.objects.filter(id=0).first()
@ -580,8 +596,8 @@ def main():
# migrate_products_to_counter() # migrate_products_to_counter()
# reset_customer_amount() # reset_customer_amount()
# migrate_invoices() # migrate_invoices()
# migrate_refillings() migrate_refillings()
# migrate_sellings() migrate_sellings()
reset_index('core', 'counter') reset_index('core', 'counter')
if __name__ == "__main__": if __name__ == "__main__":