sea_schema/postgres/writer/enumeration.rs
1use crate::postgres::def::EnumDef;
2use sea_query::{
3 extension::postgres::{Type, TypeCreateStatement},
4 Alias,
5};
6
7impl EnumDef {
8 /// Converts the [EnumDef] to a [TypeCreateStatement]
9 pub fn write(&self) -> TypeCreateStatement {
10 Type::create()
11 .as_enum(Alias::new(self.typename.as_str()))
12 .values(self.values.iter().map(|val| Alias::new(val.as_str())))
13 .to_owned()
14 }
15}