Move Account() to Accounts().MustGet()

lib/journal/performance/performance_test.go

@@ -22,12 +22,12 @@ func TestComputeFlows(t *testing.T) {
 	usd := ctx.Commodity("USD")
 	gbp := ctx.Commodity("GBP")
 	aapl := ctx.Commodity("AAPL")
-	portfolio := ctx.Account("Assets:Portfolio")
-	acc1 := ctx.Account("Assets:Acc1")
-	acc2 := ctx.Account("Assets:Acc2")
-	dividend := ctx.Account("Income:Dividends")
-	expense := ctx.Account("Expenses:Investments")
-	equity := ctx.Account("Equity:Equity")
+	portfolio := ctx.Accounts().MustGet("Assets:Portfolio")
+	acc1 := ctx.Accounts().MustGet("Assets:Acc1")
+	acc2 := ctx.Accounts().MustGet("Assets:Acc2")
+	dividend := ctx.Accounts().MustGet("Income:Dividends")
+	expense := ctx.Accounts().MustGet("Expenses:Investments")
+	equity := ctx.Accounts().MustGet("Equity:Equity")
 
 	chf.IsCurrency = true
 	usd.IsCurrency = true

lib/journal/process.go

@@ -224,7 +224,7 @@ func CloseAccounts(j *Journal, enable bool, partition date.Partition) DayFn {
 		// j.Day creates the entry for the given date as a side effect.
 		closingDays.Add(j.Day(d))
 	}
-	equityAccount := j.Registry.Account("Equity:Equity")
+	equityAccount := j.Registry.Accounts().MustGet("Equity:Equity")
 	return func(d *Day) error {
 		if closingDays.Has(d) {
 			for k, quantity := range quantities {

lib/model/account/registry.go

@@ -107,6 +107,14 @@ func (as *Registry) Create(a syntax.Account) (*Account, error) {
 	return as.Get(a.Extract())
 }
 
+func (as *Registry) MustGet(name string) *Account {
+	a, err := as.Get(name)
+	if err != nil {
+		panic(err)
+	}
+	return a
+}
+
 func isValidSegment(s string) bool {
 	if len(s) == 0 {
 		return false

lib/model/registry/registry.go

@@ -44,15 +44,6 @@ func (reg Registry) GetAccount(name string) (*Account, error) {
 	return reg.accounts.Get(name)
 }
 
-// Account returns a commodity or panics.
-func (reg Registry) Account(name string) *Account {
-	c, err := reg.GetAccount(name)
-	if err != nil {
-		panic(err)
-	}
-	return c
-}
-
 // GetCommodity returns a commodity.
 func (reg Registry) GetCommodity(name string) (*Commodity, error) {
 	return reg.commodities.Get(name)
@@ -69,15 +60,15 @@ func (reg Registry) Commodity(name string) *Commodity {
 
 // TBDAccount returns the TBD account.
 func (reg Registry) TBDAccount() *Account {
-	return reg.Account("Expenses:TBD")
+	return reg.Accounts().MustGet("Expenses:TBD")
 }
 
 // ValuationAccountFor returns the valuation account which corresponds to
 // the given Asset or Liability account.
 func (reg Registry) ValuationAccountFor(a *Account) *Account {
 	suffix := a.Split()[1:]
-	segments := append(reg.Account("Income").Split(), suffix...)
-	return reg.Account(strings.Join(segments, ":"))
+	segments := append(reg.Accounts().MustGet("Income").Split(), suffix...)
+	return reg.Accounts().MustGet(strings.Join(segments, ":"))
 }
 
 // Accounts returns the accounts.