FilesExpand file tree

 main

/

strdispatch.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

68 lines (53 loc) · 1.79 KB

 main

/

strdispatch.py

Copy path

Top

File metadata and controls

• Code
• Blame

68 lines (53 loc) · 1.79 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

"""String dispatch class to match regexps and dispatch commands.

"""

# Stdlib imports

import re

# Our own modules

from IPython.core.hooks import CommandChainDispatcher

# Code begins

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:

#print "nomatch",key # dbg

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] # only value, no priority

return