mirror of
https://gitlab.com/news-flash/article_scraper.git
synced 2025-07-08 08:30:00 +02:00
Compare commits
13 commits
article_sc
...
master
Author | SHA1 | Date | |
---|---|---|---|
|
4c9709e292 | ||
|
a23a691c31 | ||
|
9f349f8c6f | ||
|
498008f630 | ||
|
ee53f58aeb | ||
|
06990acbc0 | ||
|
f361392c04 | ||
|
9b374a28c7 | ||
|
b92500fca2 | ||
|
0978335d3b | ||
|
9f56ed03b8 | ||
|
8cfcd6d9f3 | ||
|
ca1cc47af1 |
11 changed files with 159 additions and 102 deletions
|
@ -1,15 +1,16 @@
|
|||
stages:
|
||||
- build
|
||||
|
||||
|
||||
run-build:
|
||||
stage: build
|
||||
image: rust:1.79
|
||||
image: rust:1.86
|
||||
before_script:
|
||||
- rustup component add rustfmt
|
||||
- rustup component add clippy
|
||||
- export LIBXML2=$(pkg-config libxml-2.0 --variable=libdir)/libxml2.so
|
||||
script:
|
||||
- rustc --version && cargo --version
|
||||
- echo $LIBXML2
|
||||
- cargo fmt -- --check
|
||||
- cargo clippy --all-targets --all-features -- -D warnings
|
||||
- cargo build --release
|
||||
|
|
|
@ -3,7 +3,7 @@ members = ["article_scraper", "article_scraper_cli"]
|
|||
resolver = "2"
|
||||
|
||||
[workspace.package]
|
||||
version = "2.1.1"
|
||||
version = "2.1.3"
|
||||
authors = ["Jan Lukas Gernert <jangernert@gmail.com>"]
|
||||
edition = "2021"
|
||||
license = "GPL-3.0-or-later"
|
||||
|
|
|
@ -14,17 +14,17 @@ exclude = ["resources/tests"]
|
|||
[dependencies]
|
||||
thiserror = "2.0"
|
||||
libxml = "0.3"
|
||||
reqwest = { version = "0.12", features = ["json", "native-tls", "gzip", "brotli", "stream"] }
|
||||
reqwest = { version = "0.12", features = ["stream"] }
|
||||
tokio = { version = "1", features = ["macros", "fs", "io-util"] }
|
||||
url = "2.5"
|
||||
regex = "1.10"
|
||||
regex = "1.11"
|
||||
encoding_rs = "0.8"
|
||||
chrono = "0.4"
|
||||
base64 = "0.22"
|
||||
image = "0.25"
|
||||
log = "0.4"
|
||||
rust-embed="8.4"
|
||||
once_cell = "1.19"
|
||||
rust-embed = "8.7"
|
||||
once_cell = "1.21"
|
||||
escaper = "0.1"
|
||||
futures = "0.3"
|
||||
unic-emoji-char = "0.9"
|
||||
|
|
|
@ -1 +1 @@
|
|||
Subproject commit ccde390b11893cbafdc84f74b449ddc3cc05c024
|
||||
Subproject commit 69aa220193d99427d3822fabccdfaeede56cd532
|
File diff suppressed because one or more lines are too long
|
@ -6,10 +6,10 @@ use thiserror::Error;
|
|||
|
||||
#[derive(Error, Debug)]
|
||||
pub enum ScraperError {
|
||||
#[error("")]
|
||||
#[error("Configerror {0}")]
|
||||
Config(#[from] ConfigError),
|
||||
#[error("")]
|
||||
#[error("ImageDownloadError {0}")]
|
||||
Image(#[from] ImageDownloadError),
|
||||
#[error("")]
|
||||
#[error("FullTextParserError {0}")]
|
||||
Scrap(#[from] FullTextParserError),
|
||||
}
|
||||
|
|
|
@ -158,7 +158,7 @@ fn extract_date(
|
|||
fn get_meta(context: &Context, name: &str) -> Option<String> {
|
||||
Util::get_attribute(
|
||||
context,
|
||||
&format!("//meta[contains(@name, '{}')]", name),
|
||||
&format!("//meta[contains(@name, '{name}')]"),
|
||||
"content",
|
||||
)
|
||||
.ok()
|
||||
|
|
|
@ -69,6 +69,11 @@ impl FullTextParser {
|
|||
|
||||
let html = Self::get_body(response).await?;
|
||||
|
||||
if html.is_empty() {
|
||||
log::error!("Empty response body");
|
||||
return Err(FullTextParserError::Http);
|
||||
}
|
||||
|
||||
// check for fingerprints
|
||||
let config = if config.is_none() {
|
||||
if let Some(url) = Fingerprints::detect(&html) {
|
||||
|
@ -264,10 +269,17 @@ impl FullTextParser {
|
|||
}
|
||||
|
||||
// parse html
|
||||
Self::parse_html_string_patched(html.as_str()).map_err(|err| {
|
||||
let document = Self::parse_html_string_patched(html.as_str()).map_err(|err| {
|
||||
log::error!("Parsing HTML failed for downloaded HTML {:?}", err);
|
||||
FullTextParserError::Xml
|
||||
})
|
||||
})?;
|
||||
|
||||
if document.get_root_element().is_none() {
|
||||
log::error!("document without root");
|
||||
Err(FullTextParserError::Xml)
|
||||
} else {
|
||||
Ok(document)
|
||||
}
|
||||
}
|
||||
|
||||
/// FIXME: Here are some patched functions of libxml crate.
|
||||
|
@ -354,19 +366,32 @@ impl FullTextParser {
|
|||
.send()
|
||||
.await
|
||||
.map_err(|err| {
|
||||
log::error!(
|
||||
"Downloading HTML failed: GET '{}' - '{}'",
|
||||
url.as_str(),
|
||||
err
|
||||
);
|
||||
log::error!("Downloading HTML failed: GET '{url}' - '{err}'");
|
||||
FullTextParserError::Http
|
||||
})?;
|
||||
Ok(response)
|
||||
}
|
||||
|
||||
async fn get_body(response: Response) -> Result<String, FullTextParserError> {
|
||||
if response.status().is_success() {
|
||||
let status = response.status();
|
||||
if !status.is_success() {
|
||||
log::error!("status code: {status}");
|
||||
return Err(FullTextParserError::Http);
|
||||
}
|
||||
|
||||
let headers = response.headers().clone();
|
||||
|
||||
if headers
|
||||
.get(reqwest::header::CONTENT_LENGTH)
|
||||
.and_then(|hv| hv.to_str().ok())
|
||||
.and_then(|str| str.parse::<i64>().ok())
|
||||
.map(|content_length| content_length == 0)
|
||||
.unwrap_or(false)
|
||||
{
|
||||
log::error!("Empty response body");
|
||||
return Err(FullTextParserError::Http);
|
||||
}
|
||||
|
||||
let bytes = response
|
||||
.bytes()
|
||||
.await
|
||||
|
@ -375,14 +400,14 @@ impl FullTextParser {
|
|||
match from_utf8(&bytes) {
|
||||
Ok(utf8_str) => {
|
||||
log::debug!("Valid utf-8 string");
|
||||
return Ok(utf8_str.into());
|
||||
Ok(utf8_str.into())
|
||||
}
|
||||
Err(error) => {
|
||||
log::debug!("Invalid utf-8 string");
|
||||
let lossy_string = std::string::String::from_utf8_lossy(&bytes);
|
||||
|
||||
if let Some(encoding) = Self::get_encoding_from_html(&lossy_string) {
|
||||
log::debug!("Encoding extracted from HTML: '{}'", encoding);
|
||||
log::debug!("Encoding extracted from HTML: '{encoding}'");
|
||||
if let Some(decoded_html) = Self::decode_html(&bytes, encoding) {
|
||||
let decoded_html = decoded_html.replacen(
|
||||
&format!("charset=\"{encoding}\""),
|
||||
|
@ -394,7 +419,7 @@ impl FullTextParser {
|
|||
}
|
||||
|
||||
if let Some(encoding) = Self::get_encoding_from_http_header(&headers) {
|
||||
log::debug!("Encoding extracted from headers: '{}'", encoding);
|
||||
log::debug!("Encoding extracted from headers: '{encoding}'");
|
||||
if let Some(decoded_html) = Self::decode_html(&bytes, encoding) {
|
||||
let decoded_html = decoded_html.replacen(
|
||||
&format!("charset=\"{encoding}\""),
|
||||
|
@ -405,14 +430,11 @@ impl FullTextParser {
|
|||
}
|
||||
}
|
||||
|
||||
return Err(FullTextParserError::Utf8(error));
|
||||
Err(FullTextParserError::Utf8(error))
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Err(FullTextParserError::Http)
|
||||
}
|
||||
|
||||
pub async fn download(
|
||||
url: &url::Url,
|
||||
client: &Client,
|
||||
|
@ -422,8 +444,13 @@ impl FullTextParser {
|
|||
let headers = Util::generate_headers(config, global_config)?;
|
||||
let response = Self::get_response(url, client, headers).await?;
|
||||
let body = Self::get_body(response).await?;
|
||||
if body.is_empty() {
|
||||
log::error!("Empty response body");
|
||||
Err(FullTextParserError::Http)
|
||||
} else {
|
||||
Ok(body)
|
||||
}
|
||||
}
|
||||
|
||||
fn get_encoding_from_http_header(headers: &reqwest::header::HeaderMap) -> Option<&str> {
|
||||
headers
|
||||
|
@ -695,7 +722,7 @@ impl FullTextParser {
|
|||
}
|
||||
|
||||
fn fix_iframe_size(context: &Context, site_name: &str) -> Result<(), FullTextParserError> {
|
||||
let xpath = &format!("//iframe[contains(@src, '{}')]", site_name);
|
||||
let xpath = &format!("//iframe[contains(@src, '{site_name}')]");
|
||||
let node_vec = Util::evaluate_xpath(context, xpath, false)?;
|
||||
for mut node in node_vec {
|
||||
if node.is_null() {
|
||||
|
@ -734,7 +761,7 @@ impl FullTextParser {
|
|||
) -> Result<(), FullTextParserError> {
|
||||
let xpath_tag = tag.unwrap_or("*");
|
||||
|
||||
let xpath = &format!("//{}[@{}]", xpath_tag, attribute);
|
||||
let xpath = &format!("//{xpath_tag}[@{attribute}]");
|
||||
let node_vec = Util::evaluate_xpath(context, xpath, false)?;
|
||||
for mut node in node_vec {
|
||||
if let Err(err) = node.remove_property(attribute) {
|
||||
|
@ -764,17 +791,16 @@ impl FullTextParser {
|
|||
if let Some(url) = node.get_attribute(attribute) {
|
||||
let trimmed_url = url.trim();
|
||||
|
||||
let is_hash_url = url.starts_with('#');
|
||||
if url.starts_with('#') || url.starts_with("\\#") {
|
||||
continue;
|
||||
}
|
||||
|
||||
let is_relative_url = url::Url::parse(&url)
|
||||
.err()
|
||||
.map(|err| err == url::ParseError::RelativeUrlWithoutBase)
|
||||
.unwrap_or(false);
|
||||
let is_javascript = trimmed_url.contains("javascript:");
|
||||
|
||||
if !is_hash_url && node.get_name().to_uppercase() == "A" {
|
||||
_ = node.set_attribute("target", "_blank");
|
||||
}
|
||||
|
||||
if let Some(srcset) = node.get_attribute("srcset") {
|
||||
let res = constants::SRC_SET_URL
|
||||
.captures_iter(&srcset)
|
||||
|
@ -805,9 +831,7 @@ impl FullTextParser {
|
|||
_ = node.set_attribute("srcset", res.as_str());
|
||||
}
|
||||
|
||||
if is_hash_url {
|
||||
_ = node.set_attribute(attribute, trimmed_url);
|
||||
} else if is_relative_url {
|
||||
if is_relative_url {
|
||||
let completed_url = match article_url.join(trimmed_url) {
|
||||
Ok(joined_url) => joined_url,
|
||||
Err(_) => continue,
|
||||
|
@ -920,7 +944,7 @@ impl FullTextParser {
|
|||
for xpath_strip_img_src in &config.strip_image_src {
|
||||
_ = Util::strip_node(
|
||||
context,
|
||||
&format!("//img[contains(@src,'{}')]", xpath_strip_img_src),
|
||||
&format!("//img[contains(@src,'{xpath_strip_img_src}')]"),
|
||||
);
|
||||
}
|
||||
}
|
||||
|
@ -928,7 +952,7 @@ impl FullTextParser {
|
|||
for xpath_strip_img_src in &global_config.strip_image_src {
|
||||
_ = Util::strip_node(
|
||||
context,
|
||||
&format!("//img[contains(@src,'{}')]", xpath_strip_img_src),
|
||||
&format!("//img[contains(@src,'{xpath_strip_img_src}')]"),
|
||||
);
|
||||
}
|
||||
|
||||
|
@ -1243,7 +1267,7 @@ impl FullTextParser {
|
|||
Ok(())
|
||||
}
|
||||
|
||||
fn remove_single_cell_tables(root: &mut Node) {
|
||||
pub(crate) fn remove_single_cell_tables(root: &mut Node) {
|
||||
let mut node_iter = Some(root.clone());
|
||||
|
||||
while let Some(node) = node_iter {
|
||||
|
@ -1281,7 +1305,7 @@ impl FullTextParser {
|
|||
}
|
||||
}
|
||||
|
||||
fn remove_extra_p_and_div(root: &mut Node) {
|
||||
pub(crate) fn remove_extra_p_and_div(root: &mut Node) {
|
||||
let mut node_iter = Some(root.clone());
|
||||
|
||||
while let Some(mut node) = node_iter {
|
||||
|
@ -1303,7 +1327,7 @@ impl FullTextParser {
|
|||
}
|
||||
}
|
||||
|
||||
fn remove_share_elements(root: &mut Node) {
|
||||
pub(crate) fn remove_share_elements(root: &mut Node) {
|
||||
let mut node_iter = Some(root.clone());
|
||||
|
||||
while let Some(mut node) = node_iter {
|
||||
|
@ -1323,7 +1347,7 @@ impl FullTextParser {
|
|||
}
|
||||
}
|
||||
|
||||
fn clean_attributes(root: &mut Node) -> Result<(), FullTextParserError> {
|
||||
pub(crate) fn clean_attributes(root: &mut Node) -> Result<(), FullTextParserError> {
|
||||
let mut node_iter = Some(root.clone());
|
||||
|
||||
while let Some(mut node) = node_iter {
|
||||
|
@ -1412,7 +1436,7 @@ impl FullTextParser {
|
|||
Ok(())
|
||||
}
|
||||
|
||||
fn remove_empty_nodes(root: &mut Node) {
|
||||
pub(crate) fn remove_empty_nodes(root: &mut Node) {
|
||||
let mut node_iter = Some(root.clone());
|
||||
|
||||
while let Some(mut node) = node_iter {
|
||||
|
|
|
@ -219,7 +219,9 @@ impl ImageDownloader {
|
|||
let mut image_urls = Vec::new();
|
||||
|
||||
for node in node_vec {
|
||||
image_urls.push(Self::harvest_image_urls_from_node(node)?);
|
||||
if let Ok(url) = Self::harvest_image_urls_from_node(node) {
|
||||
image_urls.push(url);
|
||||
}
|
||||
}
|
||||
|
||||
Ok(image_urls)
|
||||
|
@ -274,7 +276,7 @@ impl ImageDownloader {
|
|||
}
|
||||
|
||||
let image_base64 = base64::engine::general_purpose::STANDARD.encode(&image.data);
|
||||
let image_string = format!("data:{};base64,{}", content_type, image_base64);
|
||||
let image_string = format!("data:{content_type};base64,{image_base64}");
|
||||
let image_data_base64 = ImageDataBase64 {
|
||||
url: image.url,
|
||||
data: image_string,
|
||||
|
|
|
@ -264,17 +264,15 @@ impl Util {
|
|||
context: &Context,
|
||||
id_or_class: &str,
|
||||
) -> Result<(), FullTextParserError> {
|
||||
let xpath = &format!(
|
||||
"//*[contains(@class, '{}') or contains(@id, '{}')]",
|
||||
id_or_class, id_or_class
|
||||
);
|
||||
let xpath =
|
||||
&format!("//*[contains(@class, '{id_or_class}') or contains(@id, '{id_or_class}')]");
|
||||
|
||||
let mut ancestor = xpath.clone();
|
||||
if ancestor.starts_with("//") {
|
||||
ancestor = ancestor.chars().skip(2).collect();
|
||||
}
|
||||
|
||||
let query = &format!("{}[not(ancestor::{})]", xpath, ancestor);
|
||||
let query = &format!("{xpath}[not(ancestor::{ancestor})]");
|
||||
let node_vec = Util::evaluate_xpath(context, query, false)?;
|
||||
for mut node in node_vec {
|
||||
if node.is_null() {
|
||||
|
|
|
@ -14,6 +14,17 @@ clap = { version = "4.5", features = [ "derive" ] }
|
|||
simplelog = "0.12"
|
||||
log = "0.4"
|
||||
url = "2.5"
|
||||
reqwest = { version = "0.12", features = ["json", "native-tls", "gzip", "brotli", "stream"] }
|
||||
tokio = { version = "1", features = ["macros", "fs", "io-util", "rt-multi-thread" ] }
|
||||
indicatif = "0.17"
|
||||
reqwest = { version = "0.12", features = [
|
||||
"json",
|
||||
"native-tls",
|
||||
"gzip",
|
||||
"brotli",
|
||||
"stream",
|
||||
] }
|
||||
tokio = { version = "1", features = [
|
||||
"macros",
|
||||
"fs",
|
||||
"io-util",
|
||||
"rt-multi-thread",
|
||||
] }
|
||||
indicatif = "0.18"
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue