| """String dispatch class to match regexps and dispatch commands. |
| """ |
|
|
| |
| import re |
|
|
| |
| from IPython.core.hooks import CommandChainDispatcher |
|
|
| |
| class StrDispatch(object): |
| """Dispatch (lookup) a set of strings / regexps for match. |
| |
| Example: |
| |
| >>> dis = StrDispatch() |
| >>> dis.add_s('hei',34, priority = 4) |
| >>> dis.add_s('hei',123, priority = 2) |
| >>> dis.add_re('h.i', 686) |
| >>> print(list(dis.flat_matches('hei'))) |
| [123, 34, 686] |
| """ |
|
|
| def __init__(self): |
| self.strs = {} |
| self.regexs = {} |
|
|
| def add_s(self, s, obj, priority= 0 ): |
| """ Adds a target 'string' for dispatching """ |
|
|
| chain = self.strs.get(s, CommandChainDispatcher()) |
| chain.add(obj,priority) |
| self.strs[s] = chain |
|
|
| def add_re(self, regex, obj, priority= 0 ): |
| """ Adds a target regexp for dispatching """ |
|
|
| chain = self.regexs.get(regex, CommandChainDispatcher()) |
| chain.add(obj,priority) |
| self.regexs[regex] = chain |
|
|
| def dispatch(self, key): |
| """ Get a seq of Commandchain objects that match key """ |
| if key in self.strs: |
| yield self.strs[key] |
|
|
| for r, obj in self.regexs.items(): |
| if re.match(r, key): |
| yield obj |
| else: |
| |
| pass |
|
|
| def __repr__(self): |
| return "<Strdispatch %s, %s>" % (self.strs, self.regexs) |
|
|
| def s_matches(self, key): |
| if key not in self.strs: |
| return |
| for el in self.strs[key]: |
| yield el[1] |
|
|
| def flat_matches(self, key): |
| """ Yield all 'value' targets, without priority """ |
| for val in self.dispatch(key): |
| for el in val: |
| yield el[1] |
| return |
|
|
