added exit to script

d3i-infra · Jan 11, 2024 · 685335c · 685335c
1 parent 5658b9a
commit 685335c
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/framework/processing/py/port/script.py b/src/framework/processing/py/port/script.py
@@ -17,7 +17,7 @@
 import port.instagram as instagram
 import port.linkedin as linkedin
 
-from port.api.commands import (CommandSystemDonate, CommandUIRender)
+from port.api.commands import (CommandSystemDonate, CommandUIRender, CommandSystemExit)
 
 LOG_STREAM = io.StringIO()
 
@@ -119,6 +119,7 @@ def process(session_id):
                 LOGGER.info("Skipped ater reviewing consent: %s", platform_name)
                 yield donate_logs(f"{session_id}-tracking")
 
+    yield exit(0, "Success")
     yield render_end_page()
 
 
@@ -602,3 +603,6 @@ def prompt_file(extensions, platform):
 
 def donate(key, json_string):
     return CommandSystemDonate(key, json_string)
+
+def exit(code, info):
+    return CommandSystemExit(code, info)