1 import re 2 3 4 def calc(formula): 5 formula = re.sub(' ', '', formula) 6 formula_ret = 0 7 match_brackets = re.search(r'([^()]+)', formula) 8 if match_brackets: 9 calc_result = calc(match_brackets.group().strip("(,)")) 10 formula = formula.replace(match_brackets.group(), str(calc_result)) 11 return calc(formula) 12 else: 13 formula = formula.replace('--', '+').replace('++', '+').replace('-+', '-').replace('+-', '-') 14 while re.findall(r"[*/]", formula): 15 get_formula = re.search(r"[.d]+[*/]+[-]?[.d]+", formula) 16 if get_formula: 17 get_formula_str = get_formula.group() 18 if get_formula_str.count("*"): 19 formula_list = get_formula_str.split("*") 20 ret = float(formula_list[0]) * float(formula_list[1]) 21 else: 22 formula_list = get_formula_str.split("/") 23 ret = float(formula_list[0]) / float(formula_list[1]) 24 formula = formula.replace(get_formula_str, str(ret)).replace('--', '+').replace('++', '+') 25 formula = re.findall(r'[-]?[.d]+', formula) 26 for num in formula: 27 formula_ret += float(num) 28 return formula_ret 29 # calc_str = input('>>>') 30 calc_str = '1-2*((60-30+(-40---5)*(9-2*5/3+7/3*99/4*2998+ 10*568/ 14))-(-4*3)/(16-3*2))' 31 print(eval(calc_str)) 32 formula_result = calc(calc_str) 33 print(formula_result)
带基本验证
1 import re 2 invalid_str_re = re.compile(r'[*/]{3,}|[%]{2,}|([*/%]+d+|^[*/%]+d+|(+)+|d+[-+*/%]+$|)+(+|d+[-+*/%]+)|d+(+') 3 4 5 def decorate(func): 6 def wrapper(formula): 7 formula = re.sub(' ', '', formula) 8 valid_str = re.findall(r'[-+*/%.()d]+', formula) 9 invalid_str = invalid_str_re.findall(formula) 10 if (valid_str and len(valid_str[0]) != len(formula)) or invalid_str: 11 return False 12 return func(formula) 13 return wrapper 14 15 16 @decorate 17 def calc(formula): 18 formula = re.sub(' ', '', formula) 19 formula_ret = 0 20 match_brackets = re.search(r'([^()]+)', formula) 21 if match_brackets: 22 calc_result = calc(match_brackets.group().strip("(,)")) 23 formula = formula.replace(match_brackets.group(), str(calc_result)) 24 return calc(formula) 25 else: 26 formula = formula.replace('--', '+').replace('++', '+').replace('-+', '-').replace('+-', '-') 27 while re.findall(r"[*/]", formula): 28 get_formula = re.search(r"[.d]+[*/]+[-]?[.d]+", formula) 29 if get_formula: 30 get_formula_str = get_formula.group() 31 if get_formula_str.count("*"): 32 formula_list = get_formula_str.split("*") 33 ret = float(formula_list[0]) * float(formula_list[1]) 34 else: 35 formula_list = get_formula_str.split("/") 36 ret = float(formula_list[0]) / float(formula_list[1]) 37 formula = formula.replace(get_formula_str, str(ret)).replace('--', '+').replace('++', '+') 38 formula = re.findall(r'[-]?[.d]+', formula) 39 for num in formula: 40 formula_ret += float(num) 41 return formula_ret 42 # calc_str = input('>>>') 43 calc_str = '1-2*((60-30+(-40---5)*(9-2*5/3+7/3*99/4*2998+ 10*568/ 14))-(-4*3)/(16-3*2))' 44 formula_result = calc(calc_str) 45 print(formula_result)