import acl
import bootstrap
-import datetime_utils
-import decorator_utils
+from datetime_utils import (
+ TimeUnit, n_timeunits_from_base, datetime_to_date, date_to_datetime
+)
from dateparse.dateparse_utilsLexer import dateparse_utilsLexer # type: ignore
from dateparse.dateparse_utilsListener import dateparse_utilsListener # type: ignore
from dateparse.dateparse_utilsParser import dateparse_utilsParser # type: ignore
+from decorator_utils import decorate_matching_methods_with
logger = logging.getLogger(__name__)
pass
-@decorator_utils.decorate_matching_methods_with(
+@decorate_matching_methods_with(
debug_parse,
acl=acl.StringWildcardBasedACL(
allowed_patterns=[
'exit*',
],
denied_patterns=None,
- order_to_check_allow_deny=acl.ACL_ORDER_DENY_ALLOW,
+ order_to_check_allow_deny=acl.Order.DENY_ALLOW,
default_answer=False
)
)
# These TimeUnits are defined in datetime_utils and are used as params
# to datetime_utils.n_timeunits_from_base.
self.time_delta_unit_to_constant = {
- 'hou': datetime_utils.TimeUnit.HOURS,
- 'min': datetime_utils.TimeUnit.MINUTES,
- 'sec': datetime_utils.TimeUnit.SECONDS,
+ 'hou': TimeUnit.HOURS,
+ 'min': TimeUnit.MINUTES,
+ 'sec': TimeUnit.SECONDS,
}
self.delta_unit_to_constant = {
- 'day': datetime_utils.TimeUnit.DAYS,
- 'wor': datetime_utils.TimeUnit.WORKDAYS,
- 'wee': datetime_utils.TimeUnit.WEEKS,
- 'mon': datetime_utils.TimeUnit.MONTHS,
- 'yea': datetime_utils.TimeUnit.YEARS,
+ 'day': TimeUnit.DAYS,
+ 'wor': TimeUnit.WORKDAYS,
+ 'wee': TimeUnit.WEEKS,
+ 'mon': TimeUnit.MONTHS,
+ 'yea': TimeUnit.YEARS,
}
self.override_now_for_test_purposes = override_now_for_test_purposes
self._reset()
self.today = datetime.date.today()
else:
self.now_datetime = self.override_now_for_test_purposes
- self.today = datetime_utils.datetime_to_date(
+ self.today = datetime_to_date(
self.override_now_for_test_purposes
)
self.date: Optional[datetime.date] = None
name = name.replace('washi', 'presi')
return name
- def _figure_out_date_unit(self, orig: str) -> int:
+ def _figure_out_date_unit(self, orig: str) -> TimeUnit:
"""Figure out what unit a date expression piece is talking about."""
if 'month' in orig:
- return datetime_utils.TimeUnit.MONTHS
+ return TimeUnit.MONTHS
txt = orig.lower()[:3]
if txt in self.day_name_to_number:
return(self.day_name_to_number[txt])
if 'delta_unit' not in self.context:
raise ParseException('Missing delta_unit?!')
unit = self.context['delta_unit']
- dt = datetime_utils.n_timeunits_from_base(
+ dt = n_timeunits_from_base(
count,
unit,
- datetime_utils.date_to_datetime(self.date)
+ date_to_datetime(self.date)
)
- self.date = datetime_utils.datetime_to_date(dt)
+ self.date = datetime_to_date(dt)
def exitTimeExpr(self, ctx: dateparse_utilsParser.TimeExprContext) -> None:
# Simple time?
self.timedelta += datetime.timedelta(minutes=count)
else:
unit = self.context['time_delta_unit']
- if unit == datetime_utils.TimeUnit.SECONDS:
+ if unit == TimeUnit.SECONDS:
self.timedelta += datetime.timedelta(seconds=count)
- elif unit == datetime_utils.TimeUnit.MINUTES:
+ elif unit == TimeUnit.MINUTES:
self.timedelta = datetime.timedelta(minutes=count)
- elif unit == datetime_utils.TimeUnit.HOURS:
+ elif unit == TimeUnit.HOURS:
self.timedelta = datetime.timedelta(hours=count)
else:
raise ParseException()
self.context['time_delta_int'] = 15
self.context[
'time_delta_unit'
- ] = datetime_utils.TimeUnit.MINUTES
+ ] = TimeUnit.MINUTES
elif txt == 'half':
self.context['time_delta_int'] = 30
self.context[
'time_delta_unit'
- ] = datetime_utils.TimeUnit.MINUTES
+ ] = TimeUnit.MINUTES
else:
raise ParseException(f'Bad time fraction {ctx.getText()}')
except:
count = -count
unit = self._figure_out_date_unit(unit)
- d = datetime_utils.n_timeunits_from_base(
+ d = n_timeunits_from_base(
count,
unit,
d)
f'Bad DeltaRelativeToTodayExpr: {ctx.getText()}'
)
unit = self._figure_out_date_unit(unit)
- d = datetime_utils.n_timeunits_from_base(
+ d = n_timeunits_from_base(
count,
unit,
d)
pass
def main() -> None:
parser = DateParser()
for line in sys.stdin:
if __name__ == "__main__":
- main = bootstrap.initialize(main)
main()