mirror of
https://github.com/bspeice/dtparse
synced 2024-11-14 09:58:09 -05:00
First pass done!!!
This commit is contained in:
parent
de47318411
commit
04421ebde0
@ -5,4 +5,6 @@ authors = ["Bradlee Speice <bspeice@kcg.com>"]
|
|||||||
|
|
||||||
[dependencies]
|
[dependencies]
|
||||||
chrono = "0.4"
|
chrono = "0.4"
|
||||||
|
lazy_static = "*"
|
||||||
|
num-traits = "0.2"
|
||||||
rust_decimal = "0.8"
|
rust_decimal = "0.8"
|
520
src/lib.rs
520
src/lib.rs
@ -1,7 +1,11 @@
|
|||||||
#![allow(dead_code)]
|
#![allow(dead_code)]
|
||||||
#![allow(unused)]
|
#![allow(unused)]
|
||||||
|
|
||||||
|
#[macro_use]
|
||||||
|
extern crate lazy_static;
|
||||||
|
|
||||||
extern crate chrono;
|
extern crate chrono;
|
||||||
|
extern crate num_traits;
|
||||||
extern crate rust_decimal;
|
extern crate rust_decimal;
|
||||||
|
|
||||||
use chrono::DateTime;
|
use chrono::DateTime;
|
||||||
@ -11,6 +15,7 @@ use chrono::Local;
|
|||||||
use chrono::NaiveDateTime;
|
use chrono::NaiveDateTime;
|
||||||
use chrono::NaiveTime;
|
use chrono::NaiveTime;
|
||||||
use chrono::Utc;
|
use chrono::Utc;
|
||||||
|
use num_traits::cast::ToPrimitive;
|
||||||
use rust_decimal::Decimal;
|
use rust_decimal::Decimal;
|
||||||
use rust_decimal::Error as DecimalError;
|
use rust_decimal::Error as DecimalError;
|
||||||
use std::collections::HashMap;
|
use std::collections::HashMap;
|
||||||
@ -21,6 +26,12 @@ use std::vec::Vec;
|
|||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests;
|
mod tests;
|
||||||
|
|
||||||
|
lazy_static! {
|
||||||
|
static ref ZERO: Decimal = Decimal::new(0, 0);
|
||||||
|
static ref ONE: Decimal = Decimal::new(1, 0);
|
||||||
|
static ref TWENTY_FOUR: Decimal = Decimal::new(24, 0);
|
||||||
|
static ref SIXTY: Decimal = Decimal::new(60, 0);
|
||||||
|
}
|
||||||
|
|
||||||
#[derive(Debug, PartialEq)]
|
#[derive(Debug, PartialEq)]
|
||||||
pub enum ParseInternalError {
|
pub enum ParseInternalError {
|
||||||
@ -36,21 +47,29 @@ pub enum ParseInternalError {
|
|||||||
}
|
}
|
||||||
|
|
||||||
impl From<DecimalError> for ParseInternalError {
|
impl From<DecimalError> for ParseInternalError {
|
||||||
fn from(err: DecimalError) -> Self { ParseInternalError::InvalidDecimal }
|
fn from(err: DecimalError) -> Self {
|
||||||
|
ParseInternalError::InvalidDecimal
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl From<ParseIntError> for ParseInternalError {
|
impl From<ParseIntError> for ParseInternalError {
|
||||||
fn from(err: ParseIntError) -> Self { ParseInternalError::InvalidInteger }
|
fn from(err: ParseIntError) -> Self {
|
||||||
|
ParseInternalError::InvalidInteger
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, PartialEq)]
|
#[derive(Debug)]
|
||||||
pub enum ParseError {
|
pub enum ParseError {
|
||||||
InternalError(ParseInternalError),
|
InternalError(ParseInternalError),
|
||||||
InvalidMonth,
|
InvalidMonth,
|
||||||
|
UnrecognizedToken(String),
|
||||||
|
InvalidParseResult(ParsingResult),
|
||||||
}
|
}
|
||||||
|
|
||||||
impl From<ParseInternalError> for ParseError {
|
impl From<ParseInternalError> for ParseError {
|
||||||
fn from(err: ParseInternalError) -> Self { ParseError::InternalError(err) }
|
fn from(err: ParseInternalError) -> Self {
|
||||||
|
ParseError::InternalError(err)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
type ParseResult<I> = Result<I, ParseError>;
|
type ParseResult<I> = Result<I, ParseError>;
|
||||||
@ -279,6 +298,7 @@ fn parse_info(vec: Vec<Vec<&str>>) -> HashMap<String, usize> {
|
|||||||
m
|
m
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, PartialEq)]
|
||||||
struct ParserInfo {
|
struct ParserInfo {
|
||||||
jump: HashMap<String, usize>,
|
jump: HashMap<String, usize>,
|
||||||
weekday: HashMap<String, usize>,
|
weekday: HashMap<String, usize>,
|
||||||
@ -400,13 +420,17 @@ impl ParserInfo {
|
|||||||
|
|
||||||
// TODO: Should this be moved elsewhere?
|
// TODO: Should this be moved elsewhere?
|
||||||
fn validate(&self, res: &mut ParsingResult) -> bool {
|
fn validate(&self, res: &mut ParsingResult) -> bool {
|
||||||
if let Some(y) = res.year { res.year = Some(self.convertyear(y, res.century_specified)) };
|
if let Some(y) = res.year {
|
||||||
|
res.year = Some(self.convertyear(y, res.century_specified))
|
||||||
|
};
|
||||||
|
|
||||||
if res.tzoffset == 0 && res.tzname.is_none() || res.tzname == Some("Z".to_owned()) {
|
if res.tzoffset == Some(0) && res.tzname.is_none() || res.tzname == Some("Z".to_owned()) {
|
||||||
res.tzname = Some("UTC".to_owned());
|
res.tzname = Some("UTC".to_owned());
|
||||||
res.tzoffset = 0;
|
res.tzoffset = Some(0);
|
||||||
} else if res.tzoffset != 0 && res.tzname.is_some() && self.get_utczone(res.tzname.as_ref().unwrap()) {
|
} else if res.tzoffset != Some(0) && res.tzname.is_some()
|
||||||
res.tzoffset = 0;
|
&& self.get_utczone(res.tzname.as_ref().unwrap())
|
||||||
|
{
|
||||||
|
res.tzoffset = Some(0);
|
||||||
}
|
}
|
||||||
|
|
||||||
true
|
true
|
||||||
@ -420,7 +444,11 @@ fn days_in_month(year: i32, month: i32) -> Result<i32, ParseError> {
|
|||||||
};
|
};
|
||||||
|
|
||||||
match month {
|
match month {
|
||||||
2 => if leap_year { Ok(29) } else { Ok(28) },
|
2 => if leap_year {
|
||||||
|
Ok(29)
|
||||||
|
} else {
|
||||||
|
Ok(28)
|
||||||
|
},
|
||||||
1 | 3 | 5 | 7 | 8 | 10 | 12 => Ok(31),
|
1 | 3 | 5 | 7 | 8 | 10 | 12 => Ok(31),
|
||||||
4 | 6 | 9 | 11 => Ok(30),
|
4 | 6 | 9 | 11 => Ok(30),
|
||||||
_ => Err(ParseError::InvalidMonth),
|
_ => Err(ParseError::InvalidMonth),
|
||||||
@ -444,23 +472,24 @@ struct YMD {
|
|||||||
}
|
}
|
||||||
|
|
||||||
impl YMD {
|
impl YMD {
|
||||||
|
fn len(&self) -> usize {
|
||||||
|
self._ymd.len()
|
||||||
|
}
|
||||||
|
|
||||||
fn len(&self) -> usize { self._ymd.len() }
|
fn could_be_day(&self, val: i32) -> bool {
|
||||||
|
|
||||||
fn could_be_day(&self, val: i32) -> ParseResult<bool> {
|
|
||||||
if self.dstridx.is_some() {
|
if self.dstridx.is_some() {
|
||||||
Ok(false)
|
false
|
||||||
} else if self.mstridx.is_none() {
|
} else if self.mstridx.is_none() {
|
||||||
Ok((1 <= val) && (val <= 31))
|
(1 <= val) && (val <= 31)
|
||||||
} else if self.ystridx.is_none() {
|
} else if self.ystridx.is_none() {
|
||||||
// UNWRAP: mstridx guaranteed to have a value
|
// UNWRAP: mstridx guaranteed to have a value
|
||||||
// TODO: Justify unwrap for self._ymd
|
// TODO: Justify unwrap for self._ymd
|
||||||
let month = self._ymd[self.mstridx.unwrap()];
|
let month = self._ymd[self.mstridx.unwrap()];
|
||||||
Ok(1 <= val && (val <= days_in_month(2000, month)?))
|
1 <= val && (val <= days_in_month(2000, month).unwrap())
|
||||||
} else {
|
} else {
|
||||||
let month = self._ymd[self.mstridx.unwrap()];
|
let month = self._ymd[self.mstridx.unwrap()];
|
||||||
let year = self._ymd[self.ystridx.unwrap()];
|
let year = self._ymd[self.ystridx.unwrap()];
|
||||||
Ok(1 <= val && (val <= days_in_month(year, month)?))
|
1 <= val && (val <= days_in_month(year, month).unwrap())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -677,8 +706,8 @@ impl YMD {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Default)]
|
#[derive(Default, Debug)]
|
||||||
struct ParsingResult {
|
pub struct ParsingResult {
|
||||||
year: Option<i32>,
|
year: Option<i32>,
|
||||||
month: Option<i32>,
|
month: Option<i32>,
|
||||||
day: Option<i32>,
|
day: Option<i32>,
|
||||||
@ -688,8 +717,8 @@ struct ParsingResult {
|
|||||||
second: Option<i32>,
|
second: Option<i32>,
|
||||||
microsecond: Option<i32>,
|
microsecond: Option<i32>,
|
||||||
tzname: Option<String>,
|
tzname: Option<String>,
|
||||||
tzoffset: i32,
|
tzoffset: Option<i32>,
|
||||||
ampm: Option<bool>,
|
ampm: Option<usize>,
|
||||||
century_specified: bool,
|
century_specified: bool,
|
||||||
any_unused_tokens: Vec<String>,
|
any_unused_tokens: Vec<String>,
|
||||||
}
|
}
|
||||||
@ -700,25 +729,29 @@ struct Parser {
|
|||||||
}
|
}
|
||||||
|
|
||||||
impl Parser {
|
impl Parser {
|
||||||
|
|
||||||
pub fn parse(
|
pub fn parse(
|
||||||
&mut self,
|
&mut self,
|
||||||
timestr: String,
|
timestr: String,
|
||||||
default: Option<NaiveDateTime>,
|
default: Option<NaiveDateTime>,
|
||||||
ignoretz: bool,
|
ignoretz: bool,
|
||||||
tzinfos: Vec<String>,
|
tzinfos: Vec<String>,
|
||||||
) -> Result<DateTime<FixedOffset>, ParseError> {
|
) -> Result<(NaiveDateTime, Option<FixedOffset>, Option<Vec<String>>), ParseError> {
|
||||||
let now = Local::now().naive_local();
|
let now = Local::now().naive_local();
|
||||||
let default_date = default.unwrap_or(now).date();
|
let default_date = default.unwrap_or(now).date();
|
||||||
|
|
||||||
let default_ts = NaiveDateTime::new(default_date, NaiveTime::from_hms(0, 0, 0));
|
let default_ts = NaiveDateTime::new(default_date, NaiveTime::from_hms(0, 0, 0));
|
||||||
|
|
||||||
// TODO: What should be done with the tokens?
|
// TODO: What should be done with the tokens?
|
||||||
let (res, tokens) =
|
let (res, tokens) = self.parse_with_tokens(timestr, None, None, false, false)?;
|
||||||
self.parse_with_tokens(timestr, None, None, false, false)?;
|
|
||||||
|
|
||||||
let naive = self.build_naive(&res, default_ts);
|
let naive = self.build_naive(&res, default_ts);
|
||||||
Ok(self.build_tzaware(naive, &res, default_ts))
|
|
||||||
|
if !ignoretz {
|
||||||
|
let offset = self.build_tzaware(&naive, &res, default_ts);
|
||||||
|
Ok((naive, Some(offset.unwrap()), tokens))
|
||||||
|
} else {
|
||||||
|
Ok((naive, None, tokens))
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn parse_with_tokens(
|
fn parse_with_tokens(
|
||||||
@ -728,34 +761,178 @@ impl Parser {
|
|||||||
yearfirst: Option<bool>,
|
yearfirst: Option<bool>,
|
||||||
fuzzy: bool,
|
fuzzy: bool,
|
||||||
fuzzy_with_tokens: bool,
|
fuzzy_with_tokens: bool,
|
||||||
) -> Result<(ParsingResult, Vec<String>), ParseError> {
|
) -> Result<(ParsingResult, Option<Vec<String>>), ParseError> {
|
||||||
let fuzzy = if fuzzy_with_tokens { true } else { fuzzy };
|
let fuzzy = if fuzzy_with_tokens { true } else { fuzzy };
|
||||||
// This is probably a stylistic abomination
|
// This is probably a stylistic abomination
|
||||||
let dayfirst = if let Some(dayfirst) = dayfirst { dayfirst } else { self.info.dayfirst };
|
let dayfirst = if let Some(dayfirst) = dayfirst {
|
||||||
let yearfirst = if let Some(yearfirst) = yearfirst { yearfirst } else { self.info.yearfirst };
|
dayfirst
|
||||||
|
} else {
|
||||||
|
self.info.dayfirst
|
||||||
|
};
|
||||||
|
let yearfirst = if let Some(yearfirst) = yearfirst {
|
||||||
|
yearfirst
|
||||||
|
} else {
|
||||||
|
self.info.yearfirst
|
||||||
|
};
|
||||||
|
|
||||||
let mut res = ParsingResult::default();
|
let mut res = ParsingResult::default();
|
||||||
|
|
||||||
let l = tokenize(×tr);
|
let mut l = tokenize(×tr);
|
||||||
let skipped_idxs: Vec<usize> = Vec::new();
|
let mut skipped_idxs: Vec<usize> = Vec::new();
|
||||||
|
|
||||||
let ymd = YMD::default();
|
let mut ymd = YMD::default();
|
||||||
|
|
||||||
let len_l = l.len();
|
let len_l = l.len();
|
||||||
let mut i = 0;
|
let mut i = 0;
|
||||||
|
|
||||||
while i < len_l {
|
while i < len_l {
|
||||||
|
let value_repr = l[i].clone();
|
||||||
let value_repr = l.get(i).ok_or(ParseInternalError::ParseIndexError)?;
|
|
||||||
|
|
||||||
let value = value_repr.parse::<f32>();
|
let value = value_repr.parse::<f32>();
|
||||||
|
|
||||||
if let Ok(v) = value {
|
if let Ok(v) = value {
|
||||||
i = self.parse_numeric_token(&l, i, &self.info, &ymd, &mut res, fuzzy)?;
|
i = self.parse_numeric_token(&l, i, &self.info, &mut ymd, &mut res, fuzzy)?;
|
||||||
|
} else if let Some(value) = self.info.get_weekday(&l[i]) {
|
||||||
|
res.weekday = Some(value != 0);
|
||||||
|
} else if let Some(value) = self.info.get_month(&l[i]) {
|
||||||
|
ymd.append(value as i32, Some(YMDLabel::Month));
|
||||||
|
|
||||||
|
if i + 1 < len_l {
|
||||||
|
if l[i + 1] == "-" || l[i + 1] == "/" {
|
||||||
|
// Jan-01[-99]
|
||||||
|
let sep = &l[i + 1];
|
||||||
|
// TODO: This seems like a very unsafe unwrap
|
||||||
|
ymd.append(l[i + 2].parse::<i32>().unwrap(), None);
|
||||||
|
|
||||||
|
if i + 3 < len_l && &l[i + 3] == sep {
|
||||||
|
// Jan-01-99
|
||||||
|
ymd.append(l[i + 4].parse::<i32>().unwrap(), None);
|
||||||
|
i += 2;
|
||||||
|
}
|
||||||
|
|
||||||
|
i += 2;
|
||||||
|
} else if (i + 4 < len_l && l[i + 1] == l[i + 3] && l[i + 3] == " "
|
||||||
|
&& self.info.get_pertain(&l[i + 2]))
|
||||||
|
{
|
||||||
|
// Jan of 01
|
||||||
|
if let Some(value) = l[i + 4].parse::<i32>().ok() {
|
||||||
|
let year = self.info.convertyear(value, false);
|
||||||
|
ymd.append(year, Some(YMDLabel::Year));
|
||||||
|
}
|
||||||
|
|
||||||
|
i += 4;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else if let Some(value) = self.info.get_ampm(&l[i]) {
|
||||||
|
let is_ampm = self.ampm_valid(res.hour, res.ampm, fuzzy);
|
||||||
|
|
||||||
|
if is_ampm {
|
||||||
|
res.hour = Some(self.adjust_ampm(res.hour.unwrap(), value));
|
||||||
|
res.ampm = Some(value);
|
||||||
|
} else if fuzzy {
|
||||||
|
skipped_idxs.push(i);
|
||||||
|
}
|
||||||
|
} else if self.could_be_tzname(res.hour, res.tzname.clone(), res.tzoffset, &l[i]) {
|
||||||
|
res.tzname = Some(l[i].clone());
|
||||||
|
|
||||||
|
let tzname = res.tzname.clone().unwrap();
|
||||||
|
res.tzoffset = self.info.get_tzoffset(&tzname).map(|t| t as i32);
|
||||||
|
|
||||||
|
if i + 1 < len_l && (l[i + 1] == "+" || l[i + 1] == "-") {
|
||||||
|
// GMT+3
|
||||||
|
// According to dateutil docs - reverse the size, as GMT+3 means
|
||||||
|
// "my time +3 is GMT" not "GMT +3 is my time"
|
||||||
|
|
||||||
|
// TODO: Is there a better way of in-place modifying a vector?
|
||||||
|
let item = if l[i + 1] == "+" {
|
||||||
|
"-".to_owned()
|
||||||
|
} else {
|
||||||
|
"-".to_owned()
|
||||||
|
};
|
||||||
|
l.remove(i + 1);
|
||||||
|
l.insert(i + 1, item);
|
||||||
|
|
||||||
|
res.tzoffset = None;
|
||||||
|
|
||||||
|
if self.info.get_utczone(&tzname) {
|
||||||
|
res.tzname = None;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else if res.hour.is_some() && (l[i] == "+" || l[i] == "-") {
|
||||||
|
let signal = if l[i] == "+" { 1 } else { -1 };
|
||||||
|
let len_li = l[i].len();
|
||||||
|
|
||||||
|
let mut hour_offset: Option<i32> = None;
|
||||||
|
let mut min_offset: Option<i32> = None;
|
||||||
|
|
||||||
|
// TODO: check that l[i + 1] is integer?
|
||||||
|
if len_li == 4 {
|
||||||
|
// -0300
|
||||||
|
hour_offset = Some(l[i + 1][..2].parse::<i32>().unwrap());
|
||||||
|
min_offset = Some(l[i + 1][2..4].parse::<i32>().unwrap());
|
||||||
|
} else if i + 2 < len_l && l[i + 2] == ":" {
|
||||||
|
// -03:00
|
||||||
|
hour_offset = Some(l[i + 1].parse::<i32>().unwrap());
|
||||||
|
min_offset = Some(l[i + 3].parse::<i32>().unwrap());
|
||||||
|
i += 2;
|
||||||
|
} else if len_li <= 2 {
|
||||||
|
// -[0]3
|
||||||
|
hour_offset = Some(l[i + 1][..2].parse::<i32>().unwrap());
|
||||||
|
min_offset = Some(0);
|
||||||
|
}
|
||||||
|
|
||||||
|
res.tzoffset =
|
||||||
|
Some(signal * (hour_offset.unwrap() * 3600 + min_offset.unwrap() * 60));
|
||||||
|
|
||||||
|
let tzname = res.tzname.clone();
|
||||||
|
if i + 5 < len_l && self.info.get_jump(&l[i + 2]) && l[i + 3] == "("
|
||||||
|
&& l[i + 5] == ")" && 3 <= l[i + 4].len()
|
||||||
|
&& self.could_be_tzname(res.hour, tzname, None, &l[i + 4])
|
||||||
|
{
|
||||||
|
// (GMT)
|
||||||
|
res.tzname = Some(l[i + 4].clone());
|
||||||
|
i += 4;
|
||||||
|
}
|
||||||
|
|
||||||
|
i += 1;
|
||||||
|
} else if !self.info.get_jump(&l[i]) || fuzzy {
|
||||||
|
return Err(ParseError::UnrecognizedToken(l[i].clone()));
|
||||||
|
} else {
|
||||||
|
skipped_idxs.push(i);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
i += 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
Err(ParseError::InvalidMonth)
|
let (year, month, day) = ymd.resolve_ymd(yearfirst, dayfirst)?;
|
||||||
|
|
||||||
|
res.century_specified = ymd.century_specified;
|
||||||
|
res.year = Some(year);
|
||||||
|
res.month = Some(month);
|
||||||
|
res.day = Some(day);
|
||||||
|
|
||||||
|
if !self.info.validate(&mut res) {
|
||||||
|
Err(ParseError::InvalidParseResult(res))
|
||||||
|
} else if fuzzy_with_tokens {
|
||||||
|
let skipped_tokens = skipped_idxs.into_iter().map(|i| l[i].clone()).collect();
|
||||||
|
Ok((res, Some(skipped_tokens)))
|
||||||
|
} else {
|
||||||
|
Ok((res, None))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn could_be_tzname(
|
||||||
|
&self,
|
||||||
|
hour: Option<i32>,
|
||||||
|
tzname: Option<String>,
|
||||||
|
tzoffset: Option<i32>,
|
||||||
|
token: &str,
|
||||||
|
) -> bool {
|
||||||
|
false
|
||||||
|
}
|
||||||
|
|
||||||
|
fn ampm_valid(&self, hour: Option<i32>, ampm: Option<usize>, fuzzy: bool) -> bool {
|
||||||
|
false
|
||||||
}
|
}
|
||||||
|
|
||||||
fn build_naive(&self, res: &ParsingResult, default: NaiveDateTime) -> NaiveDateTime {
|
fn build_naive(&self, res: &ParsingResult, default: NaiveDateTime) -> NaiveDateTime {
|
||||||
@ -764,47 +941,280 @@ impl Parser {
|
|||||||
|
|
||||||
fn build_tzaware(
|
fn build_tzaware(
|
||||||
&self,
|
&self,
|
||||||
dt: NaiveDateTime,
|
dt: &NaiveDateTime,
|
||||||
res: &ParsingResult,
|
res: &ParsingResult,
|
||||||
default: NaiveDateTime,
|
default: NaiveDateTime,
|
||||||
) -> DateTime<FixedOffset> {
|
) -> Result<FixedOffset, ParseError> {
|
||||||
|
Ok(FixedOffset::east(0))
|
||||||
Local::now().with_timezone(&FixedOffset::east(0))
|
|
||||||
}
|
}
|
||||||
|
|
||||||
fn parse_numeric_token(&self, tokens: &Vec<String>, idx: usize, info: &ParserInfo, ymd: &YMD, res: &mut ParsingResult, fuzzy: bool) -> Result<usize, ParseInternalError> {
|
fn parse_numeric_token(
|
||||||
|
&self,
|
||||||
|
tokens: &Vec<String>,
|
||||||
|
idx: usize,
|
||||||
|
info: &ParserInfo,
|
||||||
|
ymd: &mut YMD,
|
||||||
|
res: &mut ParsingResult,
|
||||||
|
fuzzy: bool,
|
||||||
|
) -> Result<usize, ParseInternalError> {
|
||||||
|
let mut idx = idx;
|
||||||
let value_repr = &tokens[idx];
|
let value_repr = &tokens[idx];
|
||||||
let value = Decimal::from_str(&value_repr)?;
|
let mut value = Decimal::from_str(&value_repr).unwrap();
|
||||||
|
|
||||||
let len_li = value_repr.len();
|
let len_li = value_repr.len();
|
||||||
let len_l = tokens.len();
|
let len_l = tokens.len();
|
||||||
|
|
||||||
let mut s: Option<&str> = None;
|
|
||||||
|
|
||||||
// TODO: I miss the `x in y` syntax
|
// TODO: I miss the `x in y` syntax
|
||||||
// TODO: Decompose this logic a bit
|
// TODO: Decompose this logic a bit
|
||||||
if ymd.len() == 3 && (len_li == 2 || len_li == 4) &&
|
if ymd.len() == 3 && (len_li == 2 || len_li == 4) && res.hour.is_none()
|
||||||
res.hour.is_none() && (
|
&& (idx + 1 >= len_l
|
||||||
idx + 1 >= len_l ||
|
|| (tokens[idx + 1] != ":" && info.get_hms(&tokens[idx + 1]).is_none()))
|
||||||
(tokens[idx + 1] != ":" && info.get_hms(&tokens[idx + 1]).is_none())) {
|
{
|
||||||
|
|
||||||
// 1990101T32[59]
|
// 1990101T32[59]
|
||||||
s = Some(&tokens[idx]);
|
let s = &tokens[idx];
|
||||||
res.hour = Some(s.unwrap()[0..2].parse::<i32>()?);
|
res.hour = s[0..2].parse::<i32>().ok();
|
||||||
|
|
||||||
if len_li == 4 { res.minute = Some(s.unwrap()[2..4].parse::<i32>()?) }
|
if len_li == 4 {
|
||||||
|
res.minute = Some(s[2..4].parse::<i32>()?)
|
||||||
|
}
|
||||||
|
} else if len_li == 6 || (len_li > 6 && tokens[idx].find(".") == Some(6)) {
|
||||||
|
// YYMMDD or HHMMSS[.ss]
|
||||||
|
let s = &tokens[idx];
|
||||||
|
|
||||||
|
if ymd.len() == 0 && tokens[idx].find(".") == None {
|
||||||
|
ymd.append(s[0..2].parse::<i32>().unwrap(), None);
|
||||||
|
ymd.append(s[2..4].parse::<i32>().unwrap(), None);
|
||||||
|
ymd.append(s[4..6].parse::<i32>().unwrap(), None);
|
||||||
|
} else {
|
||||||
|
// 19990101T235959[.59]
|
||||||
|
res.hour = s[0..2].parse::<i32>().ok();
|
||||||
|
res.minute = s[2..4].parse::<i32>().ok();
|
||||||
|
|
||||||
|
let t = self.parsems(&s[4..])?;
|
||||||
|
res.second = Some(t.0);
|
||||||
|
res.microsecond = Some(t.1);
|
||||||
|
}
|
||||||
|
} else if vec![8, 12, 14].contains(&len_li) {
|
||||||
|
// YYMMDD
|
||||||
|
let s = &tokens[idx];
|
||||||
|
ymd.append(s[..4].parse::<i32>().unwrap(), Some(YMDLabel::Year));
|
||||||
|
ymd.append(s[4..6].parse::<i32>().unwrap(), None);
|
||||||
|
ymd.append(s[6..8].parse::<i32>().unwrap(), None);
|
||||||
|
|
||||||
|
if len_li > 8 {
|
||||||
|
res.hour = Some(s[8..10].parse::<i32>()?);
|
||||||
|
res.minute = Some(s[10..12].parse::<i32>()?);
|
||||||
|
|
||||||
|
if len_li > 12 {
|
||||||
|
res.second = Some(s[12..].parse::<i32>()?);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else if let Some(hms_idx) = self.find_hms_index(idx, tokens, info, true) {
|
||||||
|
// HH[ ]h or MM[ ]m or SS[.ss][ ]s
|
||||||
|
let (idx, hms) = self.parse_hms(idx, tokens, info, Some(hms_idx));
|
||||||
|
if hms.is_some() {
|
||||||
|
// TODO: This unwrap is unjustified.
|
||||||
|
self.assign_hms(res, value_repr, hms.unwrap());
|
||||||
|
}
|
||||||
|
} else if idx + 2 < len_l && tokens[idx + 1] == ":" {
|
||||||
|
// HH:MM[:SS[.ss]]
|
||||||
|
// TODO: Better story around Decimal handling
|
||||||
|
res.hour = Some(value.floor().to_i64().unwrap() as i32);
|
||||||
|
// TODO: Rescope `value` here?
|
||||||
|
value = self.to_decimal(&tokens[idx + 2]);
|
||||||
|
let min_sec = self.parse_min_sec(value);
|
||||||
|
res.minute = Some(min_sec.0);
|
||||||
|
res.second = min_sec.1;
|
||||||
|
|
||||||
|
if idx + 4 < len_l && tokens[idx + 3] == ":" {
|
||||||
|
// TODO: (x, y) = (a, b) syntax?
|
||||||
|
let ms = self.parsems(&tokens[idx + 4]).unwrap();
|
||||||
|
res.second = Some(ms.0);
|
||||||
|
res.microsecond = Some(ms.1);
|
||||||
|
|
||||||
|
idx += 2;
|
||||||
|
}
|
||||||
|
idx += 2;
|
||||||
|
} else if idx + 1 < len_l
|
||||||
|
&& (tokens[idx + 1] == "-" || tokens[idx + 1] == "/" || tokens[idx + 1] == ".")
|
||||||
|
{
|
||||||
|
// TODO: There's got to be a better way of handling the condition above
|
||||||
|
let sep = &tokens[idx + 1];
|
||||||
|
ymd.append(value_repr.parse::<i32>().unwrap(), None);
|
||||||
|
|
||||||
|
if idx + 2 < len_l && !info.get_jump(&tokens[idx + 2]) {
|
||||||
|
if let Ok(val) = tokens[idx + 2].parse::<i32>() {
|
||||||
|
ymd.append(val, None);
|
||||||
|
} else if let Some(val) = info.get_month(&tokens[idx + 2]) {
|
||||||
|
ymd.append(val as i32, Some(YMDLabel::Month));
|
||||||
|
}
|
||||||
|
|
||||||
|
if idx + 3 < len_l && &tokens[idx + 3] == sep {
|
||||||
|
if let Some(value) = info.get_month(&tokens[idx + 4]) {
|
||||||
|
ymd.append(value as i32, Some(YMDLabel::Month));
|
||||||
|
} else {
|
||||||
|
ymd.append(tokens[idx + 4].parse::<i32>().unwrap(), None);
|
||||||
|
}
|
||||||
|
|
||||||
|
idx += 2;
|
||||||
|
}
|
||||||
|
|
||||||
|
idx += 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
idx += 1
|
||||||
|
} else if idx + 1 >= len_l || info.get_jump(&tokens[idx + 1]) {
|
||||||
|
if idx + 2 < len_l && info.get_ampm(&tokens[idx + 2]).is_some() {
|
||||||
|
let hour = value.to_i64().unwrap() as i32;
|
||||||
|
let ampm = info.get_ampm(&tokens[idx + 2]).unwrap();
|
||||||
|
res.hour = Some(self.adjust_ampm(hour, ampm));
|
||||||
|
}
|
||||||
|
} else if info.get_ampm(&tokens[idx + 1]).is_some()
|
||||||
|
&& (*ZERO <= value && value < *TWENTY_FOUR)
|
||||||
|
{
|
||||||
|
// 12am
|
||||||
|
let hour = value.to_i64().unwrap() as i32;
|
||||||
|
res.hour = Some(self.adjust_ampm(hour, info.get_ampm(&tokens[idx + 1]).unwrap()));
|
||||||
|
idx += 1;
|
||||||
|
} else if ymd.could_be_day(value.to_i64().unwrap() as i32) {
|
||||||
|
ymd.append(value.to_i64().unwrap() as i32, None);
|
||||||
|
} else if !fuzzy {
|
||||||
|
return Err(ParseInternalError::ValueError("".to_owned()));
|
||||||
}
|
}
|
||||||
|
|
||||||
Ok(idx)
|
Ok(idx)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn adjust_ampm(&self, hour: i32, ampm: usize) -> i32 {
|
||||||
|
if hour < 12 && ampm == 1 {
|
||||||
|
hour + 12
|
||||||
|
} else if hour == 12 && ampm == 0 {
|
||||||
|
0
|
||||||
|
} else {
|
||||||
|
hour
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn parsems(&self, seconds_str: &str) -> Result<(i32, i32), ParseInternalError> {
|
||||||
|
if seconds_str.contains(".") {
|
||||||
|
let split: Vec<&str> = seconds_str.split(".").collect();
|
||||||
|
let (i, f): (&str, &str) = (split[0], split[1]);
|
||||||
|
|
||||||
|
let i_parse = i.parse::<i32>()?;
|
||||||
|
let f_parse = ljust(f, 6, '0').parse::<i32>()?;
|
||||||
|
Ok((i_parse, f_parse))
|
||||||
|
} else {
|
||||||
|
Ok((seconds_str.parse::<i32>()?, 0))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn find_hms_index(
|
||||||
|
&self,
|
||||||
|
idx: usize,
|
||||||
|
tokens: &Vec<String>,
|
||||||
|
info: &ParserInfo,
|
||||||
|
allow_jump: bool,
|
||||||
|
) -> Option<usize> {
|
||||||
|
let len_l = tokens.len();
|
||||||
|
let mut hms_idx = None;
|
||||||
|
|
||||||
|
if idx + 1 < len_l && info.get_hms(&tokens[idx + 1]).is_some() {
|
||||||
|
hms_idx = Some(idx + 1)
|
||||||
|
} else if allow_jump && idx + 2 < len_l && tokens[idx + 1] == " "
|
||||||
|
&& info.get_hms(&tokens[idx + 2]).is_some()
|
||||||
|
{
|
||||||
|
hms_idx = Some(idx + 2)
|
||||||
|
} else if idx > 0 && info.get_hms(&tokens[idx - 1]).is_some() {
|
||||||
|
hms_idx = Some(idx - 1)
|
||||||
|
}
|
||||||
|
// TODO: The condition for this in Python seems a bit ambiguous
|
||||||
|
else if idx == len_l - 1 && tokens[idx - 1] == " "
|
||||||
|
&& info.get_hms(&tokens[idx - 2]).is_some()
|
||||||
|
{
|
||||||
|
hms_idx = Some(idx - 2)
|
||||||
|
}
|
||||||
|
|
||||||
|
hms_idx
|
||||||
|
}
|
||||||
|
|
||||||
|
fn parse_hms(
|
||||||
|
&self,
|
||||||
|
idx: usize,
|
||||||
|
tokens: &Vec<String>,
|
||||||
|
info: &ParserInfo,
|
||||||
|
hms_index: Option<usize>,
|
||||||
|
) -> (usize, Option<usize>) {
|
||||||
|
if hms_index.is_none() {
|
||||||
|
(idx, None)
|
||||||
|
} else if hms_index.unwrap() > idx {
|
||||||
|
(
|
||||||
|
hms_index.unwrap(),
|
||||||
|
info.get_hms(&tokens[hms_index.unwrap()]),
|
||||||
|
)
|
||||||
|
} else {
|
||||||
|
(
|
||||||
|
idx,
|
||||||
|
info.get_hms(&tokens[hms_index.unwrap()]).map(|u| u + 1),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn assign_hms(&self, res: &mut ParsingResult, value_repr: &str, hms: usize) {
|
||||||
|
let value = self.to_decimal(value_repr);
|
||||||
|
|
||||||
|
if hms == 0 {
|
||||||
|
res.hour = Some(value.to_i64().unwrap() as i32);
|
||||||
|
if close_to_integer(&value) {
|
||||||
|
// TODO: High probability of issues with rounding here.
|
||||||
|
res.minute = Some((*SIXTY * (value % *ONE)).to_i64().unwrap() as i32);
|
||||||
|
}
|
||||||
|
} else if hms == 1 {
|
||||||
|
let (min, sec) = self.parse_min_sec(value);
|
||||||
|
} else if hms == 2 {
|
||||||
|
let (sec, micro) = self.parsems(value_repr).unwrap();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn to_decimal(&self, value: &str) -> Decimal {
|
||||||
|
// TODO: Actual decimals, and handling infinity
|
||||||
|
Decimal::from_str(value).unwrap()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn parse_min_sec(&self, value: Decimal) -> (i32, Option<i32>) {
|
||||||
|
let minute = value.floor().to_i64().unwrap() as i32;
|
||||||
|
let mut second = None;
|
||||||
|
|
||||||
|
let sec_remainder = value % *ONE;
|
||||||
|
if sec_remainder != *ZERO {
|
||||||
|
second = Some((*SIXTY * sec_remainder).floor().to_i64().unwrap() as i32);
|
||||||
|
}
|
||||||
|
|
||||||
|
(minute, second)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn parse_with_info(timestr: String, info: ParserInfo) -> Result<DateTime<FixedOffset>, ParseError> {
|
fn close_to_integer(value: &Decimal) -> bool {
|
||||||
|
value % *ONE == *ZERO
|
||||||
|
}
|
||||||
|
|
||||||
|
fn ljust(s: &str, chars: usize, replace: char) -> String {
|
||||||
|
if s.len() >= chars {
|
||||||
|
s[..chars].to_owned()
|
||||||
|
} else {
|
||||||
|
format!("{}{}", s, replace.to_string().repeat(chars - s.len()))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn parse_with_info(
|
||||||
|
timestr: String,
|
||||||
|
info: ParserInfo,
|
||||||
|
) -> ParseResult<(NaiveDateTime, Option<FixedOffset>, Option<Vec<String>>)> {
|
||||||
// TODO: Is `::new()` more stylistic?
|
// TODO: Is `::new()` more stylistic?
|
||||||
let mut parser = Parser { info: info };
|
let mut parser = Parser { info: info };
|
||||||
parser.parse(timestr, None, false, vec![])
|
parser.parse(timestr, None, false, vec![])
|
||||||
}
|
}
|
||||||
|
|
||||||
fn parse(timestr: String) -> Result<DateTime<FixedOffset>, ParseError> {
|
fn parse(timestr: String) -> ParseResult<(NaiveDateTime, Option<FixedOffset>)> {
|
||||||
parse_with_info(timestr, ParserInfo::default())
|
let parse_result = parse_with_info(timestr, ParserInfo::default())?;
|
||||||
|
Ok((parse_result.0, parse_result.1))
|
||||||
}
|
}
|
||||||
|
@ -1,4 +1,3 @@
|
|||||||
|
|
||||||
// WARNING
|
// WARNING
|
||||||
// This file was auto-generated using the `build_tests.py` script.
|
// This file was auto-generated using the `build_tests.py` script.
|
||||||
// Please do not edit it manually.
|
// Please do not edit it manually.
|
||||||
|
@ -1,4 +1,3 @@
|
|||||||
|
|
||||||
// WARNING
|
// WARNING
|
||||||
// This file was auto-generated using the `build_tests.py` script.
|
// This file was auto-generated using the `build_tests.py` script.
|
||||||
// Please do not edit it manually.
|
// Please do not edit it manually.
|
||||||
@ -42,18 +41,10 @@ fn test_python_compat() {
|
|||||||
);
|
);
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
tokenize("19990101T23"),
|
tokenize("19990101T23"),
|
||||||
vec![
|
vec!["19990101".to_owned(), "T".to_owned(), "23".to_owned()]
|
||||||
"19990101".to_owned(),
|
|
||||||
"T".to_owned(),
|
|
||||||
"23".to_owned(),
|
|
||||||
]
|
|
||||||
);
|
);
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
tokenize("19990101T2359"),
|
tokenize("19990101T2359"),
|
||||||
vec![
|
vec!["19990101".to_owned(), "T".to_owned(), "2359".to_owned()]
|
||||||
"19990101".to_owned(),
|
|
||||||
"T".to_owned(),
|
|
||||||
"2359".to_owned(),
|
|
||||||
]
|
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user