From 95c59de4fcf1f5793c4e0c21ea1565e81db4d58e Mon Sep 17 00:00:00 2001
From: Ludovic Stephan <ludovic.stephan@ens.fr>
Date: Wed, 29 Mar 2017 23:16:40 -0300
Subject: [PATCH] Add supplier to dev data

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

diff --git a/kfet/management/commands/loadkfetdevdata.py b/kfet/management/commands/loadkfetdevdata.py
index 8eee85a9..5c2533b8 100644
--- a/kfet/management/commands/loadkfetdevdata.py
+++ b/kfet/management/commands/loadkfetdevdata.py
@@ -12,7 +12,8 @@ from django.core.management import call_command
 
 from gestioncof.management.base import MyBaseCommand
 from gestioncof.models import CofProfile
-from kfet.models import (Account, Checkout, CheckoutStatement)
+from kfet.models import (Account, Checkout, CheckoutStatement, Supplier,
+                         SupplierArticle, Article)
 
 # Où sont stockés les fichiers json
 DATA_DIR = os.path.join(os.path.dirname(os.path.dirname(__file__)),
@@ -125,8 +126,26 @@ class Command(MyBaseCommand):
                 amount_error=0
             )
 
+        # ---
+        # Fournisseur
+        # ---
+
+        supplier, created = Supplier.objects.get_or_create(name="Panoramix")
+        if created:
+            articles = Article.objects.filter(category__name="Autres Bieres")
+            if not articles.exists():
+                articles = random.sample(Article.objects.all(), 30)
+            to_create = []
+            for article in articles:
+                to_create.append(SupplierArticle(
+                    supplier=supplier,
+                    article=article
+                ))
+
+            SupplierArticle.objects.bulk_create(to_create)
+
         # ---
         # Opérations
         # ---
 
-        call_command('createopes','100', '7')
+        call_command('createopes', '100', '7', '--transfers=20')