fasthidden & static attributes

youknowone · youknowone · commit 8f79b4e75011 · 2025-07-11T22:01:11.000+09:00
diff --git a/compiler/codegen/src/compile.rs b/compiler/codegen/src/compile.rs
@@ -8,10 +8,10 @@
 #![deny(clippy::cast_possible_truncation)]
 
 use crate::{
-    IndexSet, ToPythonName,
+    IndexMap, IndexSet, ToPythonName,
     error::{CodegenError, CodegenErrorType, PatternUnreachableReason},
     ir::{self, BlockIdx},
-    symboltable::{self, SymbolFlags, SymbolScope, SymbolTable},
+    symboltable::{self, SymbolFlags, SymbolScope, SymbolTable, SymbolTableType},
     unparse::unparse_expr,
 };
 use itertools::Itertools;
@@ -318,6 +318,8 @@ impl<'src> Compiler<'src> {
             varname_cache: IndexSet::default(),
             cellvar_cache: IndexSet::default(),
             freevar_cache: IndexSet::default(),
+            fasthidden_cache: IndexMap::default(),
+            static_attributes: None,
         };
         Compiler {
             code_stack: vec![module_code],
@@ -382,6 +384,9 @@ impl Compiler<'_> {
         let source_path = self.source_code.path.to_owned();
         let first_line_number = self.get_source_line_number();
 
+        // Get the private name from current scope if exists
+        let private = self.code_stack.last().and_then(|info| info.private.clone());
+
         let table = self.push_symbol_table();
 
         let cellvar_cache = table
@@ -400,17 +405,13 @@ impl Compiler<'_> {
             .collect();
 
         // Initialize varname_cache from SymbolTable::varnames
-        let varname_cache: IndexSet<String> = table
-            .varnames
-            .iter()
-            .cloned()
-            .collect();
+        let varname_cache: IndexSet<String> = table.varnames.iter().cloned().collect();
 
         // Qualname will be set later by set_qualname
         let qualname = None;
 
-        // Get the private name from current scope if exists
-        let private = self.code_stack.last().and_then(|info| info.private.clone());
+        // Check if this is a class scope
+        let is_class_scope = table.typ == SymbolTableType::Class;
 
         let info = ir::CodeInfo {
             flags,
@@ -430,6 +431,12 @@ impl Compiler<'_> {
             varname_cache,
             cellvar_cache,
             freevar_cache,
+            fasthidden_cache: IndexMap::default(),
+            static_attributes: if is_class_scope {
+                Some(IndexSet::default())
+            } else {
+                None
+            },
         };
         self.code_stack.push(info);
     }
diff --git a/compiler/codegen/src/ir.rs b/compiler/codegen/src/ir.rs
@@ -1,7 +1,7 @@
 use std::ops;
 
-use crate::IndexSet;
 use crate::error::InternalError;
+use crate::{IndexMap, IndexSet};
 use ruff_source_file::{OneIndexed, SourceLocation};
 use rustpython_compiler_core::bytecode::{
     CodeFlags, CodeObject, CodeUnit, ConstantData, InstrDisplayContext, Instruction, Label, OpArg,
@@ -83,6 +83,13 @@ pub struct CodeInfo {
     pub varname_cache: IndexSet<String>,
     pub cellvar_cache: IndexSet<String>,
     pub freevar_cache: IndexSet<String>,
+
+    // For tracking hidden fast locals (used in comprehensions)
+    // Maps variable name to bool (true if currently fast, false if was fast)
+    pub fasthidden_cache: IndexMap<String, bool>,
+
+    // For class scopes: attributes accessed via self.X
+    pub static_attributes: Option<IndexSet<String>>,
 }
 impl CodeInfo {
     pub fn finalize_code(mut self, optimize: u8) -> crate::InternalResult<CodeObject> {
@@ -111,6 +118,8 @@ impl CodeInfo {
             varname_cache,
             cellvar_cache,
             freevar_cache,
+            fasthidden_cache: _,
+            static_attributes: _,
         } = self;
 
         let mut instructions = Vec::new();
