Packaging stub file

ddacw · ddacw · commit 26a72e151bdf · 2021-07-21T08:25:30.000+02:00
diff --git a/modules/python/bindings/CMakeLists.txt b/modules/python/bindings/CMakeLists.txt
@@ -2,6 +2,7 @@ set(MODULE_NAME "python_bindings_generator")
 set(OPENCV_MODULE_IS_PART_OF_WORLD FALSE)
 ocv_add_module(${MODULE_NAME} INTERNAL)
 
+set(OPENCV_PYTHON_STUB_FILE "${CMAKE_CURRENT_BINARY_DIR}/__init__.pyi" CACHE INTERNAL "")
 set(OPENCV_PYTHON_SIGNATURES_FILE "${CMAKE_CURRENT_BINARY_DIR}/pyopencv_signatures.json" CACHE INTERNAL "")
 set(OPENCV_PYTHON_BINDINGS_DIR "${CMAKE_CURRENT_BINARY_DIR}" CACHE INTERNAL "")
 
@@ -67,6 +68,7 @@ set(cv2_generated_files
     "${CMAKE_CURRENT_BINARY_DIR}/pyopencv_generated_modules_content.h"
     "${CMAKE_CURRENT_BINARY_DIR}/pyopencv_generated_types.h"
     "${CMAKE_CURRENT_BINARY_DIR}/pyopencv_generated_types_content.h"
+    "${OPENCV_PYTHON_STUB_FILE}"
     "${OPENCV_PYTHON_SIGNATURES_FILE}"
 )
 
@@ -84,6 +86,13 @@ add_custom_command(
 
 add_custom_target(gen_opencv_python_source DEPENDS ${cv2_generated_files})
 
+add_custom_target(package_python_stub DEPENDS gen_opencv_python_source)
+
+add_custom_command(
+    TARGET package_python_stub
+    COMMAND ${CMAKE_COMMAND} -E copy ${OPENCV_PYTHON_STUB_FILE} ${CMAKE_BINARY_DIR}/python_loader/cv2/__init__.pyi
+)
+
 set(cv2_custom_hdr "${CMAKE_CURRENT_BINARY_DIR}/pyopencv_custom_headers.h")
 set(cv2_custom_hdr_str "//user-defined headers\n")
 foreach(uh ${opencv_userdef_hdrs})
diff --git a/modules/python/common.cmake b/modules/python/common.cmake
@@ -24,6 +24,10 @@ if(TARGET gen_opencv_python_source)
   add_dependencies(${the_module} gen_opencv_python_source)
 endif()
 
+if(TARGET package_python_stub)
+  add_dependencies(${the_module} package_python_stub)
+endif()
+
 ocv_assert(${PYTHON}_VERSION_MAJOR)
 ocv_assert(${PYTHON}_VERSION_MINOR)
 
