martin/PbDbFixer
2
1
Fork 0
Code Issues Pull requests Projects Releases 15 Wiki Activity

set foreign_key pragma correctly

Browse source
This commit is contained in:
Martin Brodbeck 2021-02-24 14:03:35 +01:00
parent da405064d4
commit 94e1dedb1e
1 changed files with 3 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
src/database.rs
View file

@ -43,8 +43,8 @@ fn get_epubs_from_database(tx: &Transaction) -> Vec<BookEntry> {
let prefix: String = row.get(1).unwrap(); let prefix: String = row.get(1).unwrap();
let filename: String = row.get(2).unwrap(); let filename: String = row.get(2).unwrap();
let filepath = format!("{}/{}", prefix, filename); let filepath = format!("{}/{}", prefix, filename);
let firstauthor: String = row.get(3).unwrap(); let firstauthor: String = row.get(3).unwrap_or_default();
let author: String = row.get(4).unwrap(); let author: String = row.get(4).unwrap_or_default();
let has_drm = match prefix.as_str() { let has_drm = match prefix.as_str() {
"/mnt/ext1/Digital Editions" => true, "/mnt/ext1/Digital Editions" => true,
_ => false, _ => false,
@ -71,8 +71,6 @@ fn get_epubs_from_database(tx: &Transaction) -> Vec<BookEntry> {
} }
fn remove_ghost_books_from_db(tx: &Transaction) -> usize { fn remove_ghost_books_from_db(tx: &Transaction) -> usize {
tx.execute("PRAGMA foreign_keys = 0", NO_PARAMS).unwrap();
let mut stmt = tx let mut stmt = tx
.prepare( .prepare(
r#" r#"
@ -115,8 +113,6 @@ fn remove_ghost_books_from_db(tx: &Transaction) -> usize {
) )
.unwrap(); .unwrap();
tx.execute("PRAGMA foreign_keys = 0", NO_PARAMS).unwrap();
num num
} }
@ -150,6 +146,7 @@ pub fn fix_db_entries() -> Statistics {
}; };
let mut conn = Connection::open(DATABASE_FILE).unwrap(); let mut conn = Connection::open(DATABASE_FILE).unwrap();
conn.pragma_update(None, "foreign_keys", &0).unwrap();
let tx = conn.transaction().unwrap(); let tx = conn.transaction().unwrap();
let book_entries = get_epubs_from_database(&tx); let book_entries = get_epubs_from_database(&tx);