Merge pull request #579 from asenyaev/asen/fix_pylin_and_pycharm

asenyaev · web-flow · commit 053e8cc1c80b · 2021-11-17T11:22:26.000+03:00
Added lines for a proper work using pylint and an autocomplete in IDE
diff --git a/setup.py b/setup.py
@@ -377,6 +377,21 @@ def _classify_installed_files_override(
 
         print("Copying files from CMake output")
 
+        # lines for a proper work using pylint and an autocomplete in IDE
+        with open(os.path.join(cmake_install_dir, "python", "cv2", "__init__.py"), 'r') as opencv_init:
+            opencv_init_lines = opencv_init.readlines()
+            extra_imports = ('\nfrom .cv2 import *\nfrom .cv2 import _registerMatType\nfrom . import mat_wrapper\nfrom . import gapi'
+                             '\nfrom . import misc\nfrom . import utils\nfrom . import data\n')
+            free_line_after_imports = 6
+            opencv_init_lines.insert(free_line_after_imports, extra_imports)
+            opencv_init_data = ""
+            for line in opencv_init_lines:
+                opencv_init_replacement = line.replace('importlib.import_module("cv2")', 'importlib.import_module("cv2.cv2")')
+                opencv_init_data = opencv_init_data + opencv_init_replacement
+
+        with open(os.path.join(cmake_install_dir, "python", "cv2", "__init__.py"), 'w') as opencv_final_init:
+            opencv_final_init.write(opencv_init_data)
+
         # add lines from the old __init__.py file to the config file
         with open(os.path.join(os.path.dirname(os.path.abspath(__file__)), 'scripts', '__init__.py'), 'r') as custom_init:
             custom_init_data = custom_init.read()
