dhalucario/sticker-usage-analyzer-rs
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Use NaiveDateTime to fix weird offsets

Browse source
This commit is contained in:
Leon Grünewald 2024-12-11 21:48:38 +01:00
parent 48cae5f709
commit 2303b4d71d
1 changed files with 17 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
src/main.rs
View file

@ -3,6 +3,7 @@ use rocket::{Rocket,Build,launch,get,routes,fs::{FileServer}};
use std::{fs::OpenOptions, io::BufReader, collections::HashMap}; use std::{fs::OpenOptions, io::BufReader, collections::HashMap};
use chrono::prelude::*; use chrono::prelude::*;
const TELEGRAM_TIMESTAMP_FORMAT: &str = "%FT%X";
#[derive(Debug, Clone, PartialEq, serde::Serialize, serde::Deserialize)] #[derive(Debug, Clone, PartialEq, serde::Serialize, serde::Deserialize)]
#[serde(rename_all = "snake_case")] #[serde(rename_all = "snake_case")]
@ -18,17 +19,19 @@ enum MediaType {
#[derive(Debug, Clone)] #[derive(Debug, Clone)]
struct ScoredChatMessage { struct ScoredChatMessage {
pub chat_message: ChatMessage, pub chat_message: ChatMessage,
pub last_used: i64, pub last_used: NaiveDateTime,
pub times: u32 pub times: u32
} }
impl From<ChatMessage> for ScoredChatMessage { impl TryFrom<ChatMessage> for ScoredChatMessage {
fn from(item: ChatMessage) -> ScoredChatMessage { type Error = chrono::format::ParseError;
ScoredChatMessage {
chat_message: item.clone(), fn try_from(value: ChatMessage) -> Result<Self, Self::Error> {
last_used: item.date_unixtime.parse().expect("Could not parse date_unixtime"), Ok(ScoredChatMessage {
chat_message: value.clone(),
last_used: NaiveDateTime::parse_from_str(&value.date, TELEGRAM_TIMESTAMP_FORMAT)?,
times: 1 times: 1
} })
} }
} }
@ -37,7 +40,7 @@ impl From<ScoredChatMessage> for Markup {
fn from(item: ScoredChatMessage) -> Markup { fn from(item: ScoredChatMessage) -> Markup {
if let Some(mime_type) = &item.chat_message.mime_type { if let Some(mime_type) = &item.chat_message.mime_type {
let message_file = item.chat_message.file.expect("No sticker file"); let message_file = item.chat_message.file.expect("No sticker file");
let message_date = DateTime::from_timestamp(item.last_used, 0).expect("Could not parse items"); let message_date = item.last_used;
match mime_type.as_str() { match mime_type.as_str() {
"image/webp" => { "image/webp" => {
return html! { return html! {
@ -84,7 +87,8 @@ struct ChatMessage {
pub media_type: Option<MediaType>, pub media_type: Option<MediaType>,
pub mime_type: Option<String>, pub mime_type: Option<String>,
pub file: Option<String>, pub file: Option<String>,
pub date_unixtime: String pub date_unixtime: String,
pub date: String
} }
@ -113,8 +117,10 @@ fn index() -> Markup {
.filter(|m| matches!(m.media_type, Some(MediaType::Sticker))) .filter(|m| matches!(m.media_type, Some(MediaType::Sticker)))
.fold(HashMap::new(), |mut acc: HashMap<String, ScoredChatMessage>, message| { .fold(HashMap::new(), |mut acc: HashMap<String, ScoredChatMessage>, message| {
if let Some(ref file) = message.file { if let Some(ref file) = message.file {
let message_date = (&message.date_unixtime).parse().expect("Could not parse date_unixtime"); let message_date = NaiveDateTime::parse_from_str(&message.date, TELEGRAM_TIMESTAMP_FORMAT)
let entry = acc.entry(file.to_owned()).or_insert_with(|| ScoredChatMessage::from(message)); .expect("Could not parse date");
let entry = acc.entry(file.to_owned()).or_insert_with(|| ScoredChatMessage::try_from(message)
.expect("Could not parse date"));
entry.last_used = entry.last_used.max(message_date); entry.last_used = entry.last_used.max(message_date);
entry.times += 1; entry.times += 1;
} }