From c1c9f99fb56f09dc83a1e439e56e973062f72433 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=D0=92=D0=B8=D0=BA=D1=82=D0=BE=D1=80=20=D0=93=D0=BB=D0=B0?=
 =?UTF-8?q?=D0=B4=D0=BA=D0=B8=D1=85?= <v.gladkikh@spider.ru>
Date: Wed, 30 Oct 2019 18:04:30 +0300
Subject: [PATCH] Add product migrate data

---
 .../management/commands/add_product.py        | 54 +++++++++++++++++++
 apps/product/models.py                        | 18 +++++++
 apps/transfer/models.py                       |  1 +
 3 files changed, 73 insertions(+)

diff --git a/apps/product/management/commands/add_product.py b/apps/product/management/commands/add_product.py
index e69de29b..981487ec 100644
--- a/apps/product/management/commands/add_product.py
+++ b/apps/product/management/commands/add_product.py
@@ -0,0 +1,54 @@
+from django.core.management.base import BaseCommand
+from product.models import ProductType, ProductSubType
+
+
+def add_type():
+    product_type = ProductType.objects.create(
+        name={'"en-GB"': "Wine"},
+        index_name=ProductType.WINE
+    )
+    return product_type.save()
+
+
+def add_subtype(id_type):
+    subtypes = ProductSubType.objects.bulk_create([
+        ProductSubType(product_type=id_type,
+                       name={"en-GB", ProductSubType.EXTRA_BRUT},
+                       index_name=ProductSubType.EXTRA_BRUT),
+        ProductSubType(product_type=id_type,
+                       name={"en-GB", ProductSubType.BRUT},
+                       index_name=ProductSubType.BRUT),
+        ProductSubType(product_type=id_type,
+                       name={"en-GB", ProductSubType.BRUT_NATURE},
+                       index_name=ProductSubType.BRUT_NATURE),
+        ProductSubType(product_type=id_type,
+                       name={"en-GB", ProductSubType.DEMI_SEC},
+                       index_name=ProductSubType.DEMI_SEC),
+        ProductSubType(product_type=id_type,
+                       name={"en-GB", ProductSubType.EXTRA_DRY},
+                       index_name=ProductSubType.EXTRA_DRY),
+        ProductSubType(product_type=id_type,
+                       name={"en-GB", ProductSubType.DOSAGE_ZERO},
+                       index_name=ProductSubType.DOSAGE_ZERO),
+        ProductSubType(product_type=id_type,
+                       name={"en-GB", ProductSubType.SEC},
+                       index_name=ProductSubType.SEC),
+        ProductSubType(product_type=id_type,
+                       name={"en-GB", ProductSubType.SEC},
+                       index_name=ProductSubType.SEC),
+        ProductSubType(product_type=id_type,
+                       name={"en-GB", ProductSubType.MOELLEUX},
+                       index_name=ProductSubType.MOELLEUX),
+        ])
+
+
+class Command(BaseCommand):
+    help = 'Add product data'
+
+    def handle(self, *args, **kwarg):
+        product_type = add_type()
+        add_subtype(product_type.id)
+
+
+
+
diff --git a/apps/product/models.py b/apps/product/models.py
index d4011fa0..332a92f6 100644
--- a/apps/product/models.py
+++ b/apps/product/models.py
@@ -46,10 +46,28 @@ class ProductSubType(TranslatedFieldsMixin, ProjectBaseMixin):
     # INDEX NAME CHOICES
     RUM = 'rum'
     OTHER = 'other'
+    EXTRA_BRUT = 'extra brut'
+    BRUT = 'brut'
+    BRUT_NATURE = 'brut nature'
+    DEMI_SEC = 'demi-sec'
+    EXTRA_DRY = 'Extra Dry'
+    DOSAGE_ZERO = 'dosage zero'
+    SEC = 'sec'
+    DOUX = 'doux'
+    MOELLEUX=  'moelleux'
 
     INDEX_NAME_TYPES = (
         (RUM, _('Rum')),
         (OTHER, _('Other')),
+        (EXTRA_BRUT, _('extra brut')),
+        (BRUT, _('brut')),
+        (BRUT_NATURE, _('brut nature')),
+        (DEMI_SEC, _('demi-sec')),
+        (EXTRA_DRY, _('Extra Dry')),
+        (DOSAGE_ZERO, _('dosage zero')),
+        (SEC, _('sec')),
+        (DOUX, _('doux')),
+        (MOELLEUX, _('moelleux'))
     )
 
     product_type = models.ForeignKey(ProductType, on_delete=models.CASCADE,
diff --git a/apps/transfer/models.py b/apps/transfer/models.py
index fecbc48b..1a978596 100644
--- a/apps/transfer/models.py
+++ b/apps/transfer/models.py
@@ -842,6 +842,7 @@ class Ads(MigrateMixin):
         managed = False
         db_table = 'ads'
 
+
 class Products(models.Model):
     establishment = models.ForeignKey('Establishments', models.DO_NOTHING, blank=True, null=True)
     brand = models.CharField(max_length=255, blank=True, null=True)