aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--autoload/beancount.vim61
1 files changed, 56 insertions, 5 deletions
diff --git a/autoload/beancount.vim b/autoload/beancount.vim
index 50a1b33..aea0a42 100644
--- a/autoload/beancount.vim
+++ b/autoload/beancount.vim
@@ -1,3 +1,12 @@
+function! s:UsingPython3()
+ if has('python3')
+ return 1
+ endif
+ return 0
+endfunction
+
+let s:using_python3 = s:UsingPython3()
+
" Equivalent to python's startswith
" Matches based on user's ignorecase preference
function! s:startswith(string, prefix)
@@ -69,6 +78,9 @@ function! beancount#complete(findstart, base)
return beancount#complete_basic(s:directives, a:base, '')
endif
+ " If we are using python3, now is a good time to load everything
+ call beancount#load_everything()
+
let l:two_tokens = searchpos('\S\+\s', "bn", line("."))[1]
let l:prev_token = strpart(getline("."), l:two_tokens, getpos(".")[2] - l:two_tokens)
" Match curriences if previous token is number
@@ -101,36 +113,75 @@ function! s:get_root()
return expand('%')
endfunction
+function! beancount#load_everything()
+ if s:using_python3
+ let l:root = s:get_root()
+python3 << EOF
+import vim
+from beancount import loader
+from beancount.core import data
+
+accounts = set()
+currencies = set()
+links = set()
+payees = set()
+tags = set()
+
+entries, errors, options_map = loader.load_file(vim.eval('l:root'))
+for index, entry in enumerate(entries):
+ if isinstance(entry, data.Open):
+ accounts.add(entry.account)
+ if entry.currencies:
+ currencies.update(entry.currencies)
+ elif isinstance(entry, data.Commodity):
+ currencies.add(entry.currency)
+ elif isinstance(entry, data.Transaction):
+ if entry.tags:
+ tags.update(entry.tags)
+ if entry.links:
+ links.update(entry.links)
+ if entry.payee:
+ payees.add(entry.payee)
+
+vim.command('let b:beancount_accounts = [{}]'.format(','.join(repr(x) for x in sorted(accounts))))
+vim.command('let b:beancount_currencies = [{}]'.format(','.join(repr(x) for x in sorted(currencies))))
+vim.command('let b:beancount_links = [{}]'.format(','.join(repr(x) for x in sorted(links))))
+vim.command('let b:beancount_payees = [{}]'.format(','.join(repr(x) for x in sorted(payees))))
+vim.command('let b:beancount_tags = [{}]'.format(','.join(repr(x) for x in sorted(tags))))
+EOF
+ endif
+endfunction
+
function! beancount#load_accounts()
- if !exists('b:beancount_accounts')
+ if !s:using_python3 && !exists('b:beancount_accounts')
let l:root = s:get_root()
let b:beancount_accounts = beancount#find_accounts(l:root)
endif
endfunction
function! beancount#load_tags()
- if !exists('b:beancount_tags')
+ if !s:using_python3 && !exists('b:beancount_tags')
let l:root = s:get_root()
let b:beancount_tags = beancount#find_tags(l:root)
endif
endfunction
function! beancount#load_links()
- if !exists('b:beancount_links')
+ if !s:using_python3 && !exists('b:beancount_links')
let l:root = s:get_root()
let b:beancount_links = beancount#find_links(l:root)
endif
endfunction
function! beancount#load_currencies()
- if !exists('b:beancount_currencies')
+ if !s:using_python3 && !exists('b:beancount_currencies')
let l:root = s:get_root()
let b:beancount_currencies = beancount#find_currencies(l:root)
endif
endfunction
function! beancount#load_payees()
- if !exists('b:beancount_payees')
+ if !s:using_python3 && !exists('b:beancount_payees')
let l:root = s:get_root()
let b:beancount_payees = beancount#find_payees(l:root)
endif