From 42097c310234fef70a0dfd2b49ee287ba9ce41d4 Mon Sep 17 00:00:00 2001
From: Sam <samuel.molina@enreda.coop>
Date: Wed, 17 Feb 2021 12:26:40 +0000
Subject: [PATCH] importing woocommerce products is working

---
 companies/views.py   | 33 +++++++++++++++++++++++----------
 utils/woocommerce.py | 34 +++++++++++++++++++++++++---------
 2 files changed, 48 insertions(+), 19 deletions(-)

diff --git a/companies/views.py b/companies/views.py
index a1d2923..18634a5 100644
--- a/companies/views.py
+++ b/companies/views.py
@@ -19,6 +19,8 @@ from companies.serializers import CompanySerializer
 from utils.tag_filters import CompanyTagFilter
 from back_latienda.permissions import IsCreator
 
+from utils import woocommerce
+
 
 class CompanyViewSet(viewsets.ModelViewSet):
       queryset = Company.objects.all()
@@ -120,23 +122,34 @@ class CompanyViewSet(viewsets.ModelViewSet):
             except Exception as e:
                   return Response({"errors":{"details": str(e),}}, status=500)
 
-      @action(detail=True, methods=['POST', ])
+      @action(detail=True, methods=['GET', ])
       def import_products(self, request, **kwargs):
             instance = self.queryset.filter(pk=kwargs['pk']).first()
             # check if it's a shop
             if instance.shop is not True:
-                  return Response('This company is not a shop')
-            # check what platform
-            platform = instance.platform
-            if platform is None:
-                  return Response('This company is not registered with any platforms')
+                  return Response({'error': 'This company is not a shop'})
             # check required credentials
             credentials = instance.credentials
             if credentials is None or credentials == {}:
-                  return Response('This company has no registered credentials')
-            # execute import
-
-            return Response()
+                  return Response({'error': 'This company has no registered credentials'})
+            # check what platform
+            platform = instance.platform
+            if platform is None:
+                  message = {'error': 'This company is not registered with any platforms'}
+            elif platform == 'WOO_COMMERCE':
+                  # recheck credentials
+                  if 'key' in credentials.keys() and 'secret' in credentials.keys():
+                        # execute import
+                        products = woocommerce.migrate_shop_products(
+                              instance.web_link,
+                              credentials['key'],
+                              credentials['secret'])
+                        message = {'details': f'{len(products)} products added for {instance.company_name}'}
+                  else:
+                        message = {"error": 'Credentials have wrong format'}
+            else:
+                  message = {'error': f'Platform {plaform} not registered'}
+            return Response(message)
 
 
 @api_view(['GET',])
diff --git a/utils/woocommerce.py b/utils/woocommerce.py
index 40ab6b6..c341176 100644
--- a/utils/woocommerce.py
+++ b/utils/woocommerce.py
@@ -67,24 +67,40 @@ def migrate_shop_products(url, key, secret, version="wc/v3"):
                 instance_data[key] = product[key]
         # remove unwanted fields
         instance_data.pop('id')
-        if instance_data.get('tags') == []:
-            instance_data.pop('tags')
-        if instance_data.get('attributes') == []:
-            instance_data.pop('attributes')
+        # extract m2m field data
+        tags = instance_data.pop('tags')
+        attributes = instance_data.pop('attributes')
+
         # create instance with serializer
+        '''
         serializer = ProductSerializer(data=instance_data)
         if serializer.is_valid():
             new = serializer.save()
+            if tags:
+                new.tags.set(tags)
+            if attributes:
+                new.attributes.set(attributes)
+            new.save()
         else:
             logging.error(f"{serializer.errors}")
             continue
         '''
         # alternative method
-        try:
-            new = Product.objects.create(**instance_data)
-        except Exception as e:
-            logging.error(f"Could not create product instance: {str(e)}")
-        '''
+        serializer = ProductSerializer(data=instance_data)
+        if serializer.is_valid():
+            try:
+                new = Product.objects.create(**serializer.validated_data)
+                if tags:
+                    new.tags.set(tags)
+                if attributes:
+                    new.attributes.set(attributes)
+                new.save()
+            except Exception as e:
+                logging.error(f"Could not create product instance: {str(e)}")
+        else:
+            logging.error(f"{serializer.errors}")
+            continue
+
         products_created.append(new)
         counter += 1
         logging.info(f"Product {instance_data.get('name')} created")