Modify MustUse to iterate the stack frame and collate options from all frames.

Options in closer frames take precedence

Author: robtaylor

amaranth/_unused.py

@@ -18,7 +18,9 @@ class MustUse:
     def __new__(cls, *args, src_loc_at=0, **kwargs):
         frame = sys._getframe(1 + src_loc_at)
         self = super().__new__(cls)
-        self._MustUse__used    = False
+
+        self._MustUse__used  = False
+        self._MustUse__frame = frame
         self._MustUse__context = dict(
             filename=frame.f_code.co_filename,
             lineno=frame.f_lineno,
@@ -31,7 +33,7 @@ def __del__(self):
         if getattr(self._MustUse__warning, "_MustUse__silence", False):
             return
         if hasattr(self, "_MustUse__used") and not self._MustUse__used:
-            if get_linter_option(self._MustUse__context["filename"],
+            if get_linter_option(self._MustUse__frame,
                                  self._MustUse__warning.__qualname__, bool, True):
                 warnings.warn_explicit(
                     f"{self!r} created but never used", self._MustUse__warning,

amaranth/_utils.py

@@ -75,6 +75,7 @@ def decorator_like(*args, **kwargs):
         return decorator_like
 
 
+
 def get_linter_options(filename):
     first_line = linecache.getline(filename, 1)
     if first_line:
@@ -84,8 +85,30 @@ def get_linter_options(filename):
     return dict()
 
 
-def get_linter_option(filename, name, type, default):
-    options = get_linter_options(filename)
+def get_linter_option(frame, name, type, default):
+    """
+    Get given linter option for a given stack frame. This iterates down the frames, collating options of the form:
+
+    .. code::
+        # amaranth: {name}=value
+
+    The earliest option value in the stack takes precedence
+
+    Returns
+    -------
+    :class:`bool` or :class:`int`
+        Option value
+    """
+
+    options = {}
+    while frame:
+        f_opts = get_linter_options(frame.f_code.co_filename)
+        options = f_opts | options
+        if frame.f_back is None:
+            break
+        else:
+            frame = frame.f_back
+
     if name not in options:
         return default