From 84527b06acd1722c6b5ce962e6945642d69c64bc Mon Sep 17 00:00:00 2001
From: Jeong YunWon <jeong@youknowone.org>
Date: Fri, 3 Apr 2020 08:09:10 +0900
Subject: [PATCH] MODULE_NAME as const

---
 derive/src/pymodule.rs | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/derive/src/pymodule.rs b/derive/src/pymodule.rs
index ac52db8d4..112715213 100644
--- a/derive/src/pymodule.rs
+++ b/derive/src/pymodule.rs
@@ -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
                     }