simonmichael/hledger
1
1
Fork 0
mirror of https://github.com/simonmichael/hledger.git synced 2024-11-09 21:22:26 +03:00
Code Issues Projects Releases Wiki Activity

add: better handling of virtual postings and default amounts

Browse Source
This commit is contained in:
Simon Michael 2009-05-24 06:54:12 +00:00
parent 8cb526f655
commit 9b8f00b54e
1 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
Commands/Add.hs
View File

@ -96,7 +96,7 @@ getPostings bestmatchps enteredps = do
else do
(amountstr, eoi) <- askFor (printf "amount %d" n) defaultamount validateamount
let amount = fromparse $ parse (someamount <|> return missingamt) "" amountstr
let p = nullrawposting{paccount=account,pamount=amount}
let p = nullrawposting{paccount=stripbrackets account,pamount=amount,ptype=postingaccounttype account}
if eoi
then if null enteredps
then return ([], True)
@ -106,14 +106,21 @@ getPostings bestmatchps enteredps = do
else getPostings bestmatchps $ enteredps ++ [p]
where
n = length enteredps + 1
realn = length enteredrealps + 1
bestmatch | isNothing bestmatchps = Nothing
| n <= length ps = Just $ ps !! (n-1)
| otherwise = Nothing
where Just ps = bestmatchps
defaultaccount = maybe Nothing (Just . paccount) bestmatch
defaultaccount = maybe Nothing (Just . showacctname) bestmatch
showacctname p = showAccountName Nothing (ptype p) $ paccount p
validateaccount = Just $ \s -> not $ null s
defaultamount | n==1 = maybe Nothing (Just . show . pamount) bestmatch -- previously used amount
| otherwise = Just $ show $ negate $ sum $ map pamount enteredps -- balancing amount
defaultamount = maybe balancingamount (Just . show . pamount) bestmatch
where balancingamount = Just $ show $ negate $ sum $ map pamount enteredrealps
enteredrealps = filter isReal enteredps
postingaccounttype ('[':_) = BalancedVirtualPosting
postingaccounttype ('(':_) = VirtualPosting
postingaccounttype _ = RegularPosting
stripbrackets = dropWhile (`elem` "([") . reverse . dropWhile (`elem` "])") . reverse
validateamount = Just $ \s ->
(null s && (not $ null enteredps)) ||
(isRight $ parse (someamount>>many spacenonewline>>eof) "" s)