mirrors/RustPython-Parser
1
0
Fork 0
mirror of https://github.com/RustPython/Parser.git synced 2025-07-15 00:55:22 +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

9
src/compile.rs
View file

@ -1015,14 +1015,16 @@ impl<O: OutputStream> Compiler<O> {
} }
fn store_docstring(&mut self, doc_str: Option<String>) { 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) // Duplicate top of stack (the function or class object)
self.emit(Instruction::Duplicate); self.emit(Instruction::Duplicate);
// Doc string value: // Doc string value:
self.emit(Instruction::LoadConst { self.emit(Instruction::LoadConst {
value: bytecode::Constant::String { value: match doc_str {
value: doc_string.to_string(), Some(doc) => bytecode::Constant::String {
value: doc.to_string(),
},
None => bytecode::Constant::None, // set docstring None if not declared
}, },
}); });
@ -1031,7 +1033,6 @@ impl<O: OutputStream> Compiler<O> {
name: "__doc__".to_string(), name: "__doc__".to_string(),
}); });
} }
}
fn compile_while( fn compile_while(
&mut self, &mut self,