adding a small switch for detecting just the street number

Ylopo · Mar 17, 2021 · dc2309e · dc2309e
1 parent 34e3373
commit dc2309e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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(