minor refactor on __repr__

PyO3 · Nov 28, 2021 · b53c551 · b53c551
1 parent 9c9e50e
commit b53c551
Showing 1 changed file with 16 additions and 14 deletions.
diff --git a/pyo3-macros-backend/src/pyclass.rs b/pyo3-macros-backend/src/pyclass.rs
@@ -425,23 +425,25 @@ fn impl_enum_class(
         .impl_all();
     let descriptors = unit_variants_as_descriptors(cls, variants.iter().map(|v| v.ident));
 
-    let variants_repr = variants.iter().map(|variant| {
-        let variant_name = variant.ident;
-        // Assuming all variants are unit variants because they are the only type we support.
-        let repr = format!("{}.{}", cls, variant_name);
-        quote! { #cls::#variant_name => #repr, }
-    });
-
-    let default_repr_impl = quote! {
-        #[allow(non_snake_case)]
-        #[pyo3(name = "__repr__")]
-        fn __pyo3__repr__(&self) -> &'static str {
-            match self {
-                #(#variants_repr)*
-                _ => unreachable!("Unsupported variant type."),
+    let default_repr_impl = {
+        let variants_repr = variants.iter().map(|variant| {
+            let variant_name = variant.ident;
+            // Assuming all variants are unit variants because they are the only type we support.
+            let repr = format!("{}.{}", cls, variant_name);
+            quote! { #cls::#variant_name => #repr, }
+        });
+        quote! {
+            #[allow(non_snake_case)]
+            #[pyo3(name = "__repr__")]
+            fn __pyo3__repr__(&self) -> &'static str {
+                match self {
+                    #(#variants_repr)*
+                    _ => unreachable!("Unsupported variant type."),
+                }
             }
         }
     };
+
     let default_impls = gen_default_slot_impls(cls, vec![default_repr_impl]);
     Ok(quote! {