Correction for arithmetical operations, They now have their dedicated information function instead of the standard one to avoid the issue of arithmetical operations appearing as alternatives inside a bigger one

Author: mpourmpoulis

library/info.py

@@ -333,24 +333,30 @@ def get_comparison_middle(root):
 
 
 # Extract Left Middle and Right from arithmetical operations
+
+def get_arithmetic(root):
+	if not match_node(root,ast.BinOp)  or match_parent(root,ast.BinOp):
+		return None
+	return root
+	
 def get_arithmetic_left(root):
-	if not match_node(root,ast.BinOp):
+	if not match_node(root,ast.BinOp)  or match_parent(root,ast.BinOp):
 		return None
 	items = get_sub_index(root,None)
 	if len(items)>=1:
 		return items[0]
 	return None
 
 def get_arithmetic_right(root):
-	if not match_node(root,ast.BinOp):
+	if not match_node(root,ast.BinOp) or match_parent(root,ast.BinOp):
 		return None
 	items = get_sub_index(root,None)
 	if len(items)>=2:
 		return items[-1]
 	return None
 
 def get_arithmetic_middle(root):
-	if not match_node(root,ast.BinOp):
+	if not match_node(root,ast.BinOp) or match_parent(root,ast.BinOp):
 		return None
 	items = get_sub_index(root,None)
 	if len(items)==3:

queries/big_roi.py

@@ -119,7 +119,7 @@ def standard(x):
 
             "attribute":((ast.Attribute),(),standard),
             "comparison":((ast.Compare),(),standard),
-            "arithmetic":((ast.BinOp),(),standard),
+            "arithmetic":((ast.BinOp),(),get_arithmetic),
             "boolean":((ast.BoolOp),(),standard),
 
             "container":((ast.Compare),(),get_container_check),