[Bindings] pass shamrock sys.argv to python (Shamrock-code#1553)

Author: tdavidcl

src/main.cpp

@@ -190,13 +190,13 @@ int main(int argc, char *argv[]) {
                         "cannot run ipython mode with > 1 processes");
                 }
 
-                shambindings::start_ipython(true);
+                shambindings::start_ipython(true, argc, argv);
 
             } else if (opts::has_option("--rscript")) {
                 __shamrock_stack_entry();
                 std::string fname = std::string(opts::get_option("--rscript"));
 
-                shambindings::run_py_file(fname, shamcomm::world_rank() == 0);
+                shambindings::run_py_file(fname, shamcomm::world_rank() == 0, argc, argv);
 
             } else {
                 if (shamcomm::world_rank() == 0) {

src/shambindings/include/shambindings/start_python.hpp

@@ -27,6 +27,13 @@ namespace shambindings {
      */
     void setpypath(std::string path);
 
+    /**
+     * @brief set the value of sys.argv
+     *
+     * This function will throw if bindings were not initialized in embed mode
+     */
+    void set_sys_argv(int argc, char *argv[]);
+
     /**
      * @brief set the value of sys.path before init from the supplied binary
      *
@@ -42,7 +49,7 @@ namespace shambindings {
      * @warning This function shall not be called if more than one processes are running
      * @param do_print print log at python startup
      */
-    void start_ipython(bool do_print);
+    void start_ipython(bool do_print, int argc, char *argv[]);
 
     /**
      * @brief run python runscript
@@ -52,7 +59,7 @@ namespace shambindings {
      * @param do_print print log at python startup
      * @param file_path path to the runscript
      */
-    void run_py_file(std::string file_path, bool do_print);
+    void run_py_file(std::string file_path, bool do_print, int argc, char *argv[]);
 
     /**
      * @brief Modify Python sys.path to point to one detected during cmake invocation

src/shambindings/src/run_ipython.py

@@ -22,4 +22,4 @@
 ###
 """
 
-start_ipython(config=c)
+start_ipython(argv=[], config=c)

src/shambindings/src/start_python.cpp

@@ -20,6 +20,7 @@
 #include "shambindings/pybindings.hpp"
 #include "shambindings/start_python.hpp"
 #include <pybind11/embed.h>
+#include <pybind11/stl.h>
 #include <cstdlib>
 #include <optional>
 #include <string>
@@ -109,10 +110,16 @@ namespace shambindings {
         py::exec(modify_path);
     }
 
-    void start_ipython(bool do_print) {
+    void set_sys_argv(int argc, char *argv[]) {
+        std::vector<std::string> cpp_argv(argv, argv + argc);
+        py::module_::import("sys").attr("argv") = py::cast(cpp_argv);
+    }
+
+    void start_ipython(bool do_print, int argc, char *argv[]) {
 
         py::scoped_interpreter guard{};
         modify_py_sys_path(do_print);
+        set_sys_argv(argc, argv);
 
         if (do_print) {
             shambase::println("--------------------------------------------");
@@ -127,9 +134,10 @@ namespace shambindings {
         }
     }
 
-    void run_py_file(std::string file_path, bool do_print) {
+    void run_py_file(std::string file_path, bool do_print, int argc, char *argv[]) {
         py::scoped_interpreter guard{};
         modify_py_sys_path(do_print);
+        set_sys_argv(argc, argv);
 
         if (do_print) {
             shambase::println("-----------------------------------");

src/shamtest/shamtest.cpp

@@ -501,6 +501,7 @@ namespace shamtest {
 
         ON_RANK_0(shamcomm::logs::print_faint_row());
         shambindings::modify_py_sys_path(shamcomm::world_rank() == 0);
+        shambindings::set_sys_argv(argc, argv);
         ON_RANK_0(shamcomm::logs::print_faint_row());
 
         // import shamrock in pybind