Update setup.py

Author: avdivan

setup.py

@@ -160,12 +160,18 @@ def main():
     # Raw paths relative to sourcetree root.
     files_outside_package_dir = {"cv2": ["LICENSE.txt", "LICENSE-3RD-PARTY.txt"]}
 
-    ci_cmake_generator = (
-        ["-G", "Visual Studio 14" + (" Win64" if is64 else "")]
-        if os.name == "nt"
-        else ["-G", "Unix Makefiles"]
-    )
-
+    if os.name == "nt": 
+        vs_env = os.environ.get("VISUAL_STUDIO")
+        if vs_env == "17":
+           generator_name = "Visual Studio 17 2022"
+           arch = "x64" if is64 else "Win32"
+           ci_cmake_generator = ["-G", generator_name, "-A", arch]
+        else:
+           generator_name = "Visual Studio 14" + (" Win64" if is64 else "")
+           ci_cmake_generator = ["-G", generator_name]
+    else:
+        ci_cmake_generator = ["-G", "Unix Makefiles"]
+        
     cmake_args = (
         (ci_cmake_generator if is_CI_build else [])
         + [
@@ -193,8 +199,8 @@ def main():
             "-DBUILD_DOCS=OFF",
             "-DPYTHON3_LIMITED_API=ON",
             "-DBUILD_OPENEXR=ON",
-            "-DLAPACK=ON",
-            
+            "-DWITH_OBSENSOR=OFF",
+            "-DWITH_MEDIAFOUNDATION=OFF",
         ]
         + (
             # CMake flags for windows/arm64 build