From 06d99b5871493c1d7c5606a1d5c2c2fba654367d Mon Sep 17 00:00:00 2001 From: Damien Elmes Date: Sat, 23 Sep 2023 16:01:46 +1000 Subject: [PATCH] Allow extracting desired retention --- rslib/src/storage/sqlite.rs | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/rslib/src/storage/sqlite.rs b/rslib/src/storage/sqlite.rs index 7da32d897..525721814 100644 --- a/rslib/src/storage/sqlite.rs +++ b/rslib/src/storage/sqlite.rs @@ -235,7 +235,7 @@ fn add_extract_custom_data_function(db: &Connection) -> rusqlite::Result<()> { ) } -/// eg. extract_fsrs_variable(card.data, 's' | 'd') -> float | null +/// eg. extract_fsrs_variable(card.data, 's' | 'd' | 'dr') -> float | null fn add_extract_fsrs_variable(db: &Connection) -> rusqlite::Result<()> { db.create_scalar_function( "extract_fsrs_variable", @@ -257,6 +257,7 @@ fn add_extract_fsrs_variable(db: &Connection) -> rusqlite::Result<()> { Ok(match key { "s" => card_data.fsrs_stability, "d" => card_data.fsrs_difficulty, + "dr" => card_data.fsrs_desired_retention, _ => panic!("invalid key: {key}"), }) },