Anri/cal7tor
1
0
Fork 0
forked from Anri/cal8tor
Code Issues 2 Pull requests Releases 8 Activity

Should be better now ?

Browse source
This commit is contained in:
Mylloon 2023-09-26 23:55:54 +02:00
parent 189b77cc4f
commit bbee1d58be
Signed by: Anri
GPG key ID: A82D63DFF8D1317F
1 changed files with 23 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
src/timetable.rs
View file

@ -1,7 +1,7 @@
use chrono::{Datelike, Duration, TimeZone, Utc}; use chrono::{Datelike, Duration, NaiveTime, TimeZone, Utc};
use regex::Regex; use regex::Regex;
use scraper::{Html, Selector}; use scraper::{Html, Selector};
use std::collections::HashMap; use std::{collections::HashMap, str::FromStr};
use crate::utils::{ use crate::utils::{
self, self,
@ -35,15 +35,16 @@ pub async fn timetable(
// Find the timetable // Find the timetable
let raw_timetable = document.select(&sel_table).next().unwrap(); let raw_timetable = document.select(&sel_table).next().unwrap();
let mut schedules = Vec::new(); let mut hours = Vec::new();
for hour in 8..=20 { for hour in 8..=20 {
for minute in &[0, 15, 30, 45] { for minute in &[0, 15, 30, 45] {
let hour_str = format!("{}h{:02}", hour, minute); let hour_str = format!("{}h{:02}", hour, minute);
schedules.push(hour_str); hours.push(hour_str);
} }
} }
let mut timetable: Vec<models::Day> = Vec::new(); let mut timetable: Vec<models::Day> = Vec::new();
let mut schedules = Vec::new();
raw_timetable raw_timetable
.select(&sel_tbody) .select(&sel_tbody)
@ -58,11 +59,15 @@ pub async fn timetable(
.captures(i.value().attr("title").unwrap()) .captures(i.value().attr("title").unwrap())
.unwrap(); .unwrap();
let day: &str = matches let day = matches
.name("day") .name("day")
.unwrap() .unwrap()
.as_str(); .as_str();
let startime = matches
.name("startime")
.unwrap()
.as_str();
let binding = i.select(&sel_b).last().unwrap().inner_html(); let binding = i.select(&sel_b).last().unwrap().inner_html();
let course = models::Course{ let course = models::Course{
@ -89,10 +94,7 @@ pub async fn timetable(
.unwrap().name("location") .unwrap().name("location")
.unwrap() .unwrap()
.as_str().to_owned(), .as_str().to_owned(),
start: schedules.iter().position(|r| r == matches start: hours.iter().position(|r| r == startime).unwrap(),
.name("startime")
.unwrap()
.as_str()).unwrap(),
size: i.value().attr("rowspan").unwrap().parse::<usize >().unwrap(), size: i.value().attr("rowspan").unwrap().parse::<usize >().unwrap(),
dtstart: None, dtstart: None,
dtend: None, dtend: None,
@ -108,6 +110,18 @@ pub async fn timetable(
courses: vec![Some(course)], courses: vec![Some(course)],
}); });
} }
let duration = Regex::new(r"(?P<h>\d{1,2})h(?P<m>\d{1,2})?")
.unwrap()
.captures(matches
.name("duration")
.unwrap()
.as_str()).unwrap();
schedules.push(format!("{}-{}", startime, NaiveTime::from_str(&startime.replace('h', ":")).unwrap().overflowing_add_signed(Duration::minutes(duration.name("h").unwrap().as_str().parse::<i64>().unwrap() * 60 + match duration.name("m") {
Some(x) => x.as_str().parse::<i64>().unwrap(),
None => 0
})).0.format("%Hh%M")));
}); });
if !check_consistency(&schedules, &timetable) { if !check_consistency(&schedules, &timetable) {