Properly fails on functions inside of declarative modules

PyO3 · Feb 23, 2024 · 391b8f9 · 391b8f9
1 parent beb75da
commit 391b8f9
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pyo3-macros-backend/src/module.rs b/pyo3-macros-backend/src/module.rs
@@ -146,6 +146,8 @@ pub fn pymodule_module_impl(mut module: syn::ItemMod) -> Result<TokenStream> {
                     ensure_spanned!(pymodule_init.is_none(), item_fn.span() => "only one pymodule_init may be specified");
                     let ident = &item_fn.sig.ident;
                     pymodule_init = Some(quote! { #ident(module)?; });
+                } else {
+                    bail_spanned!(item.span() => "only 'use' statements and and pymodule_init functions are allowed in #[pymodule]")
                 }
             }
             item => {