From 018b4565f2dd0d8136296ed95f9f322268861e4a Mon Sep 17 00:00:00 2001 From: RichardWGNR <171420035+RichardWGNR@users.noreply.github.com> Date: Sat, 13 Jul 2024 08:43:08 +0500 Subject: [PATCH] Skip Rust's escaping (#793) --- sea-query-attr/src/lib.rs | 16 ++++++++++++---- 1 file changed, 12 insertions(+), 4 deletions(-) diff --git a/sea-query-attr/src/lib.rs b/sea-query-attr/src/lib.rs index 8b67a43c..0070413b 100644 --- a/sea-query-attr/src/lib.rs +++ b/sea-query-attr/src/lib.rs @@ -58,10 +58,18 @@ pub fn enum_def(args: TokenStream, input: TokenStream) -> TokenStream { .iter() .map(|field| { let ident = &field.ident; - let string = ident - .as_ref() - .expect("#[enum_def] can only be used on structs with named fields") - .to_string(); + let string = { + let mut inner = ident + .as_ref() + .expect("#[enum_def] can only be used on structs with named fields") + .to_string(); + + if inner.starts_with("r#") { + inner = (&inner[2..]).to_string(); + } + + inner + }; let as_pascal = string.to_pascal_case(); NamingHolder { default: ident.as_ref().unwrap().clone(),