[Storage] Enable Py_LIMITED_API for extensions (#34523)

Author: jalauzon-msft

sdk/storage/azure-storage-extensions/pyproject.toml

@@ -14,7 +14,7 @@ bandit = false
 suppressed_skip_warnings = ["*"]
 
 [tool.cibuildwheel]
-build = ["cp38*", "cp39*", "cp310*", "cp311*", "cp312*", "pp38*", "pp39*", "pp310*"]
+build = ["cp38*", "pp38*", "pp39*", "pp310*"]
 test-requires = "pytest"
 test-command = "pytest {project}/tests"
 test-skip = "*-macosx_arm64"
@@ -24,6 +24,7 @@ archs = "x86_64 aarch64"
 
 [tool.cibuildwheel.macos]
 archs = "x86_64 arm64"
+environment = "CFLAGS='-Wno-implicit-function-declaration'"
 
 [tool.cibuildwheel.windows]
 archs = "AMD64"

sdk/storage/azure-storage-extensions/setup.py

@@ -6,6 +6,18 @@
 
 import os
 from setuptools import setup, Extension
+from wheel.bdist_wheel import bdist_wheel
+
+
+class bdist_wheel_abi3(bdist_wheel):
+    """Override bdist_wheel tag behavior to add abi3 tag."""
+    def get_tag(self):
+        python, abi, plat = super().get_tag()
+
+        if python.startswith("cp"):
+            return python, "abi3", plat
+        return python, abi, plat
+
 
 PACKAGE_NAME = "azure-storage-extensions"
 PACKAGE_PPRINT_NAME = "Azure Storage Extensions"
@@ -43,7 +55,9 @@
         Extension(
             'crc64',
             [os.path.join(package_folder_path, "crc64", "crc64module.c")],
+            define_macros=[("Py_LIMITED_API", "3")],
             py_limited_api=True
         ),
     ],
+    cmdclass={"bdist_wheel": bdist_wheel_abi3},
 )

tools/azure-sdk-tools/ci_tools/build.py

@@ -179,10 +179,11 @@ def create_package(
     dist = get_artifact_directory(dest_folder)
     setup_parsed = ParsedSetup.from_path(setup_directory_or_file)
 
-    if setup_parsed.ext_modules:
-        run([sys.executable, "-m", "cibuildwheel", "--output-dir", dist], cwd=setup_parsed.folder, check=True)
-
     if enable_wheel:
-        run([sys.executable, "setup.py", "bdist_wheel", "-d", dist], cwd=setup_parsed.folder, check=True)
+        if setup_parsed.ext_modules:
+            run([sys.executable, "-m", "cibuildwheel", "--output-dir", dist], cwd=setup_parsed.folder, check=True)
+        else:
+            run([sys.executable, "setup.py", "bdist_wheel", "-d", dist], cwd=setup_parsed.folder, check=True)
+
     if enable_sdist:
         run([sys.executable, "setup.py", "sdist", "-d", dist], cwd=setup_parsed.folder, check=True)