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

Merge pull request #1580 from seeeturtle/doc-f

Browse source 
Set docstring of function as None if not declared
This commit is contained in:
Aviv Palivoda 2019-11-02 19:18:11 +02:00 • committed by GitHub
commit 47c8f3756b
1 changed files with 14 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

27
src/compile.rs
View file

@ -1015,22 +1015,23 @@ impl<O: OutputStream> Compiler<O> {
}
fn store_docstring(&mut self, doc_str: Option<String>) {
if let Some(doc_string) = doc_str {
// Duplicate top of stack (the function or class object)
self.emit(Instruction::Duplicate);
// Duplicate top of stack (the function or class object)
self.emit(Instruction::Duplicate);
// Doc string value:
self.emit(Instruction::LoadConst {
value: bytecode::Constant::String {
value: doc_string.to_string(),
// Doc string value:
self.emit(Instruction::LoadConst {
value: match doc_str {
Some(doc) => bytecode::Constant::String {
value: doc.to_string(),
},
});
None => bytecode::Constant::None, // set docstring None if not declared
},
});
self.emit(Instruction::Rotate { amount: 2 });
self.emit(Instruction::StoreAttr {
name: "__doc__".to_string(),
});
}
self.emit(Instruction::Rotate { amount: 2 });
self.emit(Instruction::StoreAttr {
name: "__doc__".to_string(),
});
}
fn compile_while(