diff --git a/pyap/parser.py b/pyap/parser.py index 39228c0..3c56d63 100644 --- a/pyap/parser.py +++ b/pyap/parser.py @@ -34,7 +34,10 @@ def __init__(self, **args): package = 'pyap' + '.source_' + self.country + \ '.data' data = importlib.import_module(package) - self.rules = data.full_address + if self.rule_name == 'full_street': + self.rules = data.full_street + else: + self.rules = data.full_address except AttributeError: raise e.NoCountrySelected(