Update starters __main__ to detect interactive session

Signed-off-by: Merel Theisen <[email protected]>
kedro-org · Aug 9, 2024 · bcf1b12 · bcf1b12
1 parent c6b9e8b
commit bcf1b12
Show file tree

Hide file tree

Showing 5 changed files with 30 additions and 5 deletions.
diff --git a/...rflow-iris/{{ cookiecutter.repo_name }}/src/{{ cookiecutter.python_package }}/__main__.py b/...rflow-iris/{{ cookiecutter.repo_name }}/src/{{ cookiecutter.python_package }}/__main__.py
@@ -1,6 +1,7 @@
 """{{ cookiecutter.project_name }} file for ensuring the package is executable
 as `{{ cookiecutter.repo_name }}` and `python -m {{ cookiecutter.python_package }}`
 """
+import sys
 from pathlib import Path
 
 from kedro.framework.cli.utils import find_run_command
@@ -10,9 +11,13 @@
 def main(*args, **kwargs):
     package_name = Path(__file__).parent.name
     configure_project(package_name)
+
+    interactive = hasattr(sys, 'ps1')
+    kwargs["standalone_mode"] = not interactive
+
     run = find_run_command(package_name)
     run(*args, **kwargs)
 
 
 if __name__ == "__main__":
-    main()
+    main()
diff --git a/...pandas-viz/{{ cookiecutter.repo_name }}/src/{{ cookiecutter.python_package }}/__main__.py b/...pandas-viz/{{ cookiecutter.repo_name }}/src/{{ cookiecutter.python_package }}/__main__.py
@@ -1,6 +1,7 @@
 """{{ cookiecutter.project_name }} file for ensuring the package is executable
 as `{{ cookiecutter.repo_name }}` and `python -m {{ cookiecutter.python_package }}`
 """
+import sys
 from pathlib import Path
 
 from kedro.framework.cli.utils import find_run_command
@@ -10,9 +11,13 @@
 def main(*args, **kwargs):
     package_name = Path(__file__).parent.name
     configure_project(package_name)
+
+    interactive = hasattr(sys, 'ps1')
+    kwargs["standalone_mode"] = not interactive
+
     run = find_run_command(package_name)
     run(*args, **kwargs)
 
 
 if __name__ == "__main__":
-    main()
+    main()
diff --git a/...hts-pandas/{{ cookiecutter.repo_name }}/src/{{ cookiecutter.python_package }}/__main__.py b/...hts-pandas/{{ cookiecutter.repo_name }}/src/{{ cookiecutter.python_package }}/__main__.py
@@ -1,6 +1,7 @@
 """{{ cookiecutter.project_name }} file for ensuring the package is executable
 as `{{ cookiecutter.repo_name }}` and `python -m {{ cookiecutter.python_package }}`
 """
+import sys
 from pathlib import Path
 
 from kedro.framework.cli.utils import find_run_command
@@ -10,9 +11,13 @@
 def main(*args, **kwargs):
     package_name = Path(__file__).parent.name
     configure_project(package_name)
+
+    interactive = hasattr(sys, 'ps1')
+    kwargs["standalone_mode"] = not interactive
+
     run = find_run_command(package_name)
     run(*args, **kwargs)
 
 
 if __name__ == "__main__":
-    main()
+    main()
diff --git a/...yspark-viz/{{ cookiecutter.repo_name }}/src/{{ cookiecutter.python_package }}/__main__.py b/...yspark-viz/{{ cookiecutter.repo_name }}/src/{{ cookiecutter.python_package }}/__main__.py
@@ -1,6 +1,7 @@
 """{{ cookiecutter.project_name }} file for ensuring the package is executable
 as `{{ cookiecutter.repo_name }}` and `python -m {{ cookiecutter.python_package }}`
 """
+import sys
 from pathlib import Path
 
 from kedro.framework.cli.utils import find_run_command
@@ -10,9 +11,13 @@
 def main(*args, **kwargs):
     package_name = Path(__file__).parent.name
     configure_project(package_name)
+
+    interactive = hasattr(sys, 'ps1')
+    kwargs["standalone_mode"] = not interactive
+
     run = find_run_command(package_name)
     run(*args, **kwargs)
 
 
 if __name__ == "__main__":
-    main()
+    main()
diff --git a/...ts-pyspark/{{ cookiecutter.repo_name }}/src/{{ cookiecutter.python_package }}/__main__.py b/...ts-pyspark/{{ cookiecutter.repo_name }}/src/{{ cookiecutter.python_package }}/__main__.py
@@ -1,6 +1,7 @@
 """{{ cookiecutter.project_name }} file for ensuring the package is executable
 as `{{ cookiecutter.repo_name }}` and `python -m {{ cookiecutter.python_package }}`
 """
+import sys
 from pathlib import Path
 
 from kedro.framework.cli.utils import find_run_command
@@ -10,9 +11,13 @@
 def main(*args, **kwargs):
     package_name = Path(__file__).parent.name
     configure_project(package_name)
+
+    interactive = hasattr(sys, 'ps1')
+    kwargs["standalone_mode"] = not interactive
+
     run = find_run_command(package_name)
     run(*args, **kwargs)
 
 
 if __name__ == "__main__":
-    main()
+    main()