Compare commits
2 commits
9baff090a8
...
65007fdf50
Author | SHA1 | Date | |
---|---|---|---|
65007fdf50 | |||
a6594d4abc |
5 changed files with 15 additions and 40 deletions
6
Cargo.lock
generated
6
Cargo.lock
generated
|
@ -47,9 +47,9 @@ dependencies = [
|
|||
|
||||
[[package]]
|
||||
name = "cc"
|
||||
version = "1.0.66"
|
||||
version = "1.0.67"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "4c0496836a84f8d0495758516b8621a622beb77c0fed418570e50764093ced48"
|
||||
checksum = "e3c69b077ad434294d3ce9f1f6143a2a4b89a8a2d54ef813d85003a4fd1137fd"
|
||||
|
||||
[[package]]
|
||||
name = "cfg-if"
|
||||
|
@ -148,7 +148,7 @@ dependencies = [
|
|||
|
||||
[[package]]
|
||||
name = "pbdbfixer"
|
||||
version = "0.7.2"
|
||||
version = "0.8.0"
|
||||
dependencies = [
|
||||
"quick-xml",
|
||||
"rusqlite",
|
||||
|
|
|
@ -1,6 +1,6 @@
|
|||
[package]
|
||||
name = "pbdbfixer"
|
||||
version = "0.7.2"
|
||||
version = "0.8.0"
|
||||
authors = ["Martin Brodbeck <martin@brodbeck-online.de>"]
|
||||
edition = "2018"
|
||||
|
||||
|
|
|
@ -9,7 +9,6 @@ pub struct BookEntry {
|
|||
filepath: String,
|
||||
author: String,
|
||||
firstauthor: String,
|
||||
has_drm: bool,
|
||||
genre: String,
|
||||
first_author_letter: String,
|
||||
series: String,
|
||||
|
@ -45,10 +44,6 @@ fn get_epubs_from_database(tx: &Transaction) -> Vec<BookEntry> {
|
|||
let filepath = format!("{}/{}", prefix, filename);
|
||||
let firstauthor: String = row.get(3).unwrap_or_default();
|
||||
let author: String = row.get(4).unwrap_or_default();
|
||||
let has_drm = match prefix.as_str() {
|
||||
"/mnt/ext1/Digital Editions" => true,
|
||||
_ => false,
|
||||
};
|
||||
let genre: String = row.get(5).unwrap_or_default();
|
||||
let first_author_letter = row.get(6).unwrap_or_default();
|
||||
let series: String = row.get(7).unwrap_or_default();
|
||||
|
@ -58,7 +53,6 @@ fn get_epubs_from_database(tx: &Transaction) -> Vec<BookEntry> {
|
|||
filepath,
|
||||
firstauthor,
|
||||
author,
|
||||
has_drm,
|
||||
genre,
|
||||
first_author_letter,
|
||||
series,
|
||||
|
@ -119,7 +113,6 @@ fn remove_ghost_books_from_db(tx: &Transaction) -> usize {
|
|||
pub struct Statistics {
|
||||
pub authors_fixed: i32,
|
||||
pub ghost_books_cleaned: usize,
|
||||
pub drm_skipped: usize,
|
||||
pub genres_fixed: usize,
|
||||
pub sorting_fixed: usize,
|
||||
pub series_fixed: usize,
|
||||
|
@ -139,7 +132,6 @@ pub fn fix_db_entries() -> Statistics {
|
|||
let mut stat = Statistics {
|
||||
authors_fixed: 0,
|
||||
ghost_books_cleaned: 0,
|
||||
drm_skipped: 0,
|
||||
genres_fixed: 0,
|
||||
sorting_fixed: 0,
|
||||
series_fixed: 0,
|
||||
|
@ -152,11 +144,6 @@ pub fn fix_db_entries() -> Statistics {
|
|||
let book_entries = get_epubs_from_database(&tx);
|
||||
|
||||
for entry in book_entries {
|
||||
if entry.has_drm {
|
||||
stat.drm_skipped = stat.drm_skipped + 1;
|
||||
continue;
|
||||
}
|
||||
|
||||
if let Some(epub_metadata) = epub::get_epub_metadata(&entry.filepath) {
|
||||
// Fix firstauthor…
|
||||
let mut firstauthors = epub_metadata
|
||||
|
@ -185,7 +172,8 @@ pub fn fix_db_entries() -> Statistics {
|
|||
.unwrap_or_default()
|
||||
.to_string()
|
||||
.to_uppercase();
|
||||
if entry.first_author_letter != first_author_letter {
|
||||
|
||||
if first_author_letter != "\0" && (entry.first_author_letter != first_author_letter) {
|
||||
let mut stmt = tx
|
||||
.prepare("UPDATE books_impl SET first_author_letter = :first_letter WHERE id = :book_id")
|
||||
.unwrap();
|
||||
|
|
|
@ -151,6 +151,10 @@ pub fn get_epub_metadata(filename: &str) -> Option<EpubMetadata> {
|
|||
.unwrap()
|
||||
.as_str();
|
||||
xml_authors.insert(curr_id.clone(), XmlAut::new());
|
||||
} else {
|
||||
curr_id = "none".to_string() + xml_authors.len().to_string().as_str();
|
||||
let entry = xml_authors.entry(curr_id.clone()).or_insert(XmlAut::new());
|
||||
entry.role = "aut".to_string();
|
||||
}
|
||||
} else {
|
||||
if let Some(file_as_val) = e
|
||||
|
|
27
src/main.rs
27
src/main.rs
|
@ -25,25 +25,12 @@ fn main() {
|
|||
|
||||
if cfg!(target_arch = "arm") {
|
||||
if stat.anything_fixed() == false {
|
||||
if stat.drm_skipped == 0 {
|
||||
pocketbook::dialog(
|
||||
pocketbook::Icon::Info,
|
||||
"The database seems to be ok.\n\
|
||||
pocketbook::dialog(
|
||||
pocketbook::Icon::Info,
|
||||
"The database seems to be ok.\n\
|
||||
Nothing had to be fixed.",
|
||||
&["OK"],
|
||||
);
|
||||
} else {
|
||||
pocketbook::dialog(
|
||||
pocketbook::Icon::Info,
|
||||
&format!(
|
||||
"The database seems to be ok.\n\
|
||||
Nothing had to be fixed.\n\
|
||||
(Books skipped (DRM): {})",
|
||||
&stat.drm_skipped
|
||||
),
|
||||
&["OK"],
|
||||
);
|
||||
}
|
||||
&["OK"],
|
||||
);
|
||||
} else {
|
||||
pocketbook::dialog(
|
||||
pocketbook::Icon::Info,
|
||||
|
@ -52,13 +39,11 @@ fn main() {
|
|||
Sorting fixed: {}\n\
|
||||
Genres fixed: {}\n\
|
||||
Series fixed: {}\n\
|
||||
Books skipped (DRM): {}\n\
|
||||
Books cleaned from DB: {}",
|
||||
&stat.authors_fixed,
|
||||
&stat.sorting_fixed,
|
||||
&stat.genres_fixed,
|
||||
&stat.series_fixed,
|
||||
&stat.drm_skipped,
|
||||
&stat.ghost_books_cleaned
|
||||
),
|
||||
&["OK"],
|
||||
|
@ -70,13 +55,11 @@ fn main() {
|
|||
Sorting fixed: {}\n\
|
||||
Genres fixed: {}\n\
|
||||
Series fixed: {}\n\
|
||||
Books skipped (DRM): {}\n\
|
||||
Books cleaned from DB: {}",
|
||||
&stat.authors_fixed,
|
||||
&stat.sorting_fixed,
|
||||
&stat.genres_fixed,
|
||||
&stat.series_fixed,
|
||||
&stat.drm_skipped,
|
||||
&stat.ghost_books_cleaned
|
||||
);
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue