mschoi/RustPython
1
0
Fork 0
forked from Rust-related/RustPython
Code Pull Requests Actions Activity

MODULE_NAME as const

Browse Source
This commit is contained in:
Jeong YunWon
2020-04-03 08:09:10 +09:00
parent 5ff55f1321
commit 84527b06ac
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
derive/src/pymodule.rs
View File

@@ -180,6 +180,9 @@ pub fn impl_pymodule(attr: AttributeArgs, item: Item) -> Result<TokenStream2, Di
.collect();
let extend_mod = extract_module_items(items)?;
content.1.push(parse_quote! {
const MODULE_NAME: &str = #module_name;
});
content.1.push(parse_quote! {
pub(crate) fn extend_module(
vm: &::rustpython_vm::vm::VirtualMachine,
@@ -193,7 +196,7 @@ pub fn impl_pymodule(attr: AttributeArgs, item: Item) -> Result<TokenStream2, Di
pub(crate) fn make_module(
vm: &::rustpython_vm::vm::VirtualMachine
) -> ::rustpython_vm::pyobject::PyObjectRef {
let module = vm.new_module(#module_name, vm.ctx.new_dict());
let module = vm.new_module(MODULE_NAME, vm.ctx.new_dict());
extend_module(vm, &module);
module
}