rustfm-scraper 0.2.23

Scrapes listening history from Last.fm and stores it in a file
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49

use anyhow::{Context, Result};

use crate::files;
use crate::models::recent_tracks::Track;
use crate::models::saved_scrobbles::SavedScrobbles;

pub fn save_to_csv(scrobbles: &[Track], username: &str) -> Result<i32> {
    let file = files::build_file_path(username, "csv")?;

    let scrobbles = SavedScrobbles::from_scrobbles(scrobbles);

    let mut wtr = csv::Writer::from_path(file).unwrap();
    scrobbles.to_csv_writer(&mut wtr);
    wtr.flush().unwrap();

    Ok(scrobbles.total_saved_scrobbles())
}

pub fn append_to_csv(
    scrobbles: &[Track],
    saved_scrobbles: &mut SavedScrobbles,
    username: &str,
) -> Result<i32> {
    let file = files::build_file_path(username, "csv")?;

    saved_scrobbles.append_new_scrobbles(scrobbles);

    let mut wtr = csv::Writer::from_path(file).context("Error creating csv writer")?;
    saved_scrobbles.to_csv_writer(&mut wtr);
    wtr.flush().expect("Error flushing csv writer");

    Ok(saved_scrobbles.total_saved_scrobbles())
}

pub fn load_from_csv(username: &str) -> Result<SavedScrobbles> {
    println!("Loading saved scrobbles from `{}.csv`...", username);

    let file = files::build_file_path(username, "csv")?;

    let mut rdr = csv::Reader::from_path(file).context("Error creating csv reader")?;
    let saved_scrobbles = SavedScrobbles::from_csv_reader(&mut rdr);

    println!(
        "{} saved scrobbles retrieved from file\n",
        &saved_scrobbles.total_saved_scrobbles_formatted()
    );

    Ok(saved_scrobbles)
}