mirrors/RustPython-Parser
1
0
Fork 0
mirror of https://github.com/RustPython/Parser.git synced 2025-07-16 01:25:25 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix accessing __doc__ while inside a class definition

Browse source
This commit is contained in:
Noah 2020-09-28 23:52:59 -05:00
parent 09b8414162
commit 16a896d513
2 changed files with 14 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

22
src/compile.rs
View file

@ -989,7 +989,13 @@ impl<O: OutputStream> Compiler<O> {
// Turn code object into function object:
self.emit(Instruction::MakeFunction);
self.store_docstring(doc_str);
self.emit(Instruction::Duplicate);
self.load_docstring(doc_str);
self.emit(Instruction::Rotate { amount: 2 });
self.emit(Instruction::StoreAttr {
name: "__doc__".to_owned(),
});
self.apply_decorators(decorator_list);
self.store_name(name);
@ -1109,6 +1115,11 @@ impl<O: OutputStream> Compiler<O> {
name: "__qualname__".to_owned(),
scope: bytecode::NameScope::Free,
});
self.load_docstring(doc_str);
self.emit(Instruction::StoreName {
name: "__doc__".to_owned(),
scope: bytecode::NameScope::Free,
});
// setup annotations
if self.find_ann(body) {
self.emit(Instruction::SetupAnnotation);
@ -1175,7 +1186,6 @@ impl<O: OutputStream> Compiler<O> {
});
}
self.store_docstring(doc_str);
self.apply_decorators(decorator_list);
self.store_name(name);
@ -1184,10 +1194,9 @@ impl<O: OutputStream> Compiler<O> {
Ok(())
}
fn store_docstring(&mut self, doc_str: Option<String>) {
fn load_docstring(&mut self, doc_str: Option<String>) {
// TODO: __doc__ must be default None and no bytecodes unless it is Some
// Duplicate top of stack (the function or class object)
self.emit(Instruction::Duplicate);
// Doc string value:
self.emit(Instruction::LoadConst {
@ -1196,11 +1205,6 @@ impl<O: OutputStream> Compiler<O> {
None => bytecode::Constant::None, // set docstring None if not declared
},
});
self.emit(Instruction::Rotate { amount: 2 });
self.emit(Instruction::StoreAttr {
name: "__doc__".to_owned(),
});
}
fn compile_while(

1
src/symboltable.rs
View file

@ -531,6 +531,7 @@ impl SymbolTableBuilder {
self.enter_scope(name, SymbolTableType::Class, statement.location.row());
self.register_name("__module__", SymbolUsage::Assigned)?;
self.register_name("__qualname__", SymbolUsage::Assigned)?;
self.register_name("__doc__", SymbolUsage::Assigned)?;
self.scan_statements(body)?;
self.leave_scope();
self.scan_expressions(bases, &ExpressionContext::Load)?;