XZZ/worker/bc.py

480 lines
16 KiB
Python
Raw Normal View History

2020-03-02 17:22:32 +08:00
# by weilinfox~
# 支持四则运算,括号和浮点数,可以省略部分乘号
# 支持语法检查
2020-03-07 09:08:59 +08:00
# sys 模式只支持 Linux
2020-03-02 17:22:32 +08:00
from zzcore import StdAns
2020-03-02 22:50:19 +08:00
import math
2020-03-07 09:08:59 +08:00
import subprocess
2020-03-02 17:22:32 +08:00
2020-03-02 18:24:28 +08:00
divError = False
2020-03-02 22:50:19 +08:00
overflowError = False
2020-03-02 18:24:28 +08:00
2020-03-02 17:22:32 +08:00
#函数段
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 "多出来的小数点很孤单的啦~"
2020-03-02 18:24:28 +08:00
elif (errorNo == 6):
return "啊吓死我了0怎么做除数了啊……"
2020-03-02 22:50:19 +08:00
elif (errorNo == 7):
return "数字太大了,小身板撑不住啊……"
elif (errorNo == 8):
return "好像有奇奇怪怪的东西ummm, \"" + ch + "\"好像不能有……"
elif (errorNo == 9):
return "运算符太多了,算不过来啊……"
2020-03-02 17:22:32 +08:00
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):
#处理不带括号的表达式
2020-03-02 18:24:28 +08:00
global divError
2020-03-02 22:50:19 +08:00
global overflowError
2020-03-02 17:22:32 +08:00
mulflag = False
plusflag = False
2020-03-02 20:41:40 +08:00
if (ch[0] == '-'):
ch = '0' + ch
ch = ch.replace("--", "+")
ch = ch.replace("+-", "-")
for i in range(1, len(ch), 1):
if ((ch[i] == "+" or ch[i] == '-') and ch[i-1] != '/' and ch[i-1] != '*'):
2020-03-02 17:22:32 +08:00
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
2020-03-02 20:41:40 +08:00
if (ch[lmul] == '-' or ch[lmul] == '+'):
lmul += 1
2020-03-02 17:22:32 +08:00
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:])
2020-03-02 18:24:28 +08:00
if (num2 == 0.0):
num2 = 1
divError = True
2020-03-02 17:22:32 +08:00
num1 = float(ch[:fst])
if (ch[fst] == '*'):
2020-03-02 22:50:19 +08:00
if (num1 * num2 >= 1e16):
overflowError = True
return '1'
2020-03-02 17:22:32 +08:00
return str(num1 * num2)
elif (ch[fst] == '/'):
return str(num1 / num2)
else:
ch = exprCal(ch[:lst]) + ch[lst:]
elif (plusflag):
while (1):
2020-03-02 20:41:40 +08:00
ch = ch.replace("+-", "-")
ch = ch.replace("--", "+")
2020-03-02 17:22:32 +08:00
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] == '+'):
2020-03-02 22:50:19 +08:00
if (num1 + num2 >= 1e16):
overflowError = True
return '1'
2020-03-02 17:22:32 +08:00
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):
2020-03-02 18:24:28 +08:00
global divError
2020-03-02 22:50:19 +08:00
global overflowError
2020-03-02 17:22:32 +08:00
#预处理
#字符替换
com = com.replace(" ", "")
com = com.replace("\'", "")
com = com.replace("\"", "")
com = com.replace(",", ".")
com = com.replace("{", "(")
com = com.replace("}", ")")
com = com.replace("[", "(")
com = com.replace("]", ")")
2020-03-02 20:41:40 +08:00
com = com.replace("+-", "-")
com = com.replace("--", "+")
com = com.replace("*+", "*")
com = com.replace("/+", "/")
com = com.replace("(-", "(0-")
com = com.replace("(+", "(0+")
com = com.replace("()", "")
2020-03-02 17:22:32 +08:00
com = com.replace("x", "*")
com = com.replace("X", "*")
2020-03-05 13:46:10 +08:00
if (len(com) < 1):
return reportErr(1, "")
2020-03-02 17:22:32 +08:00
#省略乘号的支持
for i in range(1, len(com)-1, 1):
2020-03-03 19:18:46 +08:00
if (com[i] == '(' and (com[i-1].isdigit() or com[i-1] == '.' or com[i-1] == '%' or com[i-1] == ')')):
2020-03-02 17:22:32 +08:00
com = com[:i] + "*" + com[i:]
2020-03-02 20:41:40 +08:00
elif (com[i] == ')' and (com[i+1].isdigit() or com[i+1] == '.' or com[i+1] == '(' or com[i+1] == '-')):
2020-03-02 17:22:32 +08:00
com = com[:i+1] + "*" + com[i+1:]
2020-03-02 20:41:40 +08:00
#部分情况开头加0
if (com[0] == '+' or com[0] == '-' or com[0] == '.'):
com = '0' + com
2020-03-05 13:46:10 +08:00
2020-03-02 17:22:32 +08:00
#print(com)
#表达式合法性
2020-03-03 19:18:46 +08:00
#百分号合法性
pcntnum = 0
if (com[0] == '%'):
return reportErr(3, com[0])
for i in range(1, len(com)-1, 1):
if (com[i] == '%'):
pcntnum += 1
if (not com[i-1].isdigit() or (com[i+1] != '+' and com[i+1] != '-' and com[i+1] != '*' and com[i+1] != '/' and com[i+1] != '(' and com[i+1] != ')')):
return reportErr(3, com[i])
if (pcntnum > 0):
i = 1
while (i < len(com)):
if (com[i] == '%'):
nst = i-1
while (nst >= 0 and com[nst].isdigit()):
nst -= 1
# print(nst)
com = com[:nst+1] + '(' + com[nst+1:]
i += 2
i += 1
com = com.replace("%", "/100)")
#print(com)
2020-03-02 17:22:32 +08:00
#括号合法性
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(")", "")
2020-03-02 20:41:40 +08:00
if (not ncom[0].isdigit() and ncom[0] != '+' and ncom[0] != '-' and ncom[0] != '.'):
return reportErr(3, ncom[0])
elif ((ncom[0] == '+' or ncom[0] == '-') and (len(ncom) < 2 or not ncom[1].isdigit())):
2020-03-02 17:22:32 +08:00
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):
2020-03-02 20:41:40 +08:00
if (ncom[i] == "+" or ncom[i] == "-"):
if ((not ncom[i-1].isdigit() and ncom[i-1] != '.' and ncom[i-1] != "*" and ncom[i-1] != "/") or (not ncom[i+1].isdigit() and ncom[i+1] != '.')):
return reportErr(3, ncom[i])
if (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] != '.' and ncom[i+1] != "-" and ncom[i+1] != "+")):
2020-03-02 17:22:32 +08:00
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, ".")
2020-03-02 18:24:28 +08:00
ans = str(bracketExprCal(com))
if (divError):
divError = False
return reportErr(6, "")
2020-03-02 22:50:19 +08:00
elif (overflowError):
overflowError = False
return reportErr(7, "")
2020-03-02 18:24:28 +08:00
return ans
2020-03-02 17:22:32 +08:00
2020-03-02 22:50:19 +08:00
def funMain (ch):
ch = ch.replace(" ", "")
if (ch[0:5] == 'sqrt(' and ch[len(ch)-1] == ")"):
ch = ch[5:len(ch)-1]
#print(ch)
dotNum = 0
for c in ch:
if (c == '.'):
dotNum += 1
elif (not c.isdigit()):
return reportErr(8, c)
if (dotNum > 1):
return reportErr(4, "")
return str(math.sqrt(float(ch)))
elif (ch[len(ch)-1] == '!'):
ch = ch[:len(ch)-1]
#print(ch)
for c in ch:
if (not c.isdigit()):
return reportErr(8, c)
n = int(ch)
if (n == 0):
return '1'
else:
ans = 1
for i in range(2, n+1, 1):
ans *= i
return str(ans)
else:
#print(ch)
dotNum = 0
for c in ch:
if (not c.isdigit()):
flag = c
dotNum += 1
if (dotNum > 1):
return reportErr(9, "")
elif (dotNum == 0):
return ch
else:
ch = ch.replace(flag, " ")
if (flag == '+'):
num1, num2 = map(int, ch.split())
return str(num1 + num2)
elif (flag == '-'):
num1, num2 = map(int, ch.split())
return str(num1 - num2)
elif (flag == '*'):
num1, num2 = map(int, ch.split())
return str(num1 * num2)
elif (flag == '/'):
num1, num2 = map(int, ch.split())
2020-03-03 19:18:46 +08:00
if (num2 == 0):
return reportErr(6, "")
2020-03-02 22:50:19 +08:00
return str(num1 / num2)
elif (flag == '\\'):
num1, num2 = map(int, ch.split())
2020-03-03 19:18:46 +08:00
if (num2 == 0):
return reportErr(6, "")
2020-03-02 22:50:19 +08:00
return str(num1 // num2) + " R = " + str(num1 % num2)
elif (flag == '^'):
num1, num2 = map(int, ch.split())
ans = 1
for i in range(0, num2, 1):
ans *= num1
return str(ans)
else:
return reportErr(8, flag)
2020-03-07 09:08:59 +08:00
def sysbc (arg):
try:
2020-03-07 18:45:44 +08:00
arg = arg.replace("\'", "")
2020-03-11 18:16:42 +08:00
arg = "echo '''" + arg + "''' | bc"
#print(arg)
2020-03-07 18:45:44 +08:00
ret = subprocess.run(arg, shell = True, stdout = subprocess.PIPE, stderr = subprocess.STDOUT, check = True, timeout = 1)
return bytes.decode(ret.stdout).replace("\n", "")
2020-03-07 09:08:59 +08:00
except subprocess.TimeoutExpired:
return "你干了啥怎么半天算不完啦!服务器爆了打你哦~"
except:
return "啊呀呀好像出错了,一定是 SiHuan 的服务器爆了"
def sysMain (lst):
scaleFlag = False
ibaseFlag = False
obaseFlag = False
scaleSet = False
ibaseSet = False
obaseSet = False
scaleN = 0
ibaseN = 0
obaseN = 0
ans = ""
try:
for s in lst:
if (s == "bc" or s == "sys"):
continue;
elif (s == "ibase"):
ibaseFlag = True
elif (s == "obase"):
obaseFlag = True
elif (s == "scale"):
scaleFlag = True
else:
dig = True
for c in s:
if (not s.isdigit()):
dig = False
break
if (ibaseFlag or obaseFlag or scaleFlag):
if (dig):
if (ibaseFlag):
ibaseSet = True
ibaseFlag = False
ibaseN = int(s)
elif (obaseFlag):
obaseSet = True
obaseFlag = False
obaseN = int(s)
elif (scaleFlag):
scaleSet = True
scaleFlag = False
scaleN = int(s)
else:
ans += "\n" + s + " ==> 这里好像有问题哎小老弟~"
return ans
else:
opt = ""
if (scaleSet):
opt += "scale=" + str(scaleN) + ";"
if (ibaseSet):
opt += "ibase=" + str(ibaseN) + ";"
if (obaseSet):
opt += "obase=" + str(obaseN) + ";"
opt += s
ans += "\n" + opt + " ==> " + sysbc(opt)
return ans
except:
return ans + "\n啊嘞嘞怎么肥四,一定是 SiHuan 的服务器爆了"
2020-03-02 22:50:19 +08:00
2020-03-02 17:22:32 +08:00
#代码段
class Ans (StdAns):
def GETMSG(self):
ans = "汝以为咱不会算算数的嘛!"
2020-03-02 22:50:19 +08:00
if (len(self.parms) < 2):
ans = "召唤我有什么事咩?有啥不懂输 help 哦~"
elif (self.parms[1] == "help"):
ans = "你喂给我式子我算呀,我只会四则运算哦,还不会就问狸吧~\n"
2020-03-07 09:08:59 +08:00
ans += "如果输入fun会进入函数模式哦funhelp可以查看支持的函数嘿嘿\n"
ans += "输入sys允许我开挂哦syshelp能查看开挂的方法~"
2020-03-02 22:50:19 +08:00
elif (self.parms[1] == "funhelp"):
ans = "fun支持这些哦:\n"
ans += " + ==> 大整数加法\n"
ans += " - ==> 大整数减法\n"
ans += " * ==> 大整数乘法\n"
ans += " / ==> 大整数除法\n"
ans += " \\ ==> 大整数取余除法\n"
ans += " ^ ==> 整数幂\n"
ans += " ! ==> 阶乘\n"
ans += "sqrt() ==> 开平方\n"
ans += "别搞太大的数哦,服务器爆了打你哦~"
2020-03-07 09:08:59 +08:00
elif (self.parms[1] == "sys"):
ans += sysMain(self.parms)
elif (self.parms[1] == "syshelp"):
ans = "sys可是允许我开挂哦~\n"
ans += "我还支持三个指令哦:\n"
ans += "scale ==> 指定保留小数位数\n"
ans += "ibase ==> 指定输入进制\n"
ans += "obase ==> 指定输出进制\n"
ans += "别搞太大的数哦,服务器爆了打你哦~"
2020-03-02 22:50:19 +08:00
else:
if (self.parms[1] == "fun"):
funMod = True
else:
funMod = False
for s in self.parms:
if (s != "bc" and s != "help" and s != "fun" and s != "funhelp"):
if (funMod):
ans += "\n" + s + " ==> " + funMain(s)
else:
ans += "\n" + s + " ==> " + bcMain(s)
2020-03-02 17:22:32 +08:00
return ans