diff --git a/django_starfield/widgets.py b/django_starfield/widgets.py index 04abbf3a5aea4520b2bfd595d93dc13a64fb89bc..02f6d995bd904f2dd8843946461a1788af6e3da6 100644 --- a/django_starfield/widgets.py +++ b/django_starfield/widgets.py @@ -11,12 +11,9 @@ class Stars(Widget): def __init__(self, attrs=None, stars=None, codepoint=None, colour=None): super(Stars, self).__init__(attrs) - if stars is None: - self.stars = getattr(settings, 'STARFIELD_STARS', STARFIELD_STARS) - if codepoint is None: - self.codepoint = getattr(settings, 'STARFIELD_CODEPOINT', STARFIELD_CODEPOINT) - if colour is None: - self.colour = getattr(settings, 'STARFIELD_COLOUR', STARFIELD_COLOUR) + self.stars = stars if stars else getattr(settings, 'STARFIELD_STARS', STARFIELD_STARS) + self.codepoint = codepoint if codepoint else getattr(settings, 'STARFIELD_CODEPOINT', STARFIELD_CODEPOINT) + self.colour = colour if colour else getattr(settings, 'STARFIELD_COLOUR', STARFIELD_COLOUR) def get_context(self, name, value, attrs): context = super(Stars, self).get_context(name, value, attrs)