From 91f41f6fdcfddc9404cacbd1e7a3c37b0c4f0872 Mon Sep 17 00:00:00 2001 From: weilinfox Date: Mon, 2 Mar 2020 17:22:32 +0800 Subject: [PATCH] add bc command support --- worker/bc.py | 223 +++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 223 insertions(+) create mode 100644 worker/bc.py diff --git a/worker/bc.py b/worker/bc.py new file mode 100644 index 0000000..bd67e40 --- /dev/null +++ b/worker/bc.py @@ -0,0 +1,223 @@ +# by weilinfox~ +# 支持四则运算，括号和浮点数，可以省略部分乘号 +# 支持语法检查 + +from zzcore import StdAns + +#函数段 + +def reportErr (errorNo, ch): + if (errorNo == 1): + return ch + "是什么东西嘛，哼！" + elif (errorNo == 2): + return "括号一个一个对不起来了啦~" + elif (errorNo == 3): + return ch + "都不会用了嘿！" + elif (errorNo == 4): + return "一个数小数点那么多拿来做什么嘞！" + elif (errorNo == 5): + return "多出来的小数点很孤单的啦~" + else: + return "啊呀呀这是啥错误啊？问问狸去：" + ch + +def supportCh (ch): + if (ch.isdigit()): + return True; + elif (ch == "+" or ch == "-" or ch == "*" or ch == "/" or ch == '.'): + return True + elif (ch == '(' or ch == ')'): + return True + else: + return False + +def exprCal (ch): + #处理不带括号的表达式 + mulflag = False + plusflag = False + for i in range(0, len(ch), 1): + if (ch[i] == "+" or ch[i] == '-'): + plusflag = True; + elif (ch[i] == '*' or ch[i] == '/'): + mulflag = True; + if (mulflag and plusflag): + break; + if (not mulflag and not plusflag): + return ch; + elif (mulflag and plusflag): + while (1): + fmul = lmul = 0 + i = 0 + for c in ch: + if (c == '*' or c == '/'): + if (fmul == 0): + fmul = i + lmul = i + elif (fmul != 0 and (c == '+' or c == '-')): + break + i += 1 + if (fmul == 0 and lmul == 0): + break; + fmul -= 1 + lmul += 1 + while (fmul >= 0 and (ch[fmul].isdigit() or ch[fmul] == '.')): + fmul -= 1 + while (lmul < len(ch) and (ch[lmul].isdigit() or ch[lmul] == '.')): + lmul += 1 + fmul += 1 + ch = ch[:fmul] + exprCal(ch[fmul:lmul]) + ch[lmul:] + # print(ch) + return exprCal(ch) + elif (mulflag): + while (1): + fst = lst = 0 + i = 0 + for c in ch: + if (c == '*' or c == '/'): + if (fst == 0): + fst = i + else: + lst = i + break + i += 1 + if (lst == 0): + num2 = float(ch[fst+1:]) + num1 = float(ch[:fst]) + if (ch[fst] == '*'): + return str(num1 * num2) + elif (ch[fst] == '/'): + return str(num1 / num2) + else: + ch = exprCal(ch[:lst]) + ch[lst:] + elif (plusflag): + while (1): + fst = lst = 0 + i = 0 + for c in ch: + if (c == '+' or c == '-'): + if (fst == 0): + fst = i + else: + lst = i + break + i += 1 + if (lst == 0): + num2 = float(ch[fst+1:]) + num1 = float(ch[:fst]) + if (ch[fst] == '+'): + return str(num1 + num2) + elif (ch[fst] == '-'): + return str(num1 - num2) + else: + ch = exprCal(ch[:lst]) + ch[lst:] + +def bracketExprCal (ch): + #处理带括号的表达式 + fbr = lbr = -1 + i = 0 + for c in ch: + if (c == '('): + fbr = i + elif (c == ')'): + lbr = i + break + i += 1 + if (fbr == -1): + return exprCal(ch) + else: + if (lbr+1 >= len(ch)): + # print(ch) + return bracketExprCal(ch[:fbr] + bracketExprCal(ch[fbr+1:lbr])) + else: + # print(ch) + return bracketExprCal(ch[:fbr] + bracketExprCal(ch[fbr+1:lbr]) + ch[lbr+1:]) + +def bcMain(com): + #预处理 + #字符替换 + com = com.replace(" ", "") + com = com.replace("\'", "") + com = com.replace("\"", "") + com = com.replace(",", ".") + com = com.replace("{", "(") + com = com.replace("}", ")") + com = com.replace("[", "(") + com = com.replace("]", ")") + com = com.replace("x", "*") + com = com.replace("X", "*") + + #省略乘号的支持 + for i in range(1, len(com)-1, 1): + if (com[i] == '(' and (com[i-1].isdigit() or com[i-1] == '.' or com[i-1] == ')')): + com = com[:i] + "*" + com[i:] + elif (com[i] == ')' and (com[i+1].isdigit() or com[i+1] == '.' or com[i+1] == '(')): + com = com[:i+1] + "*" + com[i+1:] + + #print(com) + + #表达式合法性 + + #括号合法性 + ncom = "" + for ch in com: + if (not supportCh(ch)): + return reportErr(1, ch) + elif (ch == "(" or ch == ")"): + ncom += ch + nlen = plen = len(ncom) + while (nlen > 0): + ncom = ncom.replace("()", "") + nlen = len(ncom) + if (nlen == plen): + return reportErr(2, ncom[0]) + plen = nlen + + #运算符及小数点位置合法性 + ncom = com.replace("(", "") + ncom = ncom.replace(")", "") + if (not ncom[0].isdigit() and ncom[0] != '.'): + return reportErr(3, ncom[0]) + elif (ncom[0] == '.' and (len(ncom) < 2 or not ncom[1].isdigit())): + return reportErr(5, "") + elif (not ncom[len(ncom)-1].isdigit() and ncom[len(ncom)-1] != '.'): + return reportErr(3, ncom[len(ncom)-1]) + elif (ncom[len(ncom)-1] == '.' and (len(ncom) < 2 or not ncom[len(ncom)-2].isdigit())): + return reportErr(5, "") + for i in range(1, len(ncom)-1, 1): + if (ncom[i] == "+" or ncom[i] == "-" or ncom[i] == "*" or ncom[i] == "/"): + if ((not ncom[i-1].isdigit() and ncom[i-1] != '.') or (not ncom[i+1].isdigit() and ncom[i+1] != '.')): + return reportErr(3, ncom[i]) + elif (ncom[i] == '.'): + if (not ncom[i-1].isdigit() and not ncom[i+1].isdigit()): + return reportErr(5, "") + + #小数点数量合法性 + inNum = False + dotNum = 0 + for i in range(0, len(com), 1): + if (com[i].isdigit()): + inNum = True + elif (com[i] == "."): + inNum = True + dotNum += 1 + else: + inNum = False + if (dotNum > 1): + return reportErr(4, ".") + dotNum = 0 + if (dotNum > 1): + return reportErr(4, ".") + + return bracketExprCal(com) + +#代码段 +class Ans (StdAns): + def GETMSG(self): + ans = "汝以为咱不会算算数的嘛！" + for s in self.parms: + if (s == "help"): + ans = "你喂给我式子我算呀，还不会问狸吧~" + break + elif (s != "bc"): + ans += "\n" + s + " ==> " + bcMain(s) + return ans +