1
0
mirror of https://github.com/bspeice/dtparse synced 2024-11-14 09:58:09 -05:00

Added dmy! macro.

This commit is contained in:
Mark H. Colburn 2018-06-25 10:12:08 -07:00
parent 065eba823b
commit d7031ee279

View File

@ -647,6 +647,16 @@ impl YMD {
let mut day: Option<i32> = None; let mut day: Option<i32> = None;
let mut other: Option<i32> = None; let mut other: Option<i32> = None;
macro_rules! dmy {
($d:expr, $m:expr, $y:expr) => {
{
day = $d;
month = $m;
year = $y;
}
};
}
let mut strids: HashMap<YMDLabel, usize> = HashMap::new(); let mut strids: HashMap<YMDLabel, usize> = HashMap::new();
self.ystridx self.ystridx
.map(|u| strids.insert(YMDLabel::Year, u.clone())); .map(|u| strids.insert(YMDLabel::Year, u.clone()));
@ -668,7 +678,8 @@ impl YMD {
return Err(ParseInternalError::ValueError( return Err(ParseInternalError::ValueError(
"More than three YMD values".to_owned(), "More than three YMD values".to_owned(),
)); ));
} else if len_ymd == 1 || (self.mstridx.is_some() && len_ymd == 2) { }
if len_ymd == 1 || (self.mstridx.is_some() && len_ymd == 2) {
if self.mstridx.is_some() { if self.mstridx.is_some() {
month = Some(self._ymd[self.mstridx.unwrap()]); month = Some(self._ymd[self.mstridx.unwrap()]);
other = if len_ymd == 1 { other = if len_ymd == 1 {
@ -689,71 +700,47 @@ impl YMD {
} }
} else if len_ymd == 2 { } else if len_ymd == 2 {
if self._ymd[0] > 31 { if self._ymd[0] > 31 {
year = Some(self._ymd[0]); dmy!(None, Some(self._ymd[1]), Some(self._ymd[0]));
month = Some(self._ymd[1]);
} else if self._ymd[1] > 31 { } else if self._ymd[1] > 31 {
month = Some(self._ymd[0]); dmy!(None, Some(self._ymd[0]), Some(self._ymd[1]));
year = Some(self._ymd[1]);
} else if dayfirst && self._ymd[1] <= 12 { } else if dayfirst && self._ymd[1] <= 12 {
day = Some(self._ymd[0]); dmy!(Some(self._ymd[0]), Some(self._ymd[1]), None);
month = Some(self._ymd[1]);
} else { } else {
month = Some(self._ymd[0]); dmy!(Some(self._ymd[1]), Some(self._ymd[0]), None);
day = Some(self._ymd[1]);
} }
} else if len_ymd == 3 { } else if len_ymd == 3 {
if self.mstridx == Some(0) { if self.mstridx == Some(0) {
if self._ymd[1] > 31 { if self._ymd[1] > 31 {
month = Some(self._ymd[0]); dmy!(Some(self._ymd[2]), Some(self._ymd[0]), Some(self._ymd[1]));
year = Some(self._ymd[1]);
day = Some(self._ymd[2]);
} else { } else {
month = Some(self._ymd[0]); dmy!(Some(self._ymd[1]), Some(self._ymd[0]), Some(self._ymd[2]));
day = Some(self._ymd[1]);
year = Some(self._ymd[2]);
} }
} else if self.mstridx == Some(1) { } else if self.mstridx == Some(1) {
if self._ymd[0] > 31 || (yearfirst && self._ymd[2] <= 31) { if self._ymd[0] > 31 || (yearfirst && self._ymd[2] <= 31) {
year = Some(self._ymd[0]); dmy!(Some(self._ymd[2]), Some(self._ymd[1]), Some(self._ymd[0]));
month = Some(self._ymd[1]);
day = Some(self._ymd[2]);
} else { } else {
day = Some(self._ymd[0]); dmy!(Some(self._ymd[0]), Some(self._ymd[1]), Some(self._ymd[2]));
month = Some(self._ymd[1]);
year = Some(self._ymd[2]);
} }
} else if self.mstridx == Some(2) { } else if self.mstridx == Some(2) {
// It was in the original docs, so: WTF!? // It was in the original docs, so: WTF!?
if self._ymd[1] > 31 { if self._ymd[1] > 31 {
day = Some(self._ymd[0]); dmy!(Some(self._ymd[0]), Some(self._ymd[1]), Some(self._ymd[2]));
year = Some(self._ymd[1]);
month = Some(self._ymd[2]);
} else { } else {
year = Some(self._ymd[0]); dmy!(Some(self._ymd[1]), Some(self._ymd[2]), Some(self._ymd[0]));
day = Some(self._ymd[1]);
month = Some(self._ymd[2]);
} }
} else { } else {
if self._ymd[0] > 31 || self.ystridx == Some(0) if self._ymd[0] > 31 || self.ystridx == Some(0)
|| (yearfirst && self._ymd[1] <= 12 && self._ymd[2] <= 31) || (yearfirst && self._ymd[1] <= 12 && self._ymd[2] <= 31)
{ {
if dayfirst && self._ymd[2] <= 12 { if dayfirst && self._ymd[2] <= 12 {
year = Some(self._ymd[0]); dmy!(Some(self._ymd[1]), Some(self._ymd[2]), Some(self._ymd[0]));
day = Some(self._ymd[1]);
month = Some(self._ymd[2]);
} else { } else {
year = Some(self._ymd[0]); dmy!(Some(self._ymd[2]), Some(self._ymd[1]), Some(self._ymd[0]));
month = Some(self._ymd[1]);
day = Some(self._ymd[2]);
} }
} else if self._ymd[0] > 12 || (dayfirst && self._ymd[1] <= 12) { } else if self._ymd[0] > 12 || (dayfirst && self._ymd[1] <= 12) {
day = Some(self._ymd[0]); dmy!(Some(self._ymd[0]), Some(self._ymd[1]), Some(self._ymd[2]));
month = Some(self._ymd[1]);
year = Some(self._ymd[2]);
} else { } else {
month = Some(self._ymd[0]); dmy!(Some(self._ymd[1]), Some(self._ymd[0]), Some(self._ymd[2]));
day = Some(self._ymd[1]);
year = Some(self._ymd[2]);
} }
} }
} }