fixed phone number format

1 files changed, 17 insertions, 3 deletions

diff --git a/lib/d2/bin/d2_data_merge.py b/lib/d2/bin/d2_data_merge.py
index 2270830..0af8e94 100644
--- a/lib/d2/bin/d2_data_merge.py
+++ b/lib/d2/bin/d2_data_merge.py
@@ -218,7 +218,7 @@ class DbOccupant(object):
 
 class BaseDb(object):
 
-    NUMBER = '216-889'
+    NUMBER = '216889'
 
     def __init__(self, config, static, detail_adapter):
         self._config = config
@@ -258,8 +258,22 @@ class BaseDb(object):
         raise NotImplementedError()
 
     def _fix_phone_number(self, data):
-        if len(data) == 4:
-            return u"{0}-{1}".format(self.NUMBER, data)
+        numbers = []
+        for c in data:
+            if c.isdigit():
+                numbers.append(c)
+        if len(numbers) < 5:
+            for x, n in enumerate(self.NUMBER):
+                numbers.insert(x, n)
+        return self._add_dashes(numbers)
+
+    def _add_dashes(self, numbers_list):
+        out = []
+        for x, number in enumerate(numbers_list):
+            if x in (3, 6):
+                out.append(u'-')
+            out.append(unicode(number))
+        return u''.join(out)
 
 
 class DbFirstName(BaseDb):