Merge pull request #26 from reworkd/date-and-datetime-types

Date validation

harambe/parser/parser.py

@@ -2,8 +2,9 @@
 from pydantic import BaseModel, create_model, Field, Extra, ValidationError
 from typing import Any, Dict, List, Optional, Type
 
-from harambe.types import Schema, URL
+from harambe.parser.type_date import ParserTypeDate
 from harambe.parser.type_url import ParserTypeUrl
+from harambe.types import Schema, URL
 
 
 OBJECT_TYPE = "object"
@@ -48,7 +49,7 @@ def __init__(self, schema: Schema):
             "double": float,
             LIST_TYPE: List,
             OBJECT_TYPE: Dict[str, Any],
-            # TODO: Add support for date and datetime types
+            "date": ParserTypeDate(),
             "url": ParserTypeUrl(),
         }
 

harambe/parser/type_date.py

@@ -0,0 +1,29 @@
+from pydantic.functional_validators import AfterValidator
+from typing_extensions import Annotated
+
+import dateparser
+
+
+class ParserTypeDate:
+    def __new__(self):
+        return Annotated[str, AfterValidator(self.validate)]
+
+    def validate(date: str):
+        if not isinstance(date, str):
+            raise ValueError("Wrong input type")
+
+        # Trim whitespaces
+        date = date.strip()
+
+        # Make sure it's not empty string
+        if len(date) == 0:
+            raise ValueError("Empty input")
+
+        # Attempt to parse date string
+        try:
+            dateparser.parse(date)
+            return date
+        except ValueError:
+            pass
+
+        raise ValueError(f"Unable to parse input as date: {date}")