From 2dc3f43b126e7e6fcb2e20a5691ab1b6db8129e8 Mon Sep 17 00:00:00 2001
From: ZelinWang <zelinwang@microsoft.com>
Date: Mon, 16 Jun 2025 16:40:48 +0800
Subject: [PATCH 1/3] update

---
 azure-pipelines.yml          | 32 +++++++++----
 scripts/ci/codegen_report.py | 88 +++++++++++++++++++++++++++++++++++-
 2 files changed, 110 insertions(+), 10 deletions(-)

diff --git a/azure-pipelines.yml b/azure-pipelines.yml
index a0483050983..c64c1f14523 100644
--- a/azure-pipelines.yml
+++ b/azure-pipelines.yml
@@ -1134,8 +1134,9 @@ jobs:
           exit 1
         fi
 
-
 - job: CodegenCoverage
+  condition: in(variables['Build.Reason'], 'BatchedCI', 'IndividualCI')
+  timeoutInMinutes: 180
   displayName: "Codegen Coverage"
   continueOnError: true
   pool:
@@ -1155,14 +1156,29 @@ jobs:
       azdev extension repo add ./azure-cli-extensions
       pip install setuptools==70.0.0 wheel==0.30.0
       azdev extension add "*"
+      pip install msrestazure markupsafe==2.0.1
       # Some extension will change the dependence, so run `azdev setup` again after all extensions installed.
-      azdev setup -c $CLI_REPO_PATH -r ./azure-cli-extensions
-      # CLI repo only
-      azdev statistics list-command-table CLI --statistics-only
-      # CLI + EXT repos
-      pip install jinja2 -U
-      azdev statistics list-command-table --statistics-only > /tmp/codegen_report.json
-      python s/scripts/ci/codegen_report.py
+      azdev setup -c ./s -r ./azure-cli-extensions
+      
+      mkdir -p /tmp/module_stats
+      
+      find /mnt/vss/_work/1/s/src/azure-cli/azure/cli/command_modules/ -mindepth 1 -maxdepth 1 -type d -printf "%f\n" | grep -v '^__pycache__$' > /mnt/vss/_work/1/s/scripts/ci/core_modules.txt
+      echo "=== Core Modules ==="
+      cat /mnt/vss/_work/1/s/scripts/ci/core_modules.txt
+      
+      find /mnt/vss/_work/1/azure-cli-extensions/src/ -mindepth 1 -maxdepth 1 -type d -printf "%f\n" | grep -v '^__pycache__$' > /mnt/vss/_work/1/s/scripts/ci/extension_modules.txt
+      echo "=== Extension Modules ==="
+      cat /mnt/vss/_work/1/s/scripts/ci/extension_modules.txt
+      
+      for module in $(cat /mnt/vss/_work/1/s/scripts/ci/core_modules.txt); do
+        azdev statistics list-command-table $module --statistics-only > /tmp/module_stats/${module}.json || true
+      done
+      
+      for module in $(cat /mnt/vss/_work/1/s/scripts/ci/extension_modules.txt); do
+        azdev statistics list-command-table $module --statistics-only > /tmp/module_stats/${module}.json || true
+      done
+      azdev statistics list-command-table --statistics-only > /tmp/codegen_report.json || true
+      python /mnt/vss/_work/1/s/scripts/ci/codegen_report.py
     env:
       BUILD_ID: $(Build.BuildId)
       BUILD_BRANCH: $(Build.SourceBranchName)
diff --git a/scripts/ci/codegen_report.py b/scripts/ci/codegen_report.py
index 9091da5b6a9..82844b78b18 100644
--- a/scripts/ci/codegen_report.py
+++ b/scripts/ci/codegen_report.py
@@ -19,6 +19,88 @@
 BUILD_ID = os.environ.get('BUILD_ID', None)
 BUILD_BRANCH = os.environ.get('BUILD_BRANCH', None)
 
+def load_module_stats():
+    stats_dir = "/tmp/module_stats"
+    all_stats = {}
+
+    with open("/mnt/vss/_work/1/s/scripts/ci/core_modules.txt", "r") as f:
+        core_modules = [line.strip() for line in f.readlines()]
+
+    with open("/mnt/vss/_work/1/s/scripts/ci/extension_modules.txt", "r") as f:
+        extension_modules = [line.strip() for line in f.readlines()]
+
+    for module in core_modules + extension_modules:
+        stats_file = os.path.join(stats_dir, f"{module}.json")
+        if os.path.exists(stats_file):
+            with open(stats_file, "r") as f:
+                try:
+                    stats = json.load(f)
+                    codegenV1 = stats.get("codegenV1", 0)
+                    codegenV2 = stats.get("codegenV2", 0)
+                    total = stats.get("total", 0)
+                    manual = total - codegenV1 - codegenV2
+                    all_stats[module] = {
+                        "codegenV1": codegenV1,
+                        "codegenV2": codegenV2,
+                        "manual": manual,
+                        "total": total,
+                        "type": "core" if module in core_modules else "extension"
+                    }
+                except json.JSONDecodeError:
+                    print(f"Warning: Could not parse {stats_file}")
+    return all_stats
+
+def analyze_stats(all_stats):
+    counters = {
+        "manual": {"core": 0, "extension": 0},
+        "mixed": {"core": 0, "extension": 0},
+        "codegen": {"core": 0, "extension": 0},
+        "codegenV1": {"core": 0, "extension": 0},
+        "total": {"core": 0, "extension": 0}
+    }
+    for _, stats in all_stats.items():
+        module_type = stats["type"]
+        counters["total"][module_type] += 1
+        if stats["manual"] > 0 and (stats["codegenV1"] > 0 or stats["codegenV2"] > 0):
+            counters["mixed"][module_type] += 1
+        if stats["codegenV1"] > 0 or stats["codegenV2"] > 0:
+            counters["codegen"][module_type] += 1
+        if stats["codegenV1"] > 0:
+            counters["codegenV1"][module_type] += 1
+    counters["manual"]["core"] = counters["total"]["core"] - counters["codegen"]["core"]
+    counters["manual"]["extension"] = counters["total"]["extension"] - counters["codegen"]["extension"]
+    return counters
+
+def print_results(counters):
+    logger.info("\n===== Codegen Coverage Report =====")
+    logger.info("\n1. Manual Modules:")
+    logger.info(f"   Core: {counters['manual']['core']}")
+    logger.info(f"   Extension: {counters['manual']['extension']}")
+    logger.info("\n2. Mixed Modules:")
+    logger.info(f"   Core: {counters['mixed']['core']}")
+    logger.info(f"   Extension: {counters['mixed']['extension']}")
+    logger.info("\n3. Codegen Modules:")
+    logger.info(f"   Core: {counters['codegen']['core']}")
+    logger.info(f"   Extension: {counters['codegen']['extension']}")
+    logger.info("\n4. CodegenV1 Modules:")
+    logger.info(f"   Core: {counters['codegenV1']['core']}")
+    logger.info(f"   Extension: {counters['codegenV1']['extension']}")
+    logger.info("\n5. Total Modules:")
+    logger.info(f"   Core: {counters['total']['core']}")
+    logger.info(f"   Extension: {counters['total']['extension']}")
+
+def analyze_and_report():
+    all_stats = load_module_stats()
+    counters = analyze_stats(all_stats)
+    print_results(counters)
+    output = {
+        "detailed_stats": all_stats,
+        "summary": counters
+    }
+    logger.info("\n=== Detailed JSON Output ===")
+    logger.info(json.dumps(output, indent=2))
+    with open("/tmp/module_stats_summary.json", "w") as f:
+        json.dump(output, f, indent=2)
 
 def generate_csv():
     data = []
@@ -31,9 +113,11 @@ def generate_csv():
     is_release = True if BUILD_BRANCH == 'release' else False
     date = (datetime.datetime.utcnow() + datetime.timedelta(hours=8)).strftime("%Y-%m-%d")
     data.append([BUILD_ID, manual, codegenv1, codegenv2, total, is_release, date])
-    logger.info(f'Finish generate data for codegen report: {data}')
+    logger.info('Finish generate data for codegen report:')
+    logger.info("BUILD_ID, manual, codegenv1, codegenv2, total, is_release, date")
+    logger.info(f'{data}')
     return data
 
-
 if __name__ == '__main__':
+    analyze_and_report()
     generate_csv()

From 9d722883131386de3f8f74bb906f2028ac3c13f2 Mon Sep 17 00:00:00 2001
From: ZelinWang <zelinwang@microsoft.com>
Date: Mon, 16 Jun 2025 16:45:51 +0800
Subject: [PATCH 2/3] update

---
 azure-pipelines.yml | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/azure-pipelines.yml b/azure-pipelines.yml
index c64c1f14523..c55a61b8227 100644
--- a/azure-pipelines.yml
+++ b/azure-pipelines.yml
@@ -1159,21 +1159,21 @@ jobs:
       pip install msrestazure markupsafe==2.0.1
       # Some extension will change the dependence, so run `azdev setup` again after all extensions installed.
       azdev setup -c ./s -r ./azure-cli-extensions
-      
+
       mkdir -p /tmp/module_stats
-      
+
       find /mnt/vss/_work/1/s/src/azure-cli/azure/cli/command_modules/ -mindepth 1 -maxdepth 1 -type d -printf "%f\n" | grep -v '^__pycache__$' > /mnt/vss/_work/1/s/scripts/ci/core_modules.txt
       echo "=== Core Modules ==="
       cat /mnt/vss/_work/1/s/scripts/ci/core_modules.txt
-      
+
       find /mnt/vss/_work/1/azure-cli-extensions/src/ -mindepth 1 -maxdepth 1 -type d -printf "%f\n" | grep -v '^__pycache__$' > /mnt/vss/_work/1/s/scripts/ci/extension_modules.txt
       echo "=== Extension Modules ==="
       cat /mnt/vss/_work/1/s/scripts/ci/extension_modules.txt
-      
+
       for module in $(cat /mnt/vss/_work/1/s/scripts/ci/core_modules.txt); do
         azdev statistics list-command-table $module --statistics-only > /tmp/module_stats/${module}.json || true
       done
-      
+
       for module in $(cat /mnt/vss/_work/1/s/scripts/ci/extension_modules.txt); do
         azdev statistics list-command-table $module --statistics-only > /tmp/module_stats/${module}.json || true
       done

From 96be1da6e3c2a2a2d49d067c03f6dcc09edf548f Mon Sep 17 00:00:00 2001
From: ZelinWang <zelinwang@microsoft.com>
Date: Mon, 16 Jun 2025 16:47:22 +0800
Subject: [PATCH 3/3] update

---
 scripts/ci/codegen_report.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/scripts/ci/codegen_report.py b/scripts/ci/codegen_report.py
index 82844b78b18..b6e3810fb7d 100644
--- a/scripts/ci/codegen_report.py
+++ b/scripts/ci/codegen_report.py
@@ -47,7 +47,7 @@ def load_module_stats():
                         "type": "core" if module in core_modules else "extension"
                     }
                 except json.JSONDecodeError:
-                    print(f"Warning: Could not parse {stats_file}")
+                    logger.info(f"Warning: Could not parse {stats_file}")
     return all_stats
 
 def analyze_stats(all_stats):