74512814ec
Make these modules' names more like the heavily-used types they define (CliOpts, UIOpts, WebOpts). This is consistent with RawOptions and ReportOptions, and helps with code navigation.
60 lines
1.8 KiB
Haskell
60 lines
1.8 KiB
Haskell
{-|
|
|
|
|
Print a histogram report. (The "activity" command).
|
|
|
|
-}
|
|
|
|
module Hledger.Cli.Histogram
|
|
where
|
|
|
|
import Data.List
|
|
import Data.Maybe
|
|
import Data.Ord
|
|
import System.Console.CmdArgs.Explicit
|
|
import Text.Printf
|
|
|
|
import Hledger
|
|
import Hledger.Cli.CliOptions
|
|
import Prelude hiding (putStr)
|
|
import Hledger.Utils.UTF8IOCompat (putStr)
|
|
|
|
activitymode :: Mode RawOpts
|
|
activitymode = (defCommandMode $ ["activity"] ++ aliases) {
|
|
modeHelp = "show an ascii barchart of posting counts per interval (default: daily)" `withAliases` aliases
|
|
,modeHelpSuffix = []
|
|
,modeGroupFlags = Group {
|
|
groupUnnamed = []
|
|
,groupHidden = []
|
|
,groupNamed = [generalflagsgroup1]
|
|
}
|
|
}
|
|
where aliases = []
|
|
|
|
barchar :: Char
|
|
barchar = '*'
|
|
|
|
-- | Print a histogram of some statistic per reporting interval, such as
|
|
-- number of postings per day.
|
|
histogram :: CliOpts -> Journal -> IO ()
|
|
histogram CliOpts{reportopts_=ropts} j = do
|
|
d <- getCurrentDay
|
|
putStr $ showHistogram ropts (queryFromOpts d ropts) j
|
|
|
|
showHistogram :: ReportOpts -> Query -> Journal -> String
|
|
showHistogram opts q j = concatMap (printDayWith countBar) spanps
|
|
where
|
|
i = intervalFromOpts opts
|
|
interval | i == NoInterval = Days 1
|
|
| otherwise = i
|
|
span' = queryDateSpan (date2_ opts) q `spanDefaultsFrom` journalDateSpan (date2_ opts) j
|
|
spans = filter (DateSpan Nothing Nothing /=) $ splitSpan interval span'
|
|
spanps = [(s, filter (isPostingInDateSpan s) ps) | s <- spans]
|
|
-- same as Register
|
|
-- should count transactions, not postings ?
|
|
-- ps = sortBy (comparing postingDate) $ filterempties $ filter matchapats $ filterdepth $ journalPostings j
|
|
ps = sortBy (comparing postingDate) $ filter (q `matchesPosting`) $ journalPostings j
|
|
|
|
printDayWith f (DateSpan b _, ps) = printf "%s %s\n" (show $ fromJust b) (f ps)
|
|
|
|
countBar ps = replicate (length ps) barchar
|