diff --git a/wooify b/wooify
index 962c3bf7d23127baeb29118fca573b537779a32b..88254c7e105b8afab43a4341f0b78f50574601ba 100644
--- a/wooify
+++ b/wooify
@@ -17,7 +17,7 @@ import datetime
from xml.etree import ElementTree
WOO_IMP_EXP_FIELDS = [
-# ("FieldName", "Default Value"),
+ # ("FieldName", "Default Value"),
("Id", ""),
("Product Name", ""),
("Created Date", datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")),
@@ -80,11 +80,14 @@ WOO_IMP_EXP_FIELDS = [
("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())
+ return ''.join(ElementTree.fromstring("<body>{0}</body>"
+ .format(html_string)).itertext())
+
-def ecwid_parser(r, r_id = None):
+def ecwid_parser(r, r_id=None):
alphanum = re.compile('[\W_]+')
try:
r["description"] = strip_html(r["description"])
@@ -98,7 +101,7 @@ def ecwid_parser(r, r_id = None):
"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)]))),
+ if r["category{0}".format(c)] else None for c in range(1, 3)]))),
"SKU": r["sku"],
"Weight": r["weight"],
"Product Images": r["image"],
@@ -112,13 +115,15 @@ def ecwid_parser(r, r_id = None):
})
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 = csv.DictWriter(csv_out,
+ fieldnames=[f for f, d in WOO_IMP_EXP_FIELDS])
writer.writeheader()
for row in reader:
row = ecwid_parser(row)