diff --git a/src/database/category.rs b/src/database/category.rs index b8b73fb..083618f 100644 --- a/src/database/category.rs +++ b/src/database/category.rs @@ -4,6 +4,7 @@ use sea_orm::entity::prelude::*; use sea_orm::*; use snafu::prelude::*; +use crate::database::operator::DatabaseOperator; use crate::database::{content_folder, operation::*}; use crate::extractors::normalized_path::*; use crate::extractors::user::User; @@ -64,6 +65,13 @@ impl CategoryOperator { Self { state, user } } + pub fn db(&self) -> DatabaseOperator { + DatabaseOperator { + state: self.state.clone(), + user: self.user.clone(), + } + } + /// List categories /// /// Should not fail, unless SQLite was corrupted for some reason. diff --git a/src/database/content_folder.rs b/src/database/content_folder.rs index 2f37891..4d6ecaa 100644 --- a/src/database/content_folder.rs +++ b/src/database/content_folder.rs @@ -6,6 +6,7 @@ use snafu::prelude::*; use crate::database::category::{self, CategoryError, CategoryOperator}; use crate::database::operation::{Operation, OperationId, OperationLog, OperationType, Table}; +use crate::database::operator::DatabaseOperator; use crate::extractors::user::User; use crate::routes::content_folder::ContentFolderForm; use crate::state::AppState; @@ -64,6 +65,13 @@ impl ContentFolderOperator { Self { state, user } } + pub fn db(&self) -> DatabaseOperator { + DatabaseOperator { + state: self.state.clone(), + user: self.user.clone(), + } + } + /// list All child folders for 1 folder /// /// Should not fail, unless SQLite was corrupted for some reason.