From 3a9665dd45ad38bc98f18a091fe460f7a46ea4df Mon Sep 17 00:00:00 2001 From: Jakob Schnitzer Date: Sun, 20 Nov 2016 18:20:53 +0100 Subject: add deoplete source --- rplugin/python3/deoplete/sources/beancount.py | 75 +++++++++++++++++++++++++++ 1 file changed, 75 insertions(+) create mode 100644 rplugin/python3/deoplete/sources/beancount.py (limited to 'rplugin/python3/deoplete') diff --git a/rplugin/python3/deoplete/sources/beancount.py b/rplugin/python3/deoplete/sources/beancount.py new file mode 100644 index 0000000..4618105 --- /dev/null +++ b/rplugin/python3/deoplete/sources/beancount.py @@ -0,0 +1,75 @@ +import re + +from deoplete.source.base import Base + +try: + from beancount.loader import load_file + from beancount.core.data import Open, Transaction + HAS_BEANCOUNT = True +except ImportError: + HAS_BEANCOUNT = False + +DIRECTIVES = ['open', 'close', 'commodity', 'txn', 'balance', 'pad', 'note', + 'document', 'price', 'event', 'query', 'custom'] + + +class Source(Base): + def __init__(self, vim): + super().__init__(vim) + self.vim = vim + + self.name = 'beancount' + self.mark = '[bc]' + self.filetypes = ['beancount'] + self.min_pattern_length = 0 + + def on_init(self, context): + if not HAS_BEANCOUNT: + self.error('Importing beancount failed.') + + def on_event(self, context): + self.__make_cache(context) + + def get_complete_position(self, context): + m = re.search(r'\S*$', context['input']) + return m.start() if m else -1 + + def gather_candidates(self, context): + if re.match(r'^\d{4}[/-]\d\d[/-]\d\d \w*$', context['input']): + return DIRECTIVES + if not context['complete_str']: + return [] + first = context['complete_str'][0] + if first == '#': + return ['#' + w for w in self._tags] + elif first == '^': + return ['^' + w for w in self._links] + elif first == '"': + return ['"{}"'.format(w) for w in self._payees] + return self._accounts + + def __make_cache(self, context): + accounts = set() + links = set() + payees = set() + tags = set() + if HAS_BEANCOUNT: + entries, _, _ = load_file(self.vim.eval("expand('%')")) + else: + entries = [] + + for entry in entries: + if isinstance(entry, Open): + accounts.add(entry.account) + elif isinstance(entry, Transaction): + if entry.payee: + payees.add(entry.payee) + if hasattr(entry, 'links') and entry.links: + links.update(entry.links) + if hasattr(entry, 'tags') and entry.tags: + tags.update(entry.tags) + + self._accounts = sorted(accounts) + self._links = sorted(links) + self._payees = sorted(payees) + self._tags = sorted(tags) -- cgit v1.2.3-2-gb3c3 From 81e9f05e3a1eac547813b7d80bf7417f38be8346 Mon Sep 17 00:00:00 2001 From: Jakob Schnitzer Date: Mon, 21 Nov 2016 10:21:57 +0100 Subject: deoplete: use b:beancount_root --- rplugin/python3/deoplete/sources/beancount.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'rplugin/python3/deoplete') diff --git a/rplugin/python3/deoplete/sources/beancount.py b/rplugin/python3/deoplete/sources/beancount.py index 4618105..f595a54 100644 --- a/rplugin/python3/deoplete/sources/beancount.py +++ b/rplugin/python3/deoplete/sources/beancount.py @@ -54,7 +54,7 @@ class Source(Base): payees = set() tags = set() if HAS_BEANCOUNT: - entries, _, _ = load_file(self.vim.eval("expand('%')")) + entries, _, _ = load_file(self.vim.eval("beancount#get_root()")) else: entries = [] -- cgit v1.2.3-2-gb3c3