diff --git a/sqlalchemy_utils/types/color.py b/sqlalchemy_utils/types/color.py index 6f77816..689ce9a 100644 --- a/sqlalchemy_utils/types/color.py +++ b/sqlalchemy_utils/types/color.py @@ -24,7 +24,8 @@ class ColorType(types.TypeDecorator): # Bail if colour is not found. if colour is None: raise ImproperlyConfigured( - "'colour' is required to use 'ColorType'") + "'colour' package is required to use 'ColorType'" + ) super(ColorType, self).__init__(*args, **kwargs) self.impl = types.Unicode(max_length) diff --git a/sqlalchemy_utils/types/ip_address.py b/sqlalchemy_utils/types/ip_address.py index 457d2ca..8065762 100644 --- a/sqlalchemy_utils/types/ip_address.py +++ b/sqlalchemy_utils/types/ip_address.py @@ -1,6 +1,12 @@ import six -import ipaddress + +ipaddress = None +try: + import ipaddress +except: + pass from sqlalchemy import types +from sqlalchemy_utils import ImproperlyConfigured class IPAddressType(types.TypeDecorator): @@ -11,6 +17,11 @@ class IPAddressType(types.TypeDecorator): impl = types.Unicode(50) def __init__(self, max_length=50, *args, **kwargs): + if not ipaddress: + raise ImproperlyConfigured( + "'ipaddress' package is required to use 'IPAddressType'" + ) + super(IPAddressType, self).__init__(*args, **kwargs) self.impl = types.Unicode(max_length)