From 562778f61d52ac36d89eef66d6de3e45f0787e17 Mon Sep 17 00:00:00 2001 From: Chris Tsang Date: Wed, 29 May 2024 01:01:12 +0100 Subject: [PATCH] One last refactor --- sea-orm-macros/src/derives/active_enum.rs | 12 +++++------- 1 file changed, 5 insertions(+), 7 deletions(-) diff --git a/sea-orm-macros/src/derives/active_enum.rs b/sea-orm-macros/src/derives/active_enum.rs index c67e8f58d..b3be14733 100644 --- a/sea-orm-macros/src/derives/active_enum.rs +++ b/sea-orm-macros/src/derives/active_enum.rs @@ -40,7 +40,7 @@ impl ActiveEnum { let mut db_type = Err(Error::TT(quote_spanned! { ident_span => compile_error!("Missing macro attribute `db_type`"); })); - let mut rename_all_rule = None; + let mut rename_all = None; input .attrs @@ -72,7 +72,7 @@ impl ActiveEnum { let litstr: LitStr = meta.value()?.parse()?; enum_name = litstr.value(); } else if meta.path.is_ident("rename_all") { - rename_all_rule = Some((&meta).try_into()?); + rename_all = Some((&meta).try_into()?); } else { return Err(meta.error(format!( "Unknown attribute parameter found: {:?}", @@ -89,7 +89,7 @@ impl ActiveEnum { _ => return Err(Error::InputNotEnum), }; - let mut is_string = rename_all_rule.is_some(); + let mut is_string = rename_all.is_some(); let mut is_int = false; let mut variants = Vec::new(); @@ -135,9 +135,7 @@ impl ActiveEnum { })); } - if string_value.is_none() - && num_value.is_none() - && rename_rule.or(rename_all_rule).is_none() + if string_value.is_none() && num_value.is_none() && rename_rule.or(rename_all).is_none() { match variant.discriminant { Some((_, Expr::Lit(exprlit))) => { @@ -191,7 +189,7 @@ impl ActiveEnum { db_type: db_type?, is_string, variants, - rename_all: rename_all_rule, + rename_all, }) }