summaryrefslogtreecommitdiff
path: root/calc.py
blob: d7f030da474b1ef0bc20d2f13d5f1c36950520b7 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
from helpers import *

calc_users        = open_json_file("calc", [])
math_operators    = ["+", "-", "*", "/", "&", "|", ">", "<", "~", "%"]
ignore_operators  = ["**", "&&", "||"] # ** may be too intensive, the others cause syntax errors
calc_perm = "utils.calc"


def calc(text):
    """
    extracts a mathematical expression from `text`
    returns (expression, result) or None
    """
    expression = ""
    d_expression = ""
    should_calc = False
    for char in text:
        if char.isdigit() or (should_calc and char in [".", " "]):
            expression += char
        elif char in math_operators:
            # calculation must include at least 1 operator
            should_calc = True
            expression += char
        elif char.isalpha():
            # don't include any more text in the calculation
            if should_calc:
                d_expression = expression
            expression = ""
    if should_calc and not any(op in d_expression for op in ignore_operators):
        try:
            result = str(eval(d_expression)) # pylint: disable = W0123
        except: # pylint: disable = W0702
            # we can run into all kinds of errors here
            # most probably SyntaxError
            return None
        return (d_expression, result)
    return None


@hook.event("player.AsyncPlayerChatEvent", "monitor")
def on_calc_chat(event):
    sender = event.getPlayer()
    message = event.getMessage()
    if not event.isCancelled() and uid(sender) in calc_users and sender.hasPermission(calc_perm):
        output = calc(message)
        if type(output)in [int, float, long, complex]:
            msg(sender, "&2=== Calc: &e" + output[0] + " &2= &c" + str(output[1]).replace("420", "blazeit"))


@hook.command("calc", description="Toggles chat calculations")
def on_calc_command(sender, args):
    plugin_header(sender, "Chat Calculator")
    if not sender.hasPermission(calc_perm):
        noperm(sender)
        return True
    if not checkargs(sender, args, 0, 1):
        return True
    if not is_player(sender):
        msg(sender, "&cYou are not a player!" % sender)
        return True

    toggle(sender, calc_users, name = "Calc")
    save_json_file("calc", calc_users)

    return True