qnty 0.1.1__py3-none-any.whl → 0.1.3__py3-none-any.whl

qnty/problems/problem.py

@@ -714,6 +714,20 @@ class Problem(ValidationMixin):
 
         super().__setattr__(name, value)
 
+    def __getattr__(self, name: str) -> Any:
+        """Dynamic attribute access for composed variables and other attributes."""
+        # Avoid recursion by checking the dict directly instead of using hasattr
+        try:
+            variables = object.__getattribute__(self, "variables")
+            if name in variables:
+                return variables[name]
+        except AttributeError:
+            # variables attribute doesn't exist yet (during initialization)
+            pass
+        
+        # If not found, raise AttributeError
+        raise AttributeError(f"'{type(self).__name__}' object has no attribute '{name}'")
+
     def __getitem__(self, key: str):
         """Allow dict-like access to variables."""
         return self.get_variable(key)

qnty/quantities/field_qnty.py

@@ -738,7 +738,7 @@ class ExpressionMixin:
                 continue
 
         # Strategy 2: Look for Equation objects in scope (created with .equals())
-        frame = ScopeDiscoveryService._find_user_frame(getattr(ScopeDiscoveryService, '_get_current_frame', lambda: None)())  # type: ignore[misc]
+        frame = ScopeDiscoveryService._get_cached_user_frame()
         if frame:
             for obj in list(frame.f_locals.values()) + list(frame.f_globals.values()):
                 if isinstance(obj, Equation):
@@ -775,8 +775,20 @@ class SetterCompatibilityMixin:
         if not hasattr(self, "_setter_class"):
             self._setter_class: type | None = None
 
-    def set(self, value: float) -> TypeSafeSetter:
+    def set(self, value: float, unit: str | None = None) -> 'Self | TypeSafeSetter':
         """Create setter object for fluent API compatibility."""
+        if unit is not None:
+            # Handle direct unit setting using the specialized setter
+            if hasattr(self, "_setter_class") and self._setter_class:
+                setter = self._setter_class(self, value)
+                if hasattr(setter, unit):
+                    getattr(setter, unit)
+                    return self
+                else:
+                    raise ValueError(f"Unknown unit: {unit}")
+            else:
+                raise ValueError("Unit parameter not supported for this variable type")
+        
         if hasattr(self, "_setter_class") and self._setter_class:
             return self._setter_class(self, value)  # Correct parameter order
         else: