Add Commander

MarkKoz · 64json · commit 7072673b940c · 2019-06-11T15:50:09.000+09:00
* Capitalise module names for consistency with other tracers
diff --git a/tracer/Commander.py b/tracer/Commander.py
@@ -0,0 +1,36 @@
+import string
+from typing import Any, Dict, List
+
+from tracer import Randomize
+
+_MAX_COMMANDS = 1000000
+_MAX_OBJECTS = 100
+
+
+class Commander:
+    _keyRandomizer = Randomize.String(8, string.ascii_lowercase + string.digits)
+    _objectCount = 0
+    commands: List[Dict[str, Any]] = []
+
+    def __init__(self, **kwargs):
+        self._objectCount += 1
+        self.command(self.__class__.__name__, **kwargs)
+
+    @classmethod
+    def command(cls, method: str, key: str = _keyRandomizer.create(), **kwargs):
+        cmd = {
+            "key": key,
+            "method": method
+        }
+        cmd.update(kwargs)
+        cls.commands.append(cmd)
+
+        if len(cls.commands) > _MAX_COMMANDS:
+            raise RuntimeError("Too Many Commands")
+        elif cls._objectCount > _MAX_OBJECTS:
+            raise RuntimeError("Too Many Objects")
+
+    @classmethod
+    def destroy(cls):
+        cls._objectCount -= 1
+        cls.command("destroy")
diff --git a/tracer/Randomize.py b/tracer/Randomize.py
diff --git a/tracer/__init__.py b/tracer/__init__.py
@@ -0,0 +1,25 @@
+import atexit
+import json
+import os
+
+from tracer.Commander import Commander
+
+
+@atexit.register
+def execute():
+    if os.getenv("ALGORITHM_VISUALIZER"):
+        with open("visualization.json", "w", encoding="UTF-8") as file:
+            json.dump(Commander.commands, file)
+    else:
+        import requests
+        import webbrowser
+
+        response = requests.post(
+            "https://algorithm-visualizer.org/api/visualizations",
+            json=Commander.commands
+        )
+
+        if response.status_code == 200:
+            webbrowser.open(response.text)
+        else:
+            raise requests.HTTPError(response=response)
