XZZ/worker/bc.py

# by weilinfox~
# 支持四则运算，括号和浮点数，可以省略部分乘号
# 支持语法检查
# sys 模式只支持 Linux

from zzcore import StdAns
import math
import subprocess

divError = False
overflowError = False

#函数段

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 "多出来的小数点很孤单的啦~"
    elif (errorNo == 6):
        return "啊吓死我了，0怎么做除数了啊……"
    elif (errorNo == 7):
        return "数字太大了，小身板撑不住啊……"
    elif (errorNo == 8):
        return "好像有奇奇怪怪的东西ummm, \"" + ch + "\"好像不能有……"
    elif (errorNo == 9):
        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):
    #处理不带括号的表达式
    global divError
    global overflowError
    mulflag = False
    plusflag = False
    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] != '*'):
            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
            if (ch[lmul] == '-' or ch[lmul] == '+'):
                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:])
                if (num2 == 0.0):
                    num2 = 1
                    divError = True
                num1 = float(ch[:fst])
                if (ch[fst] == '*'):
                    if (num1 * num2 >= 1e16):
                        overflowError = True
                        return '1'
                    return str(num1 * num2)
                elif (ch[fst] == '/'):
                    return str(num1 / num2)
            else:
                ch = exprCal(ch[:lst]) + ch[lst:]
    elif (plusflag):
        while (1):
            ch = ch.replace("+-", "-")
            ch = ch.replace("--", "+")
            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] == '+'):
                    if (num1 + num2 >= 1e16):
                        overflowError = True
                        return '1'
                    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):
    global divError
    global overflowError
    #预处理
    #字符替换
    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("+-", "-")
    com = com.replace("--", "+")
    com = com.replace("*+", "*")
    com = com.replace("/+", "/")
    com = com.replace("(-", "(0-")
    com = com.replace("(+", "(0+")
    com = com.replace("()", "")
    com = com.replace("x", "*")
    com = com.replace("X", "*")

    if (len(com) < 1):
        return reportErr(1, "这")

    #省略乘号的支持
    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] == '%' 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] == '(' or com[i+1] == '-')):
            com = com[:i+1] + "*" + com[i+1:]
    #部分情况开头加0
    if (com[0] == '+' or com[0] == '-' or com[0] == '.'):
        com = '0' + com

    #print(com)

    #表达式合法性

    #百分号合法性
    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)

    #括号合法性
    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] != '+' 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())):
        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] == "-"):
            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] != "+")):
                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, ".")

    ans = str(bracketExprCal(com))
    if (divError):
        divError = False
        return reportErr(6, "")
    elif (overflowError):
        overflowError = False
        return reportErr(7, "")
    return ans

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())
                if (num2 == 0):
                    return reportErr(6, "")
                return str(num1 / num2)
            elif (flag == '\\'):
                num1, num2 = map(int, ch.split())
                if (num2 == 0):
                    return reportErr(6, "")
                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)

def sysbc (arg):
    try:
        arg = arg.replace("\'", "")
        arg = "echo '''" + arg + "''' | bc"
        #print(arg)
        ret = subprocess.run(arg, shell = True, stdout = subprocess.PIPE, stderr = subprocess.STDOUT, check = True, timeout = 1)
        return bytes.decode(ret.stdout).replace("\n", "")
    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 的服务器爆了"


#代码段
class Ans (StdAns):
    AllowGroup = [959613860]
    def GETMSG(self):
        ans = "汝以为咱不会算算数的嘛！"
        if (len(self.parms) < 2):
            ans = "召唤我有什么事咩？有啥不懂输 help 哦~"
        elif (self.parms[1] == "help"):
            ans = "你喂给我式子我算呀，我只会四则运算哦，还不会就问狸吧~\n"
            ans += "如果输入fun会进入函数模式哦，funhelp可以查看支持的函数嘿嘿\n"
            ans += "输入sys允许我开挂哦，syshelp能查看开挂的方法~"
        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 += "别搞太大的数哦，服务器爆了打你哦~"
        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 += "别搞太大的数哦，服务器爆了打你哦~"
        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)
        return ans