/
registry.py
81 lines (69 loc) · 2.27 KB
/
registry.py
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
69
70
71
72
73
74
75
76
77
78
79
80
81
# -*- coding: utf-8 -*-
# <Lettuce - Behaviour Driven Development for python>
# Copyright (C) <2010> Gabriel Falcão <gabriel@nacaolivre.org>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
import threading
world = threading.local()
world._set = False
class CleanableDict(dict):
def clear(self):
for k in self.keys():
del self[k]
class CallbackDict(CleanableDict):
def _function_matches(self, one, other):
params = 'co_filename', 'co_firstlineno'
matches = []
for param in params:
one_got = getattr(one.func_code, param)
other_got = getattr(other.func_code, param)
matches.append(one_got == other_got)
return all(matches)
def append_to(self, where, when, function):
found = False
for other_function in self[where][when]:
if self._function_matches(other_function,function):
found = True
if not found:
self[where][when].append(function)
def clear(self):
for action_dict in self.values():
for callback_list in action_dict.values():
while callback_list:
callback_list.pop()
STEP_REGISTRY = CleanableDict()
CALLBACK_REGISTRY = CallbackDict(
{
'all': {
'before': [],
'after': []
},
'step': {
'before_each': [],
'after_each': []
},
'scenario': {
'before_each': [],
'after_each': [],
'outline': []
},
'feature': {
'before_each': [],
'after_each': []
}
}
)
def clear():
STEP_REGISTRY.clear()
CALLBACK_REGISTRY.clear()