Deleted files, re-add distributer

thijsmie · Nov 28, 2023 · 1db5346 · 1db5346
1 parent 807c5ba
commit 1db5346
Show file tree

Hide file tree

Showing 8 changed files with 36 additions and 50 deletions.
diff --git a/.buildpacks b/.buildpacks
diff --git a/Procfile b/Procfile
diff --git a/config/local.example.json b/config/local.example.json
diff --git a/config/production.example.json b/config/production.example.json
diff --git a/config/production.json b/config/production.json
diff --git a/run.py b/run.py
diff --git a/runlocal.py b/runlocal.py
diff --git a/src/rollbot/plottenbakker/distributer.py b/src/rollbot/plottenbakker/distributer.py
@@ -0,0 +1,36 @@
+import sys
+from datetime import timedelta
+from rollbot.interpreter.calculator import distribute, EvaluationError
+from rollbot.varenv import VarEnv
+from lark.exceptions import UnexpectedInput
+
+
+if __name__ == "__main__":
+    varenv_name = sys.stdin.readline().strip()
+    expression = sys.stdin.readline().strip()
+
+    try:
+        env = VarEnv(varenv_name)
+    except:
+        print("No such environment")
+        sys.exit(1)
+
+    try:
+        data = distribute(expression, timedelta(seconds=5), env)
+        if not data:
+            print(f"Unprocessable input")
+            sys.exit(1)
+        xbins, data, num_rolls = data
+    except EvaluationError as e:
+        print(e.args[0])
+        sys.exit(1)
+    except UnexpectedInput as e:
+        print(f"Unexpected input: ```\n{e.get_context(expression)}```")
+        sys.exit(1)
+    except:
+        print("Server error")
+        sys.exit(1)
+
+    print(" ".join(str(x) for x in xbins))
+    print(" ".join(str(x) for x in data))
+    print(num_rolls)