Skip to content

Commit a25b3f4

Browse files
DataEraserCDataEraserC
committed
code fmt
1 parent b6816d9 commit a25b3f4

20 files changed

Lines changed: 654 additions & 599 deletions

‎file_classification_cli/src/bin/create_file.rs‎

Lines changed: 28 additions & 28 deletions
Original file line numberDiff line numberDiff line change
@@ -3,34 +3,34 @@ use file_classification_core::establish_connection;
33
use std::io::stdin;
44

55
fn main() {
6-
let connection = &mut establish_connection();
7-
8-
let mut name = String::new();
9-
let mut type_ = String::new();
10-
let mut path = String::new();
11-
12-
println!("Please input File Name:");
13-
stdin().read_line(&mut type_).unwrap();
14-
let type_ = type_.trim_end();
15-
16-
println!("Please input File Type:");
17-
stdin().read_line(&mut name).unwrap();
18-
let name = name.trim_end();
19-
20-
println!("Please input File Path:");
21-
stdin().read_line(&mut path).unwrap();
22-
let path = path.trim_end();
23-
24-
let result = create_file(connection, name, path, &type_);
25-
match result {
26-
Ok((file, group)) => {
27-
println!("\nSaved file {path} with id {}", file.id);
28-
println!("Saved group {type_} with id {}", group.id);
29-
}
30-
Err(e) => {
31-
eprintln!("An error occurred: {}", e);
32-
}
33-
}
6+
let connection = &mut establish_connection();
7+
8+
let mut name = String::new();
9+
let mut type_ = String::new();
10+
let mut path = String::new();
11+
12+
println!("Please input File Name:");
13+
stdin().read_line(&mut type_).unwrap();
14+
let type_ = type_.trim_end();
15+
16+
println!("Please input File Type:");
17+
stdin().read_line(&mut name).unwrap();
18+
let name = name.trim_end();
19+
20+
println!("Please input File Path:");
21+
stdin().read_line(&mut path).unwrap();
22+
let path = path.trim_end();
23+
24+
let result = create_file(connection, name, path, &type_);
25+
match result {
26+
Ok((file, group)) => {
27+
println!("\nSaved file {path} with id {}", file.id);
28+
println!("Saved group {type_} with id {}", group.id);
29+
}
30+
Err(e) => {
31+
eprintln!("An error occurred: {}", e);
32+
}
33+
}
3434
}
3535

3636
#[allow(dead_code)]

‎file_classification_cli/src/bin/create_file_group.rs‎

Lines changed: 17 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -3,30 +3,27 @@ use file_classification_core::establish_connection;
33
use std::io::{stdin, stdout, Write};
44

55
fn main() {
6-
let connection = &mut establish_connection();
6+
let connection = &mut establish_connection();
77

8-
let mut file_id_input = String::new();
9-
let mut group_id_input = String::new();
8+
let mut file_id_input = String::new();
9+
let mut group_id_input = String::new();
1010

11-
print!("Please input File ID: ");
12-
stdout().flush().unwrap();
13-
stdin().read_line(&mut file_id_input).unwrap();
14-
let file_id: i32 = file_id_input.trim().parse().expect("Invalid File ID");
11+
print!("Please input File ID: ");
12+
stdout().flush().unwrap();
13+
stdin().read_line(&mut file_id_input).unwrap();
14+
let file_id: i32 = file_id_input.trim().parse().expect("Invalid File ID");
1515

16-
print!("Please input Group ID: ");
17-
stdout().flush().unwrap();
18-
stdin().read_line(&mut group_id_input).unwrap();
19-
let group_id: i32 = group_id_input.trim().parse().expect("Invalid Group ID");
16+
print!("Please input Group ID: ");
17+
stdout().flush().unwrap();
18+
stdin().read_line(&mut group_id_input).unwrap();
19+
let group_id: i32 = group_id_input.trim().parse().expect("Invalid Group ID");
2020

21-
let result = create_file_group(connection, file_id, group_id);
22-
if result.is_ok() {
23-
println!("\nFile ID {} added to Group ID {}", file_id, group_id);
24-
} else {
25-
println!(
26-
"\nFailed to add File ID {} to Group ID {}",
27-
file_id, group_id
28-
);
29-
}
21+
let result = create_file_group(connection, file_id, group_id);
22+
if result.is_ok() {
23+
println!("\nFile ID {} added to Group ID {}", file_id, group_id);
24+
} else {
25+
println!("\nFailed to add File ID {} to Group ID {}", file_id, group_id);
26+
}
3027
}
3128

3229
#[allow(dead_code)]

‎file_classification_cli/src/bin/create_group.rs‎

Lines changed: 14 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -3,23 +3,23 @@ use file_classification_core::establish_connection;
33
use std::io::stdin;
44

