From 976f2d06e52c7dbf83e67a2999f0b593d19f98d7 Mon Sep 17 00:00:00 2001
From: Martin Karkowski <m.karkowski@dozema.local>
Date: Sat, 23 Jul 2022 07:34:58 +0200
Subject: [PATCH] Adding Post-Processor

---
 py-helpers/prepare_code/__init__.py       |  1 +
 py-helpers/prepare_code/post_processor.py | 27 +++++++++++++++++++++++
 2 files changed, 28 insertions(+)
 create mode 100644 py-helpers/prepare_code/post_processor.py

diff --git a/py-helpers/prepare_code/__init__.py b/py-helpers/prepare_code/__init__.py
index 645471e..6fbe8b8 100644
--- a/py-helpers/prepare_code/__init__.py
+++ b/py-helpers/prepare_code/__init__.py
@@ -5,6 +5,7 @@ __copyright__ = 'Copyright 2022 M.Karkowski'
 __version__ = '0.1.0'
 
 from .logger import get_logger
+from .post_processor import post_process
 from .js import get_parser as get_parser_js, transform as transform_js
 from .ts import get_parser as get_parser_ts, transform as transform_ts
 from .main import main
\ No newline at end of file
diff --git a/py-helpers/prepare_code/post_processor.py b/py-helpers/prepare_code/post_processor.py
new file mode 100644
index 0000000..22474e3
--- /dev/null
+++ b/py-helpers/prepare_code/post_processor.py
@@ -0,0 +1,27 @@
+
+
+replacers = {
+    "console.log": "print",
+    "console.error": "print",
+    "Error(": "Exception(",
+    "true": "True",
+    "false": "False",
+    "JSON.stringify": "json.dumps",
+    "JSON.parse": "json.loads"
+}
+
+def post_process(code: str) -> str:
+    """ Post processes the code. This results in adapting the code by replacing default 
+        elements like console.log
+
+    Args:
+        code (str): The code that have to be adapted
+
+    Returns:
+        str: The adapted code
+    """
+
+    for org, new in replacers.items():
+        code = code.replace(org, new)
+
+    return code
\ No newline at end of file