Added new get_images() function

Author: flyandlure

.gitignore

@@ -2,4 +2,5 @@ venv
 .idea
 example.py
 parents.csv
-children.csv
+children.csv
+images.csv

README.md

@@ -27,5 +27,10 @@ children = scraper.get_variants(parents)
 children.to_csv('children.csv', index=False)
 print('Children: ', len(children))
 
+
+images = scraper.get_images(parents)
+images.to_csv('images.csv', index=False)
+print('Images: ', len(images))
+
 ```
 

setup.py

@@ -8,7 +8,7 @@
 setup(
     name='ShopifyScraper',
     packages=['shopify_scraper'],
-    version='0.001',
+    version='0.002',
     license='MIT',
     description='ShopifyScraper is a Python package that scrapes Shopify products and variants to Pandas dataframes.',
     long_description=long_description,

shopify_scraper/scraper.py

@@ -108,3 +108,35 @@ def get_variants(products):
     df_variants = df_variants.merge(df_parent_data, left_on='product_id', right_on='parent_id')
     return df_variants
 
+
+def json_list_to_df(df, col):
+    """Return a Pandas dataframe based on a column that contains a list of JSON objects.
+
+    Args:
+        df (Pandas dataframe): The dataframe to be flattened.
+        col (str): The name of the column that contains the JSON objects.
+
+    Returns:
+        Pandas dataframe: A new dataframe with the JSON objects expanded into columns.
+    """
+
+    rows = []
+    for index, row in df[col].iteritems():
+        for item in row:
+            rows.append(item)
+    df = pd.DataFrame(rows)
+    return df
+
+
+def get_images(df_products):
+    """Get images from a list of products.
+
+    Args:
+        df_products (pd.DataFrame): Pandas dataframe of products from get_products()
+
+    Returns:
+        images (pd.DataFrame): Pandas dataframe of images
+    """
+
+    return json_list_to_df(df_products, 'images')
+