55
fn main() {
6-
let connection = &mut establish_connection();
6+
let connection = &mut establish_connection();
77

8-
let mut name = String::new();
8+
let mut name = String::new();
99

10-
println!("Please input Group Name:");
11-
stdin().read_line(&mut name).unwrap();
12-
let name = name.trim_end();
10+
println!("Please input Group Name:");
11+
stdin().read_line(&mut name).unwrap();
12+
let name = name.trim_end();
1313

14-
let group = create_group(connection, name);
15-
match group {
16-
Ok(group) => {
17-
println!("\nGroup {} created successfully, ID: {}", group.name, group.id);
18-
}
19-
Err(e) => {
20-
println!("\nError creating group: {}", e);
21-
}
22-
}
14+
let group = create_group(connection, name);
15+
match group {
16+
Ok(group) => {
17+
println!("\nGroup {} created successfully, ID: {}", group.name, group.id);
18+
}
19+
Err(e) => {
20+
println!("\nError creating group: {}", e);
21+
}
22+
}
2323
}
2424

2525
#[allow(dead_code)]

‎file_classification_cli/src/bin/create_group_tag.rs‎

Lines changed: 17 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -3,27 +3,27 @@ use file_classification_core::establish_connection;
33
use std::io::{stdin, stdout, Write};
44

55
fn main() {
6-
let connection = &mut establish_connection();
6+
let connection = &mut establish_connection();
77

8-
let mut group_id_input = String::new();
9-
let mut tag_id_input = String::new();
8+
let mut group_id_input = String::new();
9+
let mut tag_id_input = String::new();
1010

11-
print!("Please input Group ID: ");
12-
stdout().flush().unwrap();
13-
stdin().read_line(&mut group_id_input).unwrap();
14-
let group_id: i32 = group_id_input.trim().parse().expect("Invalid Group ID");
11+
print!("Please input Group ID: ");
12+
stdout().flush().unwrap();
13+
stdin().read_line(&mut group_id_input).unwrap();
14+
let group_id: i32 = group_id_input.trim().parse().expect("Invalid Group ID");
1515

16-
print!("Please input Tag ID: ");
17-
stdout().flush().unwrap();
18-
stdin().read_line(&mut tag_id_input).unwrap();
19-
let tag_id: i32 = tag_id_input.trim().parse().expect("Invalid Tag ID");
16+
print!("Please input Tag ID: ");
17+
stdout().flush().unwrap();
18+
stdin().read_line(&mut tag_id_input).unwrap();
19+
let tag_id: i32 = tag_id_input.trim().parse().expect("Invalid Tag ID");
2020

21-
let result = create_group_tag(connection, group_id, tag_id);
22-
if result.is_ok() {
23-
println!("\nTag ID {} added to Group ID {}", tag_id, group_id);
24-
} else {
25-
println!("\nFailed to add Tag ID {} to Group ID {}", tag_id, group_id);
26-
}
21+
let result = create_group_tag(connection, group_id, tag_id);
22+
if result.is_ok() {
23+
println!("\nTag ID {} added to Group ID {}", tag_id, group_id);
24+
} else {
25+
println!("\nFailed to add Tag ID {} to Group ID {}", tag_id, group_id);
26+
}
2727
}
2828

2929
#[allow(dead_code)]

‎file_classification_cli/src/bin/create_tag.rs‎

Lines changed: 14 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -3,23 +3,23 @@ use file_classification_core::establish_connection;
33
use std::io::stdin;
44

55
fn main() {
6-
let connection = &mut establish_connection();
6+
let connection = &mut establish_connection();
77

8-
let mut name = String::new();
8+
let mut name = String::new();
99

10-
println!("Please input Tag Name:");
11-
stdin().read_line(&mut name).unwrap();
12-
let name = name.trim_end();
10+
println!("Please input Tag Name:");
11+
stdin().read_line(&mut name).unwrap();
12+
let name = name.trim_end();
1313

14-
let tag = create_tag(connection, name);
15-
match tag {
16-
Ok(tag) => {
17-
println!("\nCreated tag {name} with id {}", tag.id);
18-
}
19-
Err(e) => {
20-
println!("\nError creating tag: {}", e);
21-
}
22-
}
14+
let tag = create_tag(connection, name);
15+
match tag {
16+
Ok(tag) => {
17+
println!("\nCreated tag {name} with id {}", tag.id);
18+
}
19+
Err(e) => {
20+
println!("\nError creating tag: {}", e);
21+
}
22+
}
2323
}
2424

2525
#[allow(dead_code)]

‎file_classification_cli/src/bin/delete_file.rs‎

Lines changed: 7 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -2,14 +2,12 @@ use file_classification_core::delete_file;
22
use file_classification_core::establish_connection;
33
use std::io;
44
fn main() {
5-
let connection = &mut establish_connection();
5+
let connection = &mut establish_connection();
66

7-
println!("Please input File ID:");
8-
let mut file_id = String::new();
9-
io::stdin()
10-
.read_line(&mut file_id)
11-
.expect("Failed to read line");
12-
let file_id: i32 = file_id.trim().parse().expect("Please type a number!");
13-
println!("Deleting file with ID: {}", file_id);
14-
delete_file(connection, file_id).expect("Error deleting file");
7+
println!("Please input File ID:");
8+
let mut file_id = String::new();
9+
io::stdin().read_line(&mut file_id).expect("Failed to read line");
10+
let file_id: i32 = file_id.trim().parse().expect("Please type a number!");
11+
println!("Deleting file with ID: {}", file_id);
12+
delete_file(connection, file_id).expect("Error deleting file");
1513
}

