python-nb : send nanobind module coal_pywrap_v2 to new Python module "coal.v2"

Author: ManifoldFR

python-nb/CMakeLists.txt

@@ -4,7 +4,7 @@ find_package(nanobind CONFIG REQUIRED)
 set(PYTHON_LIB_NAME_V2 coal_pywrap_v2)
 set(coal_pywrap_v2_SOURCES coal.cc)
 
-nanobind_add_module(${PYTHON_LIB_NAME_V2} NB_STATIC NB_SUPPRESS_WARNINGS ${coal_pywrap_v2_SOURCES})
+nanobind_add_module(${PYTHON_LIB_NAME_V2} NB_STATIC LTO NB_SUPPRESS_WARNINGS ${coal_pywrap_v2_SOURCES})
 target_link_libraries(
   ${PYTHON_LIB_NAME_V2}
   PRIVATE ${PROJECT_NAME} Boost::system
@@ -13,3 +13,25 @@ target_compile_definitions(
   ${PYTHON_LIB_NAME_V2}
   PRIVATE COAL_PYTHON_LIBNAME=${PYTHON_LIB_NAME_V2}
 )
+
+set(MODULE_DIR ${PROJECT_NAME}/v2)
+string(REPLACE "/" "." MODULE_NAME ${MODULE_DIR})
+
+set_target_properties(
+  ${PYTHON_LIB_NAME_V2}
+  PROPERTIES
+    LIBRARY_OUTPUT_DIRECTORY "${CMAKE_CURRENT_BINARY_DIR}/${MODULE_DIR}"
+)
+
+set(PYTHON_FILES __init__.py v2/__init__.py)
+
+foreach(pyfile ${PYTHON_FILES})
+  PYTHON_BUILD(${PROJECT_NAME} ${pyfile})
+endforeach(pyfile)
+
+nanobind_add_stub(
+  coal_pywrap_v2_stub
+  MODULE ${MODULE_NAME}
+  OUTPUT ${MODULE_NAME}.pyi
+  DEPENDS ${PYTHON_LIB_NAME_V2}
+)

python-nb/coal/__init__.py

@@ -0,0 +1,3 @@
+# ruff: noqa: F401, F403
+from .coal_pywrap_v2 import *  # noqa
+from .coal_pywrap_v2 import __version__