def match_named_month(s: str, now: datetime) -> List[Dict[str, Any]]:
"""
Match named month and day
:param s: textual input
:return: tuple of date parts
"""
groups = re.findall(R_NAMED_MONTH, s)
months = ['jan', 'feb', 'mar', 'apr', 'maj', 'jun', 'jul', 'aug', 'szep', 'okt', 'nov', 'dec']
res = []
groups = [(mod, m, d.lstrip('0')) if d else (mod, m, '') for mod, m, d in groups]
for group in groups:
group_res = {'match': group, 'date_parts': []}
if group[0]:
if 'jovo' in remove_accent(group[0]):
group_res['date_parts'].append(Year(now.year + 1, 'named_month'))
elif 'tavaly' in remove_accent(group[0]):
group_res['date_parts'].append(Year(now.year - 1, 'named_month'))
for i, month in enumerate(months):
if month in remove_accent(group[1]):
group_res['date_parts'].append(Month(i + 1, 'named_month'))
break
if group[2]:
group_res['date_parts'].append(Day(int(group[2]), 'named_month'))
res.append(group_res)
return res
def match_week(s: str, now: datetime) -> List[Dict[str, Any]]:
groups = re.findall(R_WEEK, s)
res = []
for group in groups:
date_parts = {'match': group, 'date_parts': []}
if 'ez' in group:
y, w = now.isocalendar()[0:2]
date_parts['date_parts'].extend([Year(y, 'week'), Week(w, 'week')])
elif 'jovo' in remove_accent(group):
y, w = (now + timedelta(days=7)).isocalendar()[0:2]
date_parts['date_parts'].extend([Year(y, 'week'), Week(w, 'week')])
elif 'mult' in remove_accent(group) or 'elozo' in remove_accent(group):
y, w = (now - timedelta(days=7)).isocalendar()[0:2]
date_parts['date_parts'].extend([Year(y, 'week'), Week(w, 'week')])
res.append(date_parts)
return res
def match_n_periods_compared_to_now(s: str, now: datetime) -> List[Dict[str, Any]]:
# TODO: implement n-periods-before-now functionality
fn = 'n_date_periods_compared_to_now'
regexes = [
(R_NWEEKS_FROM_NOW, 'w'),
(R_NDAYS_FROM_NOW, 'd'),
(R_NHOURS_FROM_NOW, 'h'),
(R_NMINS_FROM_NOW, 'm')
]
res = []
for regex, freq in regexes:
groups = re.findall(regex, s)
for group in groups:
date_parts = {'match': group, 'date_parts': []}
n = group[1]
if n:
n = word_to_num(n)
if freq == 'w':
res_dt = (now + timedelta(days=7*n))
y, m, d = res_dt.year, res_dt.month, res_dt.day
date_parts['date_parts'].extend([Year(y, fn), Month(m, fn), Day(d, fn)])
elif freq == 'd':
res_dt = (now + timedelta(days=n))
y, m, d = res_dt.year, res_dt.month, res_dt.day
date_parts['date_parts'].extend([Year(y, fn), Month(m, fn), Day(d, fn)])
elif freq == 'h':
res_dt = (now + timedelta(hours=n))
y, m, d, h = res_dt.year, res_dt.month, res_dt.day, res_dt.hour
date_parts['date_parts'].extend([Year(y, fn), Month(m, fn), Day(d, fn), Hour(h, fn)])
elif freq == 'm':
res_dt = (now + timedelta(minutes=n))
y, mo, d, h, mi = res_dt.year, res_dt.month, res_dt.day, res_dt.hour, res_dt.minute
date_parts['date_parts'].extend([Year(y, fn), Month(mo, fn),
Day(d, fn), Hour(h, fn), Minute(mi, fn)])
res.append(date_parts)
return res
def match_named_year(s: str, now: datetime) -> List[Dict[str, Any]]:
groups = re.findall(R_YEAR, s)
res = []
for group in groups:
date_parts = {'match': group, 'date_parts': []}
if 'tavalyelott' in remove_accent(group):
date_parts['date_parts'] = [Year(now.year - 2, 'named_year')]
elif 'tavaly' in remove_accent(group):
date_parts['date_parts'] = [Year(now.year - 1, 'named_year')]
elif 'iden' in remove_accent(group):
date_parts['date_parts'] = [Year(now.year, 'named_year')]
elif 'jovo' in remove_accent(group):
date_parts['date_parts'] = [Year(now.year + 1, 'named_year')]
elif 'mulva' in remove_accent(group):
num_after = word_to_num(group)
if num_after != -1:
date_parts['date_parts'] = [Year(now.year + num_after, 'named_year')]
elif 'ezelott' in remove_accent(group) or 'korabban' in remove_accent(group):
num_before = word_to_num(group)
if num_before != -1:
date_parts['date_parts'] = [Year(now.year - num_before, 'named_year')]
res.append(date_parts)
return res
def match_weekday(s: str, now: datetime, expect_future_day: bool = False) -> List[Dict[str, Any]]:
groups = re.findall(R_WEEKDAY, s)
res = []
for group in groups:
date_parts = {'match': group, 'date_parts': []}
week, day = group
n_weeks = 0
if 'jovo' in remove_accent(week):
n_weeks = 1
elif 'mult' in remove_accent(week) or 'elozo' in remove_accent(week):
n_weeks = -1
def offset_date(date):
if date.strftime("%Y-%m-%d") < now.strftime("%Y-%m-%d"):
return date + timedelta(days=7)
return date
def get_day_of_week(w, d):
return ((now - timedelta(days=now.weekday())) + timedelta(days=w * 7)) + timedelta(days=d)
if 'hetfo' in remove_accent(day):
day = offset_date(get_day_of_week(n_weeks, 0)) if expect_future_day and n_weeks >= 0\
else get_day_of_week(n_weeks, 0)
date_parts['date_parts'] = [Year(day.year, 'weekday'), Month(day.month, 'weekday'), Day(day.day, 'weekday')]
elif 'kedd' in remove_accent(day):
day = offset_date(get_day_of_week(n_weeks, 1)) if expect_future_day and n_weeks >= 0 \
else get_day_of_week(n_weeks, 1)
date_parts['date_parts'] = [Year(day.year, 'weekday'), Month(day.month, 'weekday'), Day(day.day, 'weekday')]
elif 'szerda' in remove_accent(day):
day = offset_date(get_day_of_week(n_weeks, 2)) if expect_future_day and n_weeks >= 0 \
else get_day_of_week(n_weeks, 2)
date_parts['date_parts'] = [Year(day.year, 'weekday'), Month(day.month, 'weekday'), Day(day.day, 'weekday')]
elif 'csut' in remove_accent(day):
day = offset_date(get_day_of_week(n_weeks, 3)) if expect_future_day and n_weeks >= 0 \
else get_day_of_week(n_weeks, 3)
date_parts['date_parts'] = [Year(day.year, 'weekday'), Month(day.month, 'weekday'), Day(day.day, 'weekday')]
elif 'pent' in remove_accent(day):
day = offset_date(get_day_of_week(n_weeks, 4)) if expect_future_day and n_weeks >= 0 \
else get_day_of_week(n_weeks, 4)
date_parts['date_parts'] = [Year(day.year, 'weekday'), Month(day.month, 'weekday'), Day(day.day, 'weekday')]
elif 'szom' in remove_accent(day):
day = offset_date(get_day_of_week(n_weeks, 5)) if expect_future_day and n_weeks >= 0 \
else get_day_of_week(n_weeks, 5)
date_parts['date_parts'] = [Year(day.year, 'weekday'), Month(day.month, 'weekday'), Day(day.day, 'weekday')]
elif 'vas' in remove_accent(day):
day = offset_date(get_day_of_week(n_weeks, 6)) if expect_future_day and n_weeks >= 0 \
else get_day_of_week(n_weeks, 6)
date_parts['date_parts'] = [Year(day.year, 'weekday'), Month(day.month, 'weekday'), Day(day.day, 'weekday')]
res.append(date_parts)
return res
def match_relative_day(s: str, now: datetime) -> List[Dict[str, Any]]:
groups = [*re.findall(R_TODAY, s),
*re.findall(R_TOMORROW, s),
*re.findall(R_NTOMORROW, s),
*re.findall(R_YESTERDAY, s),
*re.findall(R_NYESTERDAY, s)]
res = []
for group in groups:
if type(group) != str:
group = [m for m in group if m][0]
if 'ma' in group or 'má' in group:
res.append({'match': group, 'date_parts': [Year(now.year, 'relative_day'), Month(now.month, 'relative_day'),
Day(now.day, 'relative_day')]})
elif 'holnapu' in group:
tom2 = now + timedelta(days=2)
res.append({'match': group,
'date_parts': [Year(tom2.year, 'relative_day'), Month(tom2.month, 'relative_day'),
Day(tom2.day, 'relative_day')]})
elif 'holnap' in group:
tom = now + timedelta(days=1)
res.append({'match': group, 'date_parts': [Year(tom.year, 'relative_day'), Month(tom.month, 'relative_day'),
Day(tom.day, 'relative_day')]})
elif 'tegnapel' in group:
yes2 = now - timedelta(days=2)
res.append({'match': group,
'date_parts': [Year(yes2.year, 'relative_day'), Month(yes2.month, 'relative_day'),
Day(yes2.day, 'relative_day')]})
elif 'tegnap' in group:
yes = now - timedelta(days=1)
res.append({'match': group, 'date_parts': [Year(yes.year, 'relative_day'), Month(yes.month, 'relative_day'),
Day(yes.day, 'relative_day')]})
return res
def match_iso_date(s: str) -> List[Dict[str, Any]]:
"""
Match ISO date-like format.
:param s: textual input
:return: tuple of date parts
"""
match = re.findall(R_ISO_DATE, s)
res = []
if match:
for group in match:
group = [int(m.lstrip('0')) for m in group if m.lstrip('0')]
if len(group) == 1:
res.append({'match': group, 'date_parts': [Year(group[0], 'match_iso_date')]})
elif len(group) == 2:
res.append({'match': group,
'date_parts': [Year(group[0], 'match_iso_date'), Month(group[1], 'match_iso_date')]})
elif len(group) == 3:
res.append({'match': group,
'date_parts': [Year(group[0], 'match_iso_date'), Month(group[1], 'match_iso_date'),
Day(group[2], 'match_iso_date')]})
return res
def match_now(s: str, now: datetime) -> List[Dict[str, Any]]:
if match_weekday(s, now):
return []
match = re.match(r'.*\bmost\b.*', s.lower())
if match:
date_parts = [
Year(now.year, 'now'),
Month(now.month, 'now'),
Day(now.day, 'now'),
Hour(now.hour, 'now'),
Minute(now.minute, 'now')
]
return [{'match': 'most', 'date_parts': date_parts}]
return []
def test_match_now():
now = datetime(2020, 12, 30, 12, 1)
fn = 'now'
tf = [('kedden 8:45', []), ('most kedden', []),
('legyen most', [[
Year(now.year, fn),
Month(now.month, fn),
Day(now.day, fn),
Hour(now.hour, fn),
Minute(now.minute, fn)
]])]
for inp, exp in tf:
out = match_now(inp, now)
date_parts = []
for e in out:
date_parts.append(e['date_parts'])
assert date_parts == exp