add create product image

see saleor/saleor#5337

Author: AdamQuadmon

README.md

@@ -14,6 +14,7 @@ As of now, `saleor-gql-loader` allows to create the following entities:
 - [x] category
 - [x] product
 - [x] product_variant
+- [x] product_image
 
 PR for supporting more graphQL mutations and/or queries are more than welcome.
 
@@ -28,7 +29,7 @@ team/community._
 using Pypi:
 
 ```bash
-pip install saleor-gql-loader
+pip install saleor-gql-loader requests-toolbelt Django
 ```
 
 Or cloning the repo:

saleor_gql_loader/data_loader.py

@@ -10,7 +10,7 @@
 project for easier testing.
 
 """
-from .utils import graphql_request, override_dict, handle_errors
+from .utils import graphql_request, graphql_multipart_request, override_dict, handle_errors, get_payload
 
 
 class ETLDataLoader:
@@ -517,3 +517,33 @@ def create_product_variant(self, product_id, **kwargs):
         handle_errors(errors)
 
         return response["data"]["productVariantCreate"]["productVariant"]["id"]
+
+    def create_product_image(self, product_id, file_path):
+        """create a product image.
+
+        Parameters
+        ----------
+        product_id : str
+            id for which the product image will be created.
+        file_path : str
+            path to the image to upload.
+
+        Returns
+        -------
+        id : str
+            the id of the product image created.
+
+        Raises
+        ------
+        Exception
+            when productErrors is not an empty list.
+        """
+        body = get_payload(product_id, file_path)
+
+        response = graphql_multipart_request(
+            body, self.headers, self.endpoint_url)
+
+        errors = response["data"]["productImageCreate"]["productErrors"]
+        handle_errors(errors)
+
+        return response["data"]["productImageCreate"]["image"]["id"]

saleor_gql_loader/utils.py

@@ -7,6 +7,9 @@
 """
 import requests
 import json
+from pathlib import Path
+from requests_toolbelt import MultipartEncoder
+from django.core.serializers.json import DjangoJSONEncoder
 
 GQL_DEFAULT_ENDPOINT = "http://localhost:8000/graphql/"
 
@@ -55,6 +58,45 @@ def graphql_request(query, variables={}, headers={},
         return parsed_response
 
 
+def graphql_multipart_request(body, headers, endpoint=GQL_DEFAULT_ENDPOINT):
+    """Execute a multipart graphQL query with `body` provided on the `endpoint`.
+
+    Parameters
+    ----------
+    body : str
+        payloads of graphQL query.
+    headers : dict, optional
+        headers added to the request (important for authentication).
+    endpoint : str, optional
+        the graphQL endpoint url that will be queried, default is
+        `GQL_DEFAULT_ENDPOINT`.
+
+    Returns
+    -------
+    response : dict
+        a dictionary corresponding to the parsed JSON graphQL response.
+
+    Raises
+    ------
+    Exception
+        when `response.status_code` is not 200.
+    """
+    bodyEncoder = MultipartEncoder(body)
+    base_headers = {
+        "Content-Type": bodyEncoder.content_type,
+    }
+    override_dict(headers, base_headers)
+
+    response = requests.post(endpoint, data=bodyEncoder, headers=headers, timeout=90)
+
+    parsed_response = json.loads(response.text)
+    if response.status_code != 200:
+        raise Exception("{message}\n extensions: {extensions}".format(
+            **parsed_response["errors"][0]))
+    else:
+        return parsed_response
+
+
 def override_dict(a, overrides):
     """Override a dict with another one **only first non nested keys**.
 
@@ -99,3 +141,86 @@ def handle_errors(errors):
         txt_list = [
             "{field} : {message}".format(**error) for error in errors]
         raise Exception("\n".join(txt_list))
+
+def get_operations(product_id):
+    """Get ProductImageCreate operations
+
+    Parameters
+    ----------
+    product_id : str
+            id for which the product image will be created.
+
+    Returns
+    -------
+    query : str
+    variables: dict
+    """
+    query = """mutation ProductImageCreate($product: ID!, $image: Upload!, $alt: String) {
+                    productImageCreate(input: {alt: $alt, image: $image, product: $product}) {
+                    image{
+                        id
+                      }
+                    productErrors {
+                        field
+                        message
+                    }
+                    }
+                }"""
+    variables = {
+        "product": product_id,
+        "image": "0",
+        "alt": ''
+    }
+    return {"query": query, "variables": variables}
+
+def get_operations(product_id):
+    """Get ProductImageCreate operations
+
+    Parameters
+    ----------
+    product_id : str
+            id for which the product image will be created.
+
+    Returns
+    -------
+    query : str
+    variables: dict
+    """
+    query = """mutation ProductImageCreate($product: ID!, $image: Upload!, $alt: String) {
+                  productImageCreate(input: {alt: $alt, image: $image, product: $product}) {
+                      image{
+                        id
+                      }
+                      productErrors {
+                          field
+                          message
+                      }
+                    }
+                }"""
+    variables = {
+        "product": product_id,
+        "image": "0",
+        "alt": ''
+    }
+    return {"query": query, "variables": variables}
+
+def get_payload(product_id, file_path):
+    """Get ProductImageCreate operations
+
+    Parameters
+    ----------
+    product_id : str
+            id for which the product image will be created.
+
+    Returns
+    -------
+    query : str
+    variables: dict
+    """
+    return {
+        "operations": json.dumps(
+            get_operations(product_id), cls=DjangoJSONEncoder
+        ),
+        "map": json.dumps({'0': ["variables.image"]}, cls=DjangoJSONEncoder),
+        "0": (Path(file_path).name, open(file_path, 'rb'), 'image/png')
+    }