‎file_classification_cli/src/bin/delete_group.rs‎

Lines changed: 7 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -2,14 +2,12 @@ use file_classification_core::delete_group;
22
use file_classification_core::establish_connection;
33
use std::io;
44
fn main() {
5-
let connection = &mut establish_connection();
5+
let connection = &mut establish_connection();
66

7-
println!("Please input Group ID:");
8-
let mut group_id = String::new();
9-
io::stdin()
10-
.read_line(&mut group_id)
11-
.expect("Failed to read line");
12-
let group_id: i32 = group_id.trim().parse().expect("Please type a number!");
13-
println!("Deleting group with ID: {}", group_id);
14-
delete_group(connection, group_id).expect("Error deleting group");
7+
println!("Please input Group ID:");
8+
let mut group_id = String::new();
9+
io::stdin().read_line(&mut group_id).expect("Failed to read line");
10+
let group_id: i32 = group_id.trim().parse().expect("Please type a number!");
11+
println!("Deleting group with ID: {}", group_id);
12+
delete_group(connection, group_id).expect("Error deleting group");
1513
}

‎file_classification_cli/src/bin/delete_tag.rs‎

Lines changed: 7 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -2,14 +2,12 @@ use file_classification_core::delete_tag;
22
use file_classification_core::establish_connection;
33
use std::io;
44
fn main() {
5-
let connection = &mut establish_connection();
5+
let connection = &mut establish_connection();
66

7-
println!("Please input Tag ID:");
8-
let mut tag_id = String::new();
9-
io::stdin()
10-
.read_line(&mut tag_id)
11-
.expect("Failed to read line");
12-
let tag_id: i32 = tag_id.trim().parse().expect("Please type a number!");
13-
println!("Deleting tag with ID: {}", tag_id);
14-
delete_tag(connection, tag_id).expect("Error deleting tag");
7+
println!("Please input Tag ID:");
8+
let mut tag_id = String::new();
9+
io::stdin().read_line(&mut tag_id).expect("Failed to read line");
10+
let tag_id: i32 = tag_id.trim().parse().expect("Please type a number!");
11+
println!("Deleting tag with ID: {}", tag_id);
12+
delete_tag(connection, tag_id).expect("Error deleting tag");
1513
}

‎file_classification_cli/src/bin/list_files.rs‎

Lines changed: 24 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -1,34 +1,28 @@
11
use file_classification_core::establish_connection;
2-
use file_classification_core::{select_files, models::SearchFile}; // 引入select_files和SearchFile
2+
use file_classification_core::{models::SearchFile, select_files}; // 引入select_files和SearchFile
33

44
fn main() {
5-
let connection = &mut establish_connection();
6-
7-
// 定义一个空的 SearchFile 来进行无条件查询
8-
let search_input = SearchFile {
9-
id: None,
10-
type_: None,
11-
path: None,
12-
reference_count: None,
13-
group_id: None,
14-
};
15-
16-
// 使用 select_files 函数进行查询
17-
let results = select_files(connection, search_input, 5)
18-
.expect("Error loading files");
19-
20-
println!("Displaying {} files", results.len());
21-
for file in results {
22-
let mut output = format!("ID: {}", file.id);
23-
24-
output.push_str(&format!(", Type: '{}'", file.type_));
25-
26-
output.push_str(&format!(", Path: '{}'", file.path));
27-
28-
output.push_str(&format!(", ReferenceCount: {}", file.reference_count));
29-
30-
output.push_str(&format!(", GroupID: {}", file.group_id));
31-
32-
println!("{}", output);
33-
}
5+
let connection = &mut establish_connection();
6+
7+
// 定义一个空的 SearchFile 来进行无条件查询
8+
let search_input =
9+
SearchFile { id: None, type_: None, path: None, reference_count: None, group_id: None };
10+
11+
// 使用 select_files 函数进行查询
12+
let results = select_files(connection, search_input, 5).expect("Error loading files");
13+
14+
println!("Displaying {} files", results.len());
15+
for file in results {
16+
let mut output = format!("ID: {}", file.id);
17+
18+
output.push_str(&format!(", Type: '{}'", file.type_));
19+
20+
output.push_str(&format!(", Path: '{}'", file.path));
21+
22+
output.push_str(&format!(", ReferenceCount: {}", file.reference_count));
23+
24+
output.push_str(&format!(", GroupID: {}", file.group_id));
25+
26+
println!("{}", output);
27+
}
3428
}

0 commit comments

Comments
 (0)