remove pydantic[email]

Author: VaghinakDev

requirements.txt

@@ -1,5 +1,4 @@
 dnspython==2.1.0
-email-validator==1.1.3
 fire==0.4.0
 idna==3.3
 pydantic==1.8.2

src/superannotate_schemas/schemas/base.py

@@ -6,7 +6,6 @@
 from pydantic import BaseModel as PyDanticBaseModel
 from pydantic import conlist
 from pydantic import constr
-from pydantic import EmailStr
 from pydantic import Extra
 from pydantic import StrictInt
 from pydantic import StrictFloat
@@ -46,6 +45,18 @@ def enum_error_handling(self) -> str:
 StrictNumber = Union[StrictInt, StrictFloat]
 
 
+class EmailStr(StrictStr):
+    @classmethod
+    def validate(cls, value: Union[str]) -> Union[str]:
+        try:
+            constr(
+                regex=r"^(?=.{1,254}$)(?=.{1,64}@)[a-zA-Z0-9!#$%&'*+/=?^_`{|}~-]+(?:\.[a-zA-Z0-9!#$%&'*+/=?^_`{|}~-]+)*@[a-zA-Z0-9](?:[a-zA-Z0-9-]{0,61}[a-zA-Z0-9])?(?:\.[a-zA-Z0-9](?:[a-zA-Z0-9-]{0,61}[a-zA-Z0-9])?)*$"
+            ).validate(value)
+        except StrRegexError:
+            raise ValueError("Invalid email")
+        return value
+
+
 class BaseModel(PyDanticBaseModel):
     class Config:
         extra = Extra.allow

tests/test_cli.py

@@ -2,7 +2,6 @@
 import os
 from os.path import dirname
 from tests import LIB_PATH
-from pydantic import EmailStr
 
 from unittest import TestCase