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

CLI generate enum with numeric like variants

Browse Source
This commit is contained in:
Billy Chan 2022-03-08 17:50:01 +08:00
parent 27143a8ddc
commit 4ee6eac313
No known key found for this signature in database
GPG Key ID: A2D690CAC7DF3CC7
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
sea-orm-codegen/src/entity/active_enum.rs
View File

@ -14,10 +14,13 @@ impl ActiveEnum {
let enum_name = &self.enum_name;
let enum_iden = format_ident!("{}", enum_name.to_camel_case());
let values = &self.values;
let variants = self
.values
.iter()
.map(|v| format_ident!("{}", v.to_camel_case()));
let variants = self.values.iter().map(|v| v.trim()).map(|v| {
if v.chars().all(|c| c.is_numeric()) {
format_ident!("_{}", v)
} else {
format_ident!("{}", v.to_camel_case())
}
});
let extra_derive = with_serde.extra_derive();