martin
/
PbDbFixer
1
1
Fork 0
Code Issues Pull Requests Projects Releases 14 Wiki Activity

Skipping Adobe drm encoded files since we can't read metadata. 

fixes #1
This commit is contained in:
Martin Brodbeck 2021-02-02 19:44:31 +01:00
parent 637d0ffc74
commit 226ce76c16
1 changed files with 21 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
src/main.rs
View File

@ -304,6 +304,7 @@ struct BookEntry {
filepath: String, filepath: String,
author: String, author: String,
firstauthor: String, firstauthor: String,
has_drm: bool,
} }
fn get_epubs_from_database(tx: &Transaction) -> Vec<BookEntry> { fn get_epubs_from_database(tx: &Transaction) -> Vec<BookEntry> {
@ -331,12 +332,17 @@ fn get_epubs_from_database(tx: &Transaction) -> Vec<BookEntry> {
let filepath = format!("{}/{}", prefix, filename); let filepath = format!("{}/{}", prefix, filename);
let firstauthor: String = row.get(3).unwrap(); let firstauthor: String = row.get(3).unwrap();
let author: String = row.get(4).unwrap(); let author: String = row.get(4).unwrap();
let has_drm = match prefix.as_str() {
"/mnt/ext1/Digital Editions" => true,
_ => false,
};
let entry = BookEntry { let entry = BookEntry {
id: book_id, id: book_id,
filepath, filepath,
firstauthor, firstauthor,
author, author,
has_drm,
}; };
book_entries.push(entry); book_entries.push(entry);
@ -394,15 +400,22 @@ fn remove_ghost_books_from_db(tx: &Transaction) -> usize {
struct Statistics { struct Statistics {
authors_fixed: i32, authors_fixed: i32,
ghost_books_cleaned: usize, ghost_books_cleaned: usize,
drm_skipped: usize,
} }
fn fix_db_entries(tx: &Transaction, book_entries: &Vec<BookEntry>) -> Statistics { fn fix_db_entries(tx: &Transaction, book_entries: &Vec<BookEntry>) -> Statistics {
let mut stat = Statistics { let mut stat = Statistics {
authors_fixed: 0, authors_fixed: 0,
ghost_books_cleaned: 0, ghost_books_cleaned: 0,
drm_skipped: 0,
}; };
for entry in book_entries { for entry in book_entries {
if entry.has_drm {
stat.drm_skipped = stat.drm_skipped + 1;
continue;
}
let file = File::open(entry.filepath.as_str()); let file = File::open(entry.filepath.as_str());
let file = match file { let file = match file {
Err(_) => continue, Err(_) => continue,
@ -489,16 +502,20 @@ fn main() {
pocketbook::dialog( pocketbook::dialog(
pocketbook::Icon::Info, pocketbook::Icon::Info,
&format!( &format!(
"Authors fixed: {}\nBooks cleaned from DB: {}", "Authors fixed: {}\n\
&stat.authors_fixed, &stat.ghost_books_cleaned Books skipped (DRM): {}\n\
Books cleaned from DB: {}",
&stat.authors_fixed, &stat.drm_skipped, &stat.ghost_books_cleaned
), ),
&["OK"], &["OK"],
); );
} }
} else { } else {
println!( println!(
"Authors fixed: {}\nBooks cleaned from DB: {}", "Authors fixed: {}\n\
&stat.authors_fixed, &stat.ghost_books_cleaned Books skipped (DRM): {}\n\
Books cleaned from DB: {}",
&stat.authors_fixed, &stat.drm_skipped, &stat.ghost_books_cleaned
); );
} }
} }