openstack/deb-python-sqlalchemy-utils
Code Issues Proposed changes

Add PhoneNumber python_type, refs #248

Browse Source
This commit is contained in:
Konsta Vesterinen
2016-10-20 09:55:19 +03:00
parent b1f84e5998
commit ec7e2fefec
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
sqlalchemy_utils/types/phone_number.py
View File

@@ -153,9 +153,7 @@ class PhoneNumberType(types.TypeDecorator, ScalarCoercible):
""" """
STORE_FORMAT = 'e164' STORE_FORMAT = 'e164'
impl = types.Unicode(20) impl = types.Unicode(20)
python_type = PhoneNumber
def python_type(self, text):
return self._coerce(text)
def __init__(self, region='US', max_length=20, *args, **kwargs): def __init__(self, region='US', max_length=20, *args, **kwargs):
# Bail if phonenumbers is not found. # Bail if phonenumbers is not found.

3
tests/types/test_phonenumber.py
View File

@@ -150,6 +150,9 @@ class TestPhoneNumberType(object):
session.refresh(user) session.refresh(user)
assert user.phone_number is None 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') @pytest.mark.usefixtures('user')
def test_phone_number_is_none(self, session, User): def test_phone_number_is_none(self, session, User):
phone_number = None phone_number = None