feat(python): add add_custom_domains

FGRibreau · Nov 9, 2021 · 4b668de · 4b668de
1 parent c3245c5
commit 4b668de
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 0 deletions.
diff --git a/platform/python/MailChecker.py b/platform/python/MailChecker.py
@@ -34,3 +34,7 @@ def is_blacklisted(cls, email):
     @classmethod
     def is_valid_email_format(cls, email):
         return bool(email) and cls.valid_matcher.search(email) is not None
+
+    @classmethod
+    def add_custom_domains(cls, domains = []):
+        cls.blacklist.update(domains)
diff --git a/platform/python/MailChecker.tmpl.py b/platform/python/MailChecker.tmpl.py
@@ -34,3 +34,7 @@ def is_blacklisted(cls, email):
     @classmethod
     def is_valid_email_format(cls, email):
         return bool(email) and cls.valid_matcher.search(email) is not None
+
+    @classmethod
+    def add_custom_domains(cls, domains = []):
+        cls.blacklist.update(domains)
diff --git a/test/platform.python.test.py b/test/platform.python.test.py
@@ -55,5 +55,16 @@ def test_return_false_for_blacklisted_domains_and_their_subdomains(self):
       # Should not be invalid as a subdomain of a valid domain.
       self.valid("test@%s.gmail.com" % blacklisted_domain)
 
+  def test_add_custom_domains(self):
+    self.valid('foo@youtube.com')
+    self.valid('foo@google.com')
+    self.valid('ok@gmail.com')
+
+    MailChecker.add_custom_domains(['youtube.com', 'google.com'])
+
+    self.invalid('foo@youtube.com')
+    self.invalid('foo@google.com')
+    self.valid('ok@gmail.com')
+
 if __name__ == '__main__':
     unittest.main()