Add PhoneNumber python_type, refs #248

kvesteri · Oct 20, 2016 · ec7e2fe · ec7e2fe
1 parent b1f84e5
commit ec7e2fe
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/sqlalchemy_utils/types/phone_number.py b/sqlalchemy_utils/types/phone_number.py
@@ -153,9 +153,7 @@ class User(self.Base):
     """
     STORE_FORMAT = 'e164'
     impl = types.Unicode(20)
-
-    def python_type(self, text):
-        return self._coerce(text)
+    python_type = PhoneNumber
 
     def __init__(self, region='US', max_length=20, *args, **kwargs):
         # Bail if phonenumbers is not found.

diff --git a/tests/types/test_phonenumber.py b/tests/types/test_phonenumber.py
@@ -150,6 +150,9 @@ def test_empty_phone_number_is_equiv_to_none(self, session, User):
         session.refresh(user)
         assert user.phone_number is None
 
+    def test_uses_phonenumber_class_as_python_type(self):
+        assert PhoneNumberType().python_type is PhoneNumber
+
     @pytest.mark.usefixtures('user')
     def test_phone_number_is_none(self, session, User):
         phone_number = None