From f512c1bdd742ca2c1f4cbdd9899ad6ae1e7021b0 Mon Sep 17 00:00:00 2001
From: Ludovic Stephan <ludovic.stephan@ens.fr>
Date: Sat, 18 Mar 2017 18:02:23 -0300
Subject: [PATCH] Add better checkout creation

---
 kfet/management/commands/loadkfetdevdata.py | 21 +++++++++++++++++----
 1 file changed, 17 insertions(+), 4 deletions(-)

diff --git a/kfet/management/commands/loadkfetdevdata.py b/kfet/management/commands/loadkfetdevdata.py
index 5b3e9913..328a50a4 100644
--- a/kfet/management/commands/loadkfetdevdata.py
+++ b/kfet/management/commands/loadkfetdevdata.py
@@ -12,7 +12,8 @@ from django.contrib.auth.models import User, Group, Permission, ContentType
 
 from gestioncof.management.base import MyBaseCommand
 from gestioncof.models import CofProfile
-from kfet.models import Account, Article, OperationGroup, Operation, Checkout
+from kfet.models import Account, Article, OperationGroup, Operation,\
+                        Checkout, CheckoutStatement
 
 # Où sont stockés les fichiers json
 DATA_DIR = os.path.join(os.path.dirname(os.path.dirname(__file__)),
@@ -96,13 +97,25 @@ class Command(MyBaseCommand):
         # Caisse
         # ---
 
-        checkout, _ = Checkout.objects.get_or_create(
+        checkout, created = Checkout.objects.get_or_create(
             created_by=Account.objects.get(trigramme='000'),
             name='Chaudron',
-            valid_from=timezone.now(),
-            valid_to=timezone.now() + timedelta(days=365)
+            defaults={
+                'valid_from': timezone.now(),
+                'valid_to': timezone.now() + timedelta(days=730)
+            },
         )
 
+        if created:
+            CheckoutStatement.objects.create(
+                by=Account.objects.get(trigramme='000'),
+                checkout=checkout,
+                balance_old=0,
+                balance_new=0,
+                amount_taken=0,
+                amount_error=0
+            )
+
         # ---
         # Opérations
         # ---