From e769f36b7a31c91b1b57741eddf6a1609578b1d3 Mon Sep 17 00:00:00 2001
From: Vlad Emelianov <vlad.emelianov.nz@gmail.com>
Date: Thu, 26 Dec 2024 23:16:05 +0300
Subject: [PATCH] Fix full packages

---
 .github/workflows/integration_full.yml                 | 5 +++--
 mypy_boto3_builder/generators/aiobotocore_generator.py | 7 ++++++-
 mypy_boto3_builder/generators/boto3_generator.py       | 7 ++++++-
 mypy_boto3_builder/generators/types_boto3_generator.py | 7 ++++++-
 4 files changed, 21 insertions(+), 5 deletions(-)

diff --git a/.github/workflows/integration_full.yml b/.github/workflows/integration_full.yml
index 23cda00c..408820c2 100644
--- a/.github/workflows/integration_full.yml
+++ b/.github/workflows/integration_full.yml
@@ -26,8 +26,8 @@ jobs:
           - "types-aiobotocore-custom"
         python_version:
           - "3.8"
-          # - "3.10"
-          # - "3.13"
+          - "3.10"
+          - "3.13"
     steps:
       - uses: actions/checkout@v4
       - name: Set up uv
@@ -39,4 +39,5 @@ jobs:
           uv run scripts/integration.py \
             --product ${{ matrix.product }} \
             --python ${{ matrix.python_version }} \
+            --wheel \
             -d
\ No newline at end of file
diff --git a/mypy_boto3_builder/generators/aiobotocore_generator.py b/mypy_boto3_builder/generators/aiobotocore_generator.py
index 53d622cd..756c2e3b 100644
--- a/mypy_boto3_builder/generators/aiobotocore_generator.py
+++ b/mypy_boto3_builder/generators/aiobotocore_generator.py
@@ -152,11 +152,16 @@ def generate_full_stubs(self) -> Package | None:
                 "*.pyi",
             )
         package.set_description(package.get_short_description("All-in-one type annotations"))
-        self._generate_full_stubs_services(package)
         self.package_writer.write_package(
             package,
             template_path=TemplatePath.types_aiobotocore_full,
         )
+        self._generate_full_stubs_services(package)
+        self.setup_package_writer.write_package(
+            package,
+            template_path=TemplatePath.types_aiobotocore_full,
+            include_template_names=("setup.py.jinja2",),
+        )
         return package
 
     def generate_custom_stubs(self) -> Package:
diff --git a/mypy_boto3_builder/generators/boto3_generator.py b/mypy_boto3_builder/generators/boto3_generator.py
index 529ff7aa..ff7da06c 100644
--- a/mypy_boto3_builder/generators/boto3_generator.py
+++ b/mypy_boto3_builder/generators/boto3_generator.py
@@ -192,12 +192,17 @@ def generate_full_stubs(self) -> Package | None:
                 "*.pyi",
             )
         package.set_description(package.get_short_description("All-in-one type annotations"))
-        self._generate_full_stubs_services(package)
         self.package_writer.write_package(
             package=package,
             template_path=TemplatePath.types_boto3_full,
             static_files_path=None,
         )
+        self._generate_full_stubs_services(package)
+        self.setup_package_writer.write_package(
+            package,
+            template_path=TemplatePath.types_boto3_full,
+            include_template_names=("setup.py.jinja2",),
+        )
         return package
 
     def generate_custom_stubs(self) -> Package:
diff --git a/mypy_boto3_builder/generators/types_boto3_generator.py b/mypy_boto3_builder/generators/types_boto3_generator.py
index 369c30d6..d979cdab 100644
--- a/mypy_boto3_builder/generators/types_boto3_generator.py
+++ b/mypy_boto3_builder/generators/types_boto3_generator.py
@@ -155,12 +155,17 @@ def generate_full_stubs(self) -> Package | None:
                 "*.pyi",
             )
         package.set_description(package.get_short_description("All-in-one type annotations"))
-        self._generate_full_stubs_services(package)
         self.package_writer.write_package(
             package=package,
             template_path=TemplatePath.types_boto3_full,
             static_files_path=None,
         )
+        self._generate_full_stubs_services(package)
+        self.setup_package_writer.write_package(
+            package,
+            template_path=TemplatePath.types_boto3_full,
+            include_template_names=("setup.py.jinja2",),
+        )
         return package
 
     def generate_custom_stubs(self) -> Package: