diff options
author | xentac | 2016-11-21 14:04:08 -0800 |
---|---|---|
committer | GitHub | 2016-11-21 14:04:08 -0800 |
commit | d337960f109ec6f53fb0d449a6e0d10c690d2b30 (patch) | |
tree | bba96bd7fd79bbca27c1258be94a39b518397218 /rplugin/python3/deoplete/sources | |
parent | 391df18ca139a611864bfea11e203976f16494fc (diff) | |
parent | 81e9f05e3a1eac547813b7d80bf7417f38be8346 (diff) | |
download | vim-beancount-d337960f109ec6f53fb0d449a6e0d10c690d2b30.tar.gz |
Merge pull request #31 from yagebu/master
add deoplete source
Diffstat (limited to 'rplugin/python3/deoplete/sources')
-rw-r--r-- | rplugin/python3/deoplete/sources/beancount.py | 75 |
1 files changed, 75 insertions, 0 deletions
diff --git a/rplugin/python3/deoplete/sources/beancount.py b/rplugin/python3/deoplete/sources/beancount.py new file mode 100644 index 0000000..f595a54 --- /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("beancount#get_root()")) + 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) |