datetime parser for humans

Author: bharathgs

dateandtimeparser/datetime_parser.py

@@ -1,30 +1,46 @@
+import attr
+from .loggers import logger
 from collections import namedtuple
 from .errors import DateParserException
 from .core_date_parser import DateParser
-from .loggers import logger
 from .core_datetime_formats import DateTimeFormats
+from ._validators import _datetime_format_type_validator, _end_year_validator, _mode_type_validator
 
 
+@attr.s(slots=True)
 class DateTimeParser:
     """
-    DateTime Adaptor class for DateParser
+    Adaptor class for CoreDateParser
     """
+    text = attr.ib(validator=attr.validators.instance_of(str))
+    start_year = attr.ib(validator=attr.validators.instance_of(int))
+    end_year = attr.ib(validator=attr.validators.instance_of(int))
+    mode = attr.ib(default="date", validator=[
+                   _mode_type_validator, attr.validators.instance_of(str)])
+    datetime_format = attr.ib(
+        default=None, validator=_datetime_format_type_validator)
 
-    def __init__(self, text, start_year, end_year, datetime_format=None):
-        self.text = text
-        self.start_year = start_year
-        self.end_year = end_year
-        self._datetime_format = datetime_format
-        if self._datetime_format == None:
-            self._datetime_format = DateTimeFormats.FORMATS
+    def __attrs_post_init__(self):
+        object.__setattr__(self, "datetime_format",
+                           self.datetime_format_handler(self.datetime_format))
+        object.__setattr__(self, "_dateparser_formatter",
+                           self._format_datetime)
+        object.__setattr__(self, "datetime", self._parser(
+            self.text, self.start_year, self.end_year, self.datetime_format, self._dateparser_formatter))
+
+    @staticmethod
+    def datetime_format_handler(datetime_format):
+        if datetime_format == None:
+            return DateTimeFormats.FORMATS
         else:
-            self._datetime_format = DateTimeFormats.FORMATS+self._datetime_format
-        self._dateparser_formatter = self._format_datetime
-        self.datetime = self._parser(self.text, self.start_year, self.end_year,
-                                     self._datetime_format, self._dateparser_formatter)
+            return DateTimeFormats.FORMATS+datetime_format
 
-    def __repr__(self):
-        return f"DateTimeParser(start={self.start_year}, end={self.end_year}, format={self._datetime_format})"
+    @staticmethod
+    def _format_datetime(datetime_object):
+        _datetime = namedtuple(
+            "datetime", ["date", "token_span", "token_index", "format"])
+        return _datetime(datetime_object[0], (datetime_object[1], datetime_object[2]),
+                         (datetime_object[4], datetime_object[5]), datetime_object[3])
 
     @staticmethod
     def _parser(text, start_year, end_year, datetime_format, formatter):
@@ -35,14 +51,5 @@ def _parser(text, start_year, end_year, datetime_format, formatter):
             dt = DP.parse_string(text)
             _dt = [formatter(i) for i in dt]
             return _dt
-        except DateParserException as DPEX:
-            logger.info(f"ERROR: {DPEX}")
+        except DateParserException:
             return None
-
-    @staticmethod
-    def _format_datetime(datetime_object):
-        _datetime = namedtuple(
-            "datetime", ["date", "token_span", "token_index", "format"])
-        return _datetime(datetime_object[0],
-                         (datetime_object[1], datetime_object[2]),
-                         (datetime_object[4], datetime_object[5]), datetime_object[3])