new hledger-balance-csv and hledger-register-csv examples in extra, generating CSV output

extra/hledger-balance-csv.hs

@@ -0,0 +1,32 @@
+#!/usr/bin/env runhaskell
+{-|
+hledger-balance-csv [OPTIONS] [ARGS]
+
+Show a balance report as CSV.
+-}
+
+import Hledger
+import Hledger.Cli
+import System.Console.CmdArgs.Explicit
+import Text.CSV
+
+
+argsmode = 
+  (defCommandMode ["balance-csv"]) {
+     modeHelp = "show matched postings accounts and their balances as CSV"
+    ,modeGroupFlags = Group {
+       groupNamed = []
+      ,groupUnnamed = inputflags ++ reportflags ++ helpflags
+      ,groupHidden = []
+      }
+    }
+
+main = getCliOpts argsmode >>= printBalanceCsv
+
+printBalanceCsv opts = withJournalDo opts $
+  \CliOpts{reportopts_=ropts} j -> do
+    d <- getCurrentDay
+    let (items,_) = accountsReport ropts (queryFromOpts d ropts) j
+    putStrLn $ printCSV $
+      ["account","balance"] :
+      [[a, showMixedAmountWithoutPrice b] | (a, _, _, b) <- items]

extra/hledger-register-csv.hs

@@ -0,0 +1,57 @@
+#!/usr/bin/env runhaskell
+{-|
+hledger-register-csv [OPTIONS] [ARGS]
+
+Show a register report as CSV.
+-}
+
+module Main
+where
+
+import Hledger
+import Hledger.Cli
+import System.Console.CmdArgs.Explicit
+import Text.CSV
+
+
+argsmode :: Mode RawOpts
+argsmode = (defCommandMode ["register-csv"]) {
+  modeHelp = "show matched postings and running total as CSV"
+ ,modeGroupFlags = Group {
+     groupNamed = [
+         ("Input",inputflags)
+        ,("Reporting",reportflags)
+        ,("Misc",helpflags)
+        ]
+    ,groupUnnamed = []
+    ,groupHidden = []
+    }
+ }
+
+main :: IO ()
+main = do
+  opts <- getCliOpts argsmode
+  withJournalDo opts $
+    \CliOpts{reportopts_=ropts} j -> do
+      d <- getCurrentDay
+      putStrLn $ printCSV $ postingsReportAsCsv $ postingsReport ropts (queryFromOpts d ropts) j
+    
+postingsReportAsCsv :: PostingsReport -> CSV
+postingsReportAsCsv (_,is) =
+  ["date","description","account","amount","balance"]
+  :
+  map postingsReportItemAsCsvRecord is
+
+postingsReportItemAsCsvRecord :: PostingsReportItem -> Record
+postingsReportItemAsCsvRecord (_, _, p, b) = [date,desc,acct,amt,bal]
+  where
+    date = showDate $ postingDate p
+    desc = maybe "" tdescription $ ptransaction p
+    acct = bracket $ paccount p
+      where
+        bracket = case ptype p of
+                             BalancedVirtualPosting -> (\s -> "["++s++"]")
+                             VirtualPosting -> (\s -> "("++s++")")
+                             _ -> id
+    amt = showMixedAmountWithoutPrice $ pamount p
+    bal = showMixedAmountWithoutPrice b