config-able function blacklist

Author: Fallen-Breath

readme.md

@@ -5,6 +5,8 @@ A simple in game calculator plugin for [MCDReforged](https://github.com/Fallen-B
 
 Requires [simpleeval](https://pypi.org/project/simpleeval/) module
 
+Configure file: `config/simple_calculator/config.json`
+
 ## Usage
 
 It will calculate the expression with an extra prefix `==` such as `==1+1`. All unrelated chars will be ignored

simple_calculator/calculator.py

@@ -45,14 +45,16 @@ def _calculate(self, expression: str) -> Optional[RESULT]:
 
 
 class SimpleevalCalculator(Calculator):
-	def _calculate(self, expression: str) -> Optional[RESULT]:
+	def __init__(self, config: Optional[Configure]):
+		super().__init__(config)
+		
 		def merge(a: dict, b: dict):
 			ret = a.copy()
 			ret.update(b)
 			return ret
 
 		simpleeval.MAX_POWER = self.config.max_power_length
-		s = simpleeval.SimpleEval(
+		core = simpleeval.SimpleEval(
 			operators=merge(simpleeval.DEFAULT_OPERATORS, {
 				ast.BitXor: operator.xor,
 				ast.BitAnd: operator.and_,
@@ -64,13 +66,16 @@ def merge(a: dict, b: dict):
 		for k, v in math.__dict__.items():
 			if not k.startswith('_'):
 				if type(v) in [int, float]:
-					s.names[k] = v
-				elif callable(v) and k not in ['exp', 'expm1', 'ldexp', 'pow', 'factorial']:
-					s.functions[k] = v
-		s.functions.update({
+					core.names[k] = v
+				elif callable(v) and k not in self.config.function_blacklist:
+					core.functions[k] = v
+		core.functions.update({
 			'hex': lambda x: hex(x).replace('0x', '', 1).rstrip('L').upper(),
 			'bin': lambda x: bin(x).replace('0b', '', 1).rstrip('L'),
 			'oct': lambda x: oct(x).replace('0o', '', 1).rstrip('L'),
 			'bool': bool
 		})
-		return s.eval(expression)
+		self.core = core
+
+	def _calculate(self, expression: str) -> Optional[RESULT]:
+		return self.core.eval(expression)

simple_calculator/configure.py

@@ -1,7 +1,12 @@
+from typing import List
+
 from mcdreforged.api.utils import Serializable
 
 
 class Configure(Serializable):
 	max_result_length = 1000  # maximum result string length
 	max_power_length = 10000  # simpleeval.MAX_POWER override
 	enable_simpleeval = True  # if enable simpleeval for better calculator if it's doable
+	function_blacklist: List[str] = [
+		'exp', 'expm1', 'ldexp', 'pow', 'factorial'
+	]