134 lines
4.2 KiB
Rust
134 lines
4.2 KiB
Rust
use super::{post, Post};
|
|
use actix_admin::prelude::*;
|
|
use chrono::NaiveDateTime;
|
|
use sea_orm::entity::prelude::*;
|
|
use serde::{Deserialize, Serialize};
|
|
|
|
#[derive(
|
|
Clone,
|
|
Debug,
|
|
PartialEq,
|
|
DeriveEntityModel,
|
|
Deserialize,
|
|
Serialize,
|
|
DeriveActixAdmin,
|
|
DeriveActixAdminModel,
|
|
DeriveActixAdminViewModel,
|
|
)]
|
|
#[sea_orm(table_name = "comment")]
|
|
pub struct Model {
|
|
#[sea_orm(primary_key)]
|
|
#[serde(skip_deserializing)]
|
|
#[actix_admin(primary_key)]
|
|
pub id: i32,
|
|
pub comment: String,
|
|
#[sea_orm(column_type = "Text")]
|
|
#[actix_admin(html_input_type = "email", list_regex_mask = "^([a-zA-Z]*)")]
|
|
pub user: String,
|
|
#[sea_orm(column_type = "DateTime")]
|
|
pub insert_date: DateTime,
|
|
pub is_visible: bool,
|
|
#[actix_admin(select_list = "Post")]
|
|
pub post_id: Option<i32>,
|
|
pub my_decimal: Decimal,
|
|
}
|
|
|
|
#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
|
|
pub enum Relation {
|
|
#[sea_orm(
|
|
belongs_to = "super::post::Entity",
|
|
from = "Column::PostId",
|
|
to = "super::post::Column::Id"
|
|
)]
|
|
Post,
|
|
}
|
|
|
|
impl Related<super::post::Entity> for Entity {
|
|
fn to() -> RelationDef {
|
|
Relation::Post.def()
|
|
}
|
|
}
|
|
|
|
impl ActiveModelBehavior for ActiveModel {}
|
|
|
|
impl ActixAdminModelValidationTrait<ActiveModel> for Entity {
|
|
fn validate(model: &ActiveModel) -> HashMap<String, String> {
|
|
let mut errors = HashMap::new();
|
|
if model.my_decimal.clone().unwrap() < Decimal::from(100 as i16) {
|
|
errors.insert(
|
|
"my_decimal".to_string(),
|
|
"Must be larger than 100".to_string(),
|
|
);
|
|
}
|
|
|
|
errors
|
|
}
|
|
}
|
|
|
|
#[async_trait]
|
|
impl ActixAdminModelFilterTrait<Entity> for Entity {
|
|
fn get_filter() -> Vec<ActixAdminModelFilter<Entity>> {
|
|
vec![
|
|
ActixAdminModelFilter::<Entity> {
|
|
name: "User".to_string(),
|
|
filter_type: ActixAdminModelFilterType::Text,
|
|
filter: |q: sea_orm::Select<Entity>, v| -> sea_orm::Select<Entity> {
|
|
q.apply_if(v, |query, val: String| query.filter(Column::User.eq(val)))
|
|
},
|
|
values: None,
|
|
},
|
|
ActixAdminModelFilter::<Entity> {
|
|
name: "Insert Date After".to_string(),
|
|
filter_type: ActixAdminModelFilterType::DateTime,
|
|
filter: |q: sea_orm::Select<Entity>, v| -> sea_orm::Select<Entity> {
|
|
q.apply_if(v, |query, val: String| {
|
|
query.filter(
|
|
Column::InsertDate.gte(
|
|
NaiveDateTime::parse_from_str(&val, "%Y-%m-%dT%H:%M").unwrap(),
|
|
),
|
|
)
|
|
})
|
|
},
|
|
values: None,
|
|
},
|
|
ActixAdminModelFilter::<Entity> {
|
|
name: "Is Visible".to_string(),
|
|
filter_type: ActixAdminModelFilterType::Checkbox,
|
|
filter: |q: sea_orm::Select<Entity>, v| -> sea_orm::Select<Entity> {
|
|
q.apply_if(v, |query, val: String| {
|
|
query.filter(Column::IsVisible.eq(val))
|
|
})
|
|
},
|
|
values: None,
|
|
},
|
|
ActixAdminModelFilter::<Entity> {
|
|
name: "Post".to_string(),
|
|
filter_type: ActixAdminModelFilterType::SelectList,
|
|
filter: |q: sea_orm::Select<Entity>, v| -> sea_orm::Select<Entity> {
|
|
q.apply_if(v, |query, val: String| query.filter(Column::PostId.eq(val)))
|
|
},
|
|
values: None,
|
|
},
|
|
]
|
|
}
|
|
|
|
async fn get_filter_values(
|
|
filter: &ActixAdminModelFilter<Entity>,
|
|
db: &DatabaseConnection,
|
|
) -> Option<Vec<(String, String)>> {
|
|
match filter.name.as_str() {
|
|
"Post" => Some({
|
|
Post::find()
|
|
.order_by_asc(post::Column::Id)
|
|
.all(db)
|
|
.await
|
|
.unwrap()
|
|
.iter()
|
|
.map(|p| (p.id.to_string(), p.title.to_string()))
|
|
.collect()
|
|
}),
|
|
_ => None,
|
|
}
|
|
}
|
|
}
|