Add a second arity to tree-grid to specify column order

Author: matthewdowney

src/excel_clj/core.clj

@@ -163,12 +163,15 @@
 (defn tree-grid
   "Build a lazy sheet grid from `tree`, whose leaves are shaped key->number.
 
-  E.g. (tree {:assets {:cash {:usd 100 :eur 100}}})
+  E.g. (tree-grid {:assets {:cash {:usd 100 :eur 100}}})
 
   See the comment form below this definition for examples."
-  [tree]
-  (let [ks (into [""] (keys (tree/fold + tree)))]
-    (table-grid ks (tree->rows tree))))
+  ([tree]
+   (let [ks (into [""] (keys (tree/fold + tree)))]
+     (tree-grid ks tree)))
+  ([ks tree]
+   (let [ks (into [""] (remove #{""}) ks)] ;; force the "" col to come first
+     (table-grid ks (tree->rows tree)))))
 
 
 (comment

test/excel_clj/core_test.clj

@@ -41,6 +41,7 @@
   (let [temp-file (io/file (temp ".xlsx"))]
     (try
       (testing "Example code snippet writes successfully."
+        (println "Writing example workbook...")
         (write! example-workbook-data temp-file))
       (finally
         (io/delete-file temp-file)))))
@@ -50,6 +51,7 @@
   (let [temp-file (io/file (temp ".xlsx"))]
     (try
       (testing "Example code snippet writes successfully."
+        (println "Writing example template...")
         (let [template (clojure.java.io/resource "uptime-template.xlsx")
               new-data {"raw" (table-grid example-template-data)}]
           (append! new-data template "filled-in-template.xlsx")))