Added category export.

wooify

@@ -140,30 +140,42 @@ def ecwid_parser(r, r_id=None):
 def expand_category(name, cat_id, slugify=None, delimiter=None, products=None):
     """Build a rich category entry from minimal input."""
     slugify = slugify or UniqueSlugify()
+    clean_name = (name[name.rfind(delimiter) + len(delimiter):]
+                  if delimiter and delimiter in name else name)
     return {
         'Id': cat_id,
         'Term Taxonomy Id': cat_id,
-        'Name': (name[name.rfind(delimiter):]
-                 if delimiter and delimiter in name else name),
-        'Slug': slugify(name),
-        'Parent Raw Name': (name[:name.rfind(delimiter)]
-                            if delimiter and delimiter in name
-                            else ''),
+        'Name': clean_name,
+        'Slug': slugify(clean_name),
+        'Parent Name': (name[:name.rfind(delimiter)]
+                        if delimiter and delimiter in name
+                        else ''),
         'Count': (sum([1 for p in products if name in p['Categories']])
                   if products else 0),
     }
 
 
-def add_missing_parents(categories):
+def add_missing_parents(categories, products, delimiter, slugify=None):
     """Find categories whose parents do not exist, and recursively add
        those parents to the list."""
+    def split_category_name(name):
+        """Return the name of each of this categories ancestors, and itself."""
+        return [(name[:name.rfind(delimiter)]
+                 if delimiter and delimiter in name
+                 else '') for i in range(name.count(delimiter))] + [name]
+    slugify = slugify or UniqueSlugify()
+    categories_with_parents = categories.copy()
     for i, c in categories.items():
-        if not c['Parent Raw Name']:
-            continue
-        if c['Parent Raw Name'] in categories:
+        try:
+            for ancestor in split_category_name(c['Parent Name']):
+                if ancestor not in categories_with_parents:
+                    cat_id = len(categories_with_parents) + 1
+                    categories_with_parents[ancestor] = expand_category(
+                        ancestor, cat_id, slugify, delimiter,
+                        products)
+        except KeyError:
             continue
-        # TODO: this
-    return categories
+    return categories_with_parents
 
 
 def make_woocommerce_csv(input_file, products_file, categories_file=None):
@@ -185,26 +197,27 @@ def make_woocommerce_csv(input_file, products_file, categories_file=None):
             raw_category_names.extend(product['Categories'])
         raw_category_names = list(set(raw_category_names))  # Uniquify
         categories = {}
-        slugify = UniqueSlugify()
+        slugify = UniqueSlugify(to_lower=True)
         for idx, name in enumerate(raw_category_names):
             categories[name] = expand_category(name, idx, slugify,
                                                ECWID_CATEGORY_DELIMITER,
                                                products)
-        categories = add_missing_parents(categories)
+        categories = add_missing_parents(categories, products,
+                                         ECWID_CATEGORY_DELIMITER, slugify)
         for i, c in categories.items():
-            if c['Parent Raw Name']:
-                parent = categories[c['Parent Raw Name']]
+            if c['Parent Name']:
+                parent = categories[c['Parent Name']]
                 c['Parent Id'] = parent['Id']
                 c['Parent Slug'] = parent['Slug']
             else:
                 c['Parent Id'] = ''
                 c['Parent Slug'] = ''
-            del c['Parent Raw Name']
+            del c['Parent Name']
 
         for product in products:
             products_writer.writerow(product)
 
-        if csv_gategories:
+        if csv_categories:
             categories_writer = csv.DictWriter(
                 csv_categories,
                 fieldnames=[f for f, d in WOO_IMP_EXP_CATEGORY_FIELDS])