initial commit

7f1e0d1d · Anton Sarukhanov · 7f1e0d1d · 7f1e0d1d
Commit 7f1e0d1d authored 8 years ago by Anton Sarukhanov
--- a/README.md
+++ b/README.md
+# Wooify
+
+Wooify is a script to prepare eCommerce product data for import into the
+[Woo Import Export](https://codecanyon.net/item/woo-import-export/13694764)
+WordPress plugin.
+
+## Supported Sources
+
+* [Ecwid](https://support.ecwid.com/hc/en-us/articles/207099979-Import-Export)
+
--- a/wooify
+++ b/wooify
+#!/usr/bin/env python3
+
+"""
+Prepare eCommerce product data for import into the
+"Woo Import Export" WordPress plugin.
+
+Supported sources:
+
+    - Ecwid
+"""
+
+import sys
+import csv
+import re
+import json
+import datetime
+from xml.etree import ElementTree
+
+WOO_IMP_EXP_FIELDS = [
+#   ("FieldName", "Default Value"),
+    ("Id", ""),
+    ("Product Name", ""),
+    ("Created Date", datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")),
+    ("Description", ""),
+    ("Product Type", "simple"),
+    ("Categories", json.dumps([])),
+    ("Price", 0),
+    ("Short Description", ""),
+    ("Product Status", "publish"),
+    ("Permalink", ""),
+    ("Tags", ""),
+    ("SKU", ""),
+    ("Sale Price", ""),
+    ("Visibility", "visible"),
+    ("On Sale", "no"),
+    ("Stock Status", "instock"),
+    ("Regular Price", 0),
+    ("Total Sales", 0),
+    ("Downloadable", "no"),
+    ("Virtual", "no"),
+    ("Purchase Note", ""),
+    ("Weight", ""),
+    ("Length", ""),
+    ("Width", ""),
+    ("Height", ""),
+    ("Unit", "in"),
+    ("Sold Individually", "yes"),
+    ("Manage Stock", "no"),
+    ("Stock", ""),
+    ("Backorders Allowed", "false"),
+    ("Backorders", "no"),
+    ("Purchaseable", "yes"),
+    ("Featured", "no"),
+    ("Is Taxable", "yes"),
+    ("Tax Status", "taxable"),
+    ("Tax Class", ""),
+    ("Product Images", "false"),
+    ("Product Image Set", "no"),
+    ("Download Limit", ""),
+    ("Download Expiry", ""),
+    ("Downloadable Files", ""),
+    ("Download Type", ""),
+    ("Product URL", ""),
+    ("Button Text", ""),
+    ("Shipping Required", "yes"),
+    ("Shipping Taxable", "yes"),
+    ("Shipping Class", ""),
+    ("Shipping Class Id", ""),
+    ("Average Rating", 0),
+    ("Rating Count", 0),
+    ("Related Ids", ""),
+    ("Upsell Ids", ""),
+    ("Cross Sell Ids", ""),
+    ("Attributes", ""),
+    ("Custom Fields", ""),
+    ("Product Parent id", 0),
+    ("Variation Description", ""),
+    ("Menu Order", 0),
+    ("Comment Status", "closed"),
+    ("Ping Status", "open")
+]
+
+def strip_html(html_string):
+    """Remove all HTML tags from a string."""
+    return ''.join(ElementTree.fromstring("<body>{0}</body>".format(html_string)).itertext())
+
+def ecwid_parser(r, r_id = None):
+    alphanum = re.compile('[\W_]+')
+    try:
+        r["description"] = strip_html(r["description"])
+    except ElementTree.ParseError:
+        print("Malformed HTML: "
+              "{0} ({1})".format(r["name"], (r["sku"])))
+    woo_row = dict(WOO_IMP_EXP_FIELDS)
+    woo_row.update({
+        "Id": r["product_id"],
+        "Product Name": r["name"],
+        "Categories": json.dumps(list(filter(None, [
+            {"name": r["category{0}".format(c)],
+             "slug": alphanum.sub('', r["category{0}".format(c)]).lower()}
+            if r["category{0}".format(c)] else None for c in range(1,3)]))),
+        "SKU": r["sku"],
+        "Weight": r["weight"],
+        "Product Images": r["image"],
+        "Product Image Set": "yes" if r["image"] else "no",
+        "Short Description": r["seo_description"],
+        "Description": r["description"],
+        "Price": r["price"],
+        "Regular Price": r["recommended_price"] or r["price"],
+        "Visibility": "visible" if r["enabled"] == "yes" else "invisible",
+        "Purchaseable": r["enabled"] or "no",
+    })
+    return woo_row
+
+def make_woocommerce_csv(input_file, output_file):
+    """Convert input_file, a CSV of eCommerce products, into
+       a format that can be imported by WooCommerce."""
+    count = 0
+    with open(input_file) as csv_in, open(output_file, "w") as csv_out:
+        reader = csv.DictReader(csv_in, delimiter=';')
+        writer = csv.DictWriter(csv_out, fieldnames=[f for f, d in WOO_IMP_EXP_FIELDS])
+        writer.writeheader()
+        for row in reader:
+            row = ecwid_parser(row)
+            writer.writerow(row)
+            count += 1
+    return count
+
+
+if __name__ == '__main__':
+    try:
+        output_file = sys.argv[2]
+    except IndexError:
+        output_file = 'woocommerce_products.csv'
+
+    try:
+        input_file = sys.argv[1]
+    except IndexError:
+        print("\n  Usage: {0} input_file [output_file]\n".format(sys.argv[0]))
+        sys.exit()
+
+    make_woocommerce_csv(input_file, output_file)