Coverage for src/robotide/namespace/local_namespace.py: 86%

1# Copyright 2008-2015 Nokia Networks 

2# Copyright 2016- Robot Framework Foundation 

3# 

4# Licensed under the Apache License, Version 2.0 (the "License"); 

5# you may not use this file except in compliance with the License. 

6# You may obtain a copy of the License at 

7# 

8# http://www.apache.org/licenses/LICENSE-2.0 

9# 

10# Unless required by applicable law or agreed to in writing, software 

11# distributed under the License is distributed on an "AS IS" BASIS, 

12# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 

13# See the License for the specific language governing permissions and 

14# limitations under the License. 

15 

16from robotide import utils 2 ctx1df

17from robotide.spec.iteminfo import LocalVariableInfo 2 ctx1df

18 

19 

20def local_namespace(controller, namespace, row=None): 2 ctx1df

21 if row is not None: # can be 0! 27 ctx1dophqlijrsmtuvwxkzAygnbBcea

22 return LocalRowNamespace(controller, namespace, row) 22 ctx1ophqlijrsmtuvwxkgnbcea

23 return LocalMacroNamespace(controller, namespace) 5 ctx1dzAyB

24 

25 

26class LocalMacroNamespace(object): 2 ctx1df

27 

28 def __init__(self, controller, namespace): 2 ctx1df

29 self._controller = controller 27 ctx1dophqlijrsmtuvwxkzAygnbBcea

30 self.namespace = namespace 27 ctx1dophqlijrsmtuvwxkzAygnbBcea

31 

32 def get_suggestions(self, start): 2 ctx1df

33 # print(f"DEBUG: local_namespace.py LocalMacroNamespace get_suggestions ENTER start={start}") 

34 return self.namespace.get_suggestions_for(self._controller, start) 5 ctx1zbcea

35 

36 def has_name(self, value): 2 ctx1df

37 for sug in self.namespace.get_suggestions_for(self._controller, value): 20 ctx1ophqlijrsmtuvwxkAygn

38 if sug.name == value: 38 ↛ 40line 38 didn't jump to line 40 because the condition on line 38 was always true4 ctx1hjyg

39 return True 4 ctx1hjyg

40 try: 

41 if value in sug.assign: 

42 return True 

43 except AttributeError: 

44 pass 

45 return False 18 ctx1ophqlijrsmtuvwxkAn

46 

47 def update_words_cache(self, words_list: list, reset=False): 2 ctx1df

48 return self.namespace.update_words_cache(words_list, reset) 2 ctx1dz

49 

50 

51class LocalRowNamespace(LocalMacroNamespace): 2 ctx1df

52 

53 def __init__(self, controller, namespace, row): 2 ctx1df

54 LocalMacroNamespace.__init__(self, controller, namespace) 22 ctx1ophqlijrsmtuvwxkgnbcea

55 self._row = row 22 ctx1ophqlijrsmtuvwxkgnbcea

56 

57 def get_suggestions(self, start): 2 ctx1df

58 suggestions = LocalMacroNamespace.get_suggestions(self, start) 4 ctx1bcea

59 # print(f"DEBUG: suggesters.py LocalRowNamespace get_suggestions after LocalMacroNamespace start={start}\n" 

60 # f"suggestions={suggestions}") 

61 if self._could_be_variable(start): 4 ctx1bcea

62 suggestions = self._harvest_local_variables(start, suggestions) 4 ctx1bcea

63 else: 

64 suggestions = self._harvest_local_variables('${'+start, suggestions) 1 ctx1a

65 suggestions = self._harvest_local_variables('@{'+start, suggestions) 1 ctx1a

66 suggestions = self._harvest_local_variables('&{'+start, suggestions) 1 ctx1a

67 return suggestions 4 ctx1bcea

68 

69 def _harvest_local_variables(self, start, suggestions): 2 ctx1df

70 matching_assignments = set() 4 ctx1bcea

71 for row, step in enumerate(self._controller.steps): 71 ↛ 77line 71 didn't jump to line 77 because the loop on line 71 didn't complete4 ctx1bcea

72 if self._row == row: 4 ctx1bcea

73 break 4 ctx1bcea

74 matching_assignments = matching_assignments.union( 4 ctx1bcea

75 val.replace('=', '').strip() for val in step.assignments if 

76 val.startswith(start)) 

77 if matching_assignments: 4 ctx1bcea

78 local_variables = [LocalVariableInfo(name) for name 3 ctx1bca

79 in matching_assignments] 

80 suggestions = sorted(self._remove_duplicates(suggestions, 3 ctx1bca

81 local_variables)) 

82 return suggestions 4 ctx1bcea

83 

84 @staticmethod 2 ctx1df

85 def _could_be_variable(start): 2 ctx1df

86 return len(start) == 0 or start[0] in ['$', '@', '&'] 4 ctx1bcea

87 

88 def has_name(self, value): 2 ctx1df

89 if self._row is not None: 89 ↛ 95line 89 didn't jump to line 95 because the condition on line 89 was always true18 ctx1ophqlijrsmtuvwxkgn

90 for row, step in enumerate(self._controller.steps): 90 ↛ 95line 90 didn't jump to line 95 because the loop on line 90 didn't complete18 ctx1ophqlijrsmtuvwxkgn

91 if self._row == row: 18 ctx1ophqlijrsmtuvwxkgn

92 break 18 ctx1ophqlijrsmtuvwxkgn

93 if step.is_assigning(value): 6 ctx1limkgn

94 return True 3 ctx1ikg

95 return LocalMacroNamespace.has_name(self, value) 18 ctx1ophqlijrsmtuvwxkgn

96 

97 @staticmethod 2 ctx1df

98 def _remove_duplicates(suggestions, local_variables): 2 ctx1df

99 def is_unique(gvar): 3 ctx1bca

100 if hasattr(gvar, 'name'): 100 ↛ 103line 100 didn't jump to line 103 because the condition on line 100 was always true3 ctx1bca

101 return utils.normalize(gvar.name) not in [utils.normalize(lvar.name) for lvar in local_variables] 3 ctx1bca

102 else: 

103 return utils.normalize(gvar) not in [utils.normalize(lvar.name) for lvar in local_variables] 

104 unique = [gvar for gvar in suggestions if is_unique(gvar)] 3 ctx1bca

105 return unique + local_variables 3 ctx1bca