From 0488436c5831564a9746707ec3f54e403fffbf98 Mon Sep 17 00:00:00 2001
From: Anderson Bravalheri <andersonbravalheri@gmail.com>
Date: Sun, 19 Jun 2022 08:26:34 +0100
Subject: [PATCH 1/3] test_expand: Add example for annotated assignment without
 value

---
 setuptools/tests/config/test_expand.py | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

diff --git a/setuptools/tests/config/test_expand.py b/setuptools/tests/config/test_expand.py
index a1f543861e..523779a8ed 100644
--- a/setuptools/tests/config/test_expand.py
+++ b/setuptools/tests/config/test_expand.py
@@ -85,13 +85,17 @@ def test_read_attr(self, tmp_path, monkeypatch):
         values = expand.read_attr('lib.mod.VALUES', {'lib': 'pkg/sub'}, tmp_path)
         assert values['c'] == (0, 1, 1)
 
-    def test_read_annotated_attr(self, tmp_path):
+    @pytest.mark.parametrize(
+        "example",
+        [
+            "VERSION: str\nVERSION = '0.1.1'\nraise SystemExit(1)\n",
+            "VERSION: str = '0.1.1'\nraise SystemExit(1)\n",
+        ]
+    )
+    def test_read_annotated_attr(self, tmp_path, example):
         files = {
             "pkg/__init__.py": "",
-            "pkg/sub/__init__.py": (
-                "VERSION: str = '0.1.1'\n"
-                "raise SystemExit(1)\n"
-            ),
+            "pkg/sub/__init__.py": example,
         }
         write_files(files, tmp_path)
         # Make sure this attribute can be read statically

From 6384f26597800c89e5605931a306fbb68127ce13 Mon Sep 17 00:00:00 2001
From: Anderson Bravalheri <andersonbravalheri@gmail.com>
Date: Sun, 19 Jun 2022 08:19:51 +0100
Subject: [PATCH 2/3] config.expand: Refactor StaticModule

---
 setuptools/config/expand.py | 25 ++++++++++++-------------
 1 file changed, 12 insertions(+), 13 deletions(-)

diff --git a/setuptools/config/expand.py b/setuptools/config/expand.py
index a0dd7c2a3f..75af6018c8 100644
--- a/setuptools/config/expand.py
+++ b/setuptools/config/expand.py
@@ -66,24 +66,23 @@ def __init__(self, name: str, spec: ModuleSpec):
         vars(self).update(locals())
         del self.self
 
+    def _find_assignments(self) -> Iterator[Tuple[ast.AST, Optional[ast.AST]]]:
+        for statement in self.module.body:
+            if isinstance(statement, ast.Assign):
+                yield from ((target, statement.value) for target in statement.targets)
+            elif isinstance(statement, ast.AnnAssign):
+                yield (statement.target, statement.value)
+
     def __getattr__(self, attr):
         """Attempt to load an attribute "statically", via :func:`ast.literal_eval`."""
         try:
-            for statement in self.module.body:
-                if isinstance(statement, ast.Assign):
-                    targets = statement.targets
-                    value = statement.value
-                elif isinstance(statement, ast.AnnAssign):
-                    targets = [statement.target]
-                    value = statement.value
-                else:
-                    continue
-                for target in targets:
-                    if isinstance(target, ast.Name) and target.id == attr:
-                        return ast.literal_eval(value)
+            return next(
+                ast.literal_eval(value)
+                for target, value in self._find_assignments()
+                if isinstance(target, ast.Name) and target.id == attr
+            )
         except Exception as e:
             raise AttributeError(f"{self.name} has no attribute {attr}") from e
-        raise AttributeError(f"{self.name} has no attribute {attr}")
 
 
 def glob_relative(

From 0b1d090e48b16bff59e6e9e1d33b0d6dca65a7ee Mon Sep 17 00:00:00 2001
From: Anderson Bravalheri <andersonbravalheri@gmail.com>
Date: Sun, 19 Jun 2022 08:21:25 +0100
Subject: [PATCH 3/3] config.expand.StaticModule: handle scenarios when
 annotated assignment does not have a value

---
 setuptools/config/expand.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/setuptools/config/expand.py b/setuptools/config/expand.py
index 75af6018c8..ed7564047a 100644
--- a/setuptools/config/expand.py
+++ b/setuptools/config/expand.py
@@ -66,11 +66,11 @@ def __init__(self, name: str, spec: ModuleSpec):
         vars(self).update(locals())
         del self.self
 
-    def _find_assignments(self) -> Iterator[Tuple[ast.AST, Optional[ast.AST]]]:
+    def _find_assignments(self) -> Iterator[Tuple[ast.AST, ast.AST]]:
         for statement in self.module.body:
             if isinstance(statement, ast.Assign):
                 yield from ((target, statement.value) for target in statement.targets)
-            elif isinstance(statement, ast.AnnAssign):
+            elif isinstance(statement, ast.AnnAssign) and statement.value:
                 yield (statement.target, statement.value)
 
     def __getattr__(self, attr):