Add search in a website

This commit is contained in:
Rémi BERTHO 2020-07-21 20:37:26 +02:00
parent cf6106155a
commit 5bb5ed9fdb
Signed by: dalan
GPG Key ID: EE3B917931C07B64
4 changed files with 52 additions and 9 deletions

View File

@ -3,7 +3,7 @@
* [x] Récupération prix darty avec [scraper](https://crates.io/crates/scraper) et [reqwest](https://crates.io/crates/reqwest)
* [x] Ajout du support de la Fnac, …
* [x] Récupération URL ligne de commande avec [clap](https://crates.io/crates/clap)
* [ ] Ajout de SearchParser pour rechercher un article sur tous les parseurs - recherche sur un seul parser
* [ ] Ajout de SearchParser pour rechercher un article sur tous les parseurs
* [x] Commande de liste des différents parseurs
* [ ] Ajout des pays avec [celes](https://crates.io/crates/celes) : recherche uniquement sur les parser du pays et parseur multi pays (amazon par exemple)
* [ ] Lecture des URLs depuis un fichier avec [toml](https://crates.io/crates/toml)

View File

@ -19,40 +19,58 @@ fn main() -> Result<()> {
.author("Rémi BERTHO <remi.bertho@dalan.fr>")
.about("Check price")
.subcommand(
SubCommand::with_name("check")
.about("Check from an URL")
SubCommand::with_name("get")
.about("Get a price from an URL")
.arg(Arg::with_name("URL").required(true).multiple(true).help("The URL to get price")),
)
.subcommand(
SubCommand::with_name("search")
.about("Search an object")
.arg(Arg::with_name("parser")
.short("w")
.long("website")
.value_name("WEBSITE")
.help("Website name")
.takes_value(true))
.arg(Arg::with_name("name").required(true).multiple(true).help("The name of the object")),
)
.subcommand(SubCommand::with_name("list").about("List the parsers"))
.get_matches();
let price_checker = PriceChecker::new().unwrap();
match matches.subcommand() {
("check", Some(check_matches)) => {
("get", Some(check_matches)) => {
for url_str in check_matches.values_of("URL").unwrap() {
let url = Url::parse(url_str)?;
println!("{}", price_checker.get_price(&url)?);
}
}
("search", Some(check_matches)) => {
if let Some(parser) = check_matches.value_of("parser")
{
for name in check_matches.values_of("name").unwrap() {
let res = price_checker.search(name)?;
if let Some(url) = price_checker.search(parser, name)? {
let price = price_checker.get_price(&url)?;
println!(" - {}", name);
println!(" * {}", url);
println!(" * {}", price);
}
}
} else {
for name in check_matches.values_of("name").unwrap() {
let res = price_checker.search_all(name)?;
if res.is_empty() {
println!("«{}» not found", name);
} else {
for (parser_name, url) in res {
let price = price_checker.get_price(&url)?;
println!(" - {}", parser_name);
println!(" - {} : {}", name, parser_name);
println!(" * {}", url);
println!(" * {}", price);
}
}
}
}
}
("list", _) => {
println!("Price parsers:");
for parser in price_checker.parser_list.get_price().iter() {

View File

@ -5,7 +5,7 @@ pub mod fnac;
pub mod ldlc;
use crate::price_result::PriceResult;
use anyhow::Result;
use anyhow::{anyhow, Result};
use arraygen::Arraygen;
use scraper::Html;
use url::Url;
@ -27,7 +27,7 @@ pub trait PriceParser: Parser {
fn parse_price(&self, html: &Html) -> Result<PriceResult>;
}
pub trait SearchParser: Parser {
pub trait SearchParser: PriceParser {
/// Return the search URL
fn search_url(&self, name: &str) -> Url;
/// Return the first occurence of result of the page if any
@ -39,11 +39,13 @@ macro_rules! gen_list {
#[derive(Arraygen, Debug)]
#[gen_array(pub fn get_price: & dyn PriceParser)]
#[gen_array(pub fn get_search: & dyn SearchParser)]
#[gen_array(pub fn get_all: & dyn Parser)]
pub struct List {
$(
$(
#[in_array($array)]
)*
#[in_array(get_all)]
$module: $module::$name
),*
}
@ -71,6 +73,16 @@ gen_list!(
[amazon::Amazon: get_price, get_search]
);
impl List {
pub fn get_parser(&self, name: &str) -> Result<&dyn SearchParser> {
Ok(*self
.get_search()
.iter()
.find(|&&parser| parser.name().to_lowercase() == name.to_lowercase())
.ok_or(anyhow!("Cannot find the parser {}", name))?)
}
}
#[test]
fn test_parser_list() {
let parser_list = List::new().unwrap();

View File

@ -45,7 +45,7 @@ impl PriceChecker {
}
/// Search an object in all parsers
pub fn search(&self, name: &str) -> Result<Vec<(&'static str, Url)>> {
pub fn search_all(&self, name: &str) -> Result<Vec<(&'static str, Url)>> {
let mut urls = Vec::new();
for parser in self.parser_list.get_search().iter() {
let search_url = parser.search_url(name);
@ -56,6 +56,19 @@ impl PriceChecker {
}
Ok(urls)
}
/// Search an object in a parser
pub fn search(&self, parser_name: &str, name: &str) -> Result<Option<Url>> {
let parser = self.parser_list.get_parser(parser_name)?;
let search_url = parser.search_url(name);
let document = self.get_html(&search_url)?;
if let Some(url) = parser.search(&document)? {
Ok(Some(url))
}
else {
Ok(None)
}
}
}
#[test]