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

Changed PhoneNumber string representation

Browse Source
This commit is contained in:
Vesa Uimonen
2013-03-27 14:41:45 +02:00
parent ffda0eac51
commit c03bcc4874
2 changed files with 11 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
sqlalchemy_utils/types.py
View File

@@ -49,6 +49,12 @@ class PhoneNumber(phonenumbers.phonenumber.PhoneNumber):
def is_valid_number(self): def is_valid_number(self):
return phonenumbers.is_valid_number(self._phone_number) return phonenumbers.is_valid_number(self._phone_number)
def __unicode__(self):
return self.national
def __str__(self):
return unicode(self).encode('utf-8')
class PhoneNumberType(types.TypeDecorator): class PhoneNumberType(types.TypeDecorator):
""" """

5
tests/test_phonenumber_type.py
View File

@@ -39,6 +39,11 @@ class TestPhoneNumber(object):
assert phone_number.international == u'+358 40 1234567' assert phone_number.international == u'+358 40 1234567'
assert phone_number.national == u'040 1234567' assert phone_number.national == u'040 1234567'
def test_phone_number_str_repr(self):
phone_number = PhoneNumber('+358401234567')
assert phone_number.__unicode__() == phone_number.national
assert phone_number.__str__() == phone_number.national.encode('utf-8')
class TestPhoneNumberType(TestCase): class TestPhoneNumberType(TestCase):
def setup_method(self, method): def setup_method(self, method):