From 8c44164415f64e5f10ae5b022c84f542a74cdef1 Mon Sep 17 00:00:00 2001
From: Étienne Loks <etienne.loks@iggdrasil.net>
Date: Sun, 18 Feb 2018 15:50:43 +0100
Subject: Importer: INSEE code formater

---
 ishtar_common/data_importer.py  | 29 +++++++++++++++++++++++++++++
 ishtar_common/models_imports.py |  4 +++-
 2 files changed, 32 insertions(+), 1 deletion(-)

diff --git a/ishtar_common/data_importer.py b/ishtar_common/data_importer.py
index b2dd0d5a0..ffe6c221d 100644
--- a/ishtar_common/data_importer.py
+++ b/ishtar_common/data_importer.py
@@ -269,6 +269,35 @@ class FloatFormater(Formater):
                 'value': value})
 
 
+class InseeFormater(Formater):
+    """
+    Formater for "code INSEE" (statistic institute ine France)
+    The syntax "CodeINSEE-Year" is accepted (Ishtar trick) in order to manage
+    old INSEE (year is the date of creation)
+    """
+    ERROR = _(u"\"{value}\" is not an appropriate INSEE code")
+
+    def format(self, value):
+        value = value.strip()
+        exp = value.split('-')
+        code = exp[0]
+        try:
+            int(code)
+        except ValueError:
+            raise ValueError(unicode(self.ERROR).format(value))
+        while len(code) < 5:
+            code = "0" + code
+        if len(exp) > 2:
+            raise ValueError(unicode(self.ERROR).format(value))
+        elif len(exp) == 1:
+            return code
+        try:
+            datetime.datetime.strptime(exp[1], '%Y')
+        except ValueError:
+            raise ValueError(unicode(self.ERROR).format(value))
+        return code + u"-" + exp[1]
+
+
 class YearFormater(Formater):
     def format(self, value):
         value = value.strip()
diff --git a/ishtar_common/models_imports.py b/ishtar_common/models_imports.py
index cd868cc4a..4a5654781 100644
--- a/ishtar_common/models_imports.py
+++ b/ishtar_common/models_imports.py
@@ -42,7 +42,7 @@ from ishtar_common.utils import create_slug, \
     get_all_related_m2m_objects_with_model
 from ishtar_common.data_importer import Importer, ImportFormater, \
     IntegerFormater, FloatFormater, UnicodeFormater, DateFormater, \
-    TypeFormater, YearFormater, StrToBoolean, FileFormater
+    TypeFormater, YearFormater, StrToBoolean, FileFormater, InseeFormater
 
 logger = logging.getLogger(__name__)
 
@@ -641,6 +641,7 @@ IMPORTER_TYPES = (
     ('DateFormater', _(u"Date")),
     ('TypeFormater', _(u"Type")),
     ('YearFormater', _(u"Year")),
+    ('InseeFormater', _(u"INSEE code")),
     ('StrToBoolean', _(u"String to boolean")),
     ('FileFormater', pgettext_lazy("filesystem", u"File")),
     ('UnknowType', _(u"Unknow type"))
@@ -655,6 +656,7 @@ IMPORTER_TYPES_DCT = {
     'YearFormater': YearFormater,
     'StrToBoolean': StrToBoolean,
     'FileFormater': FileFormater,
+    'InseeFormater': InseeFormater,
     'UnknowType': None,
 }
 
-- 
cgit v1.2.3