caisin/sea-orm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

One last refactor

Browse Source
This commit is contained in:
Chris Tsang 2024-05-29 01:01:12 +01:00
parent e3f64b1892
commit 562778f61d
1 changed files with 5 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
sea-orm-macros/src/derives/active_enum.rs
View File

@ -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,
})
}