✨ Update IGN/INSEE import script

1 files changed, 11 insertions, 3 deletions

diff --git a/ishtar_common/management/commands/import_insee_comm_csv.py b/ishtar_common/management/commands/import_insee_comm_csv.py
index 7962efd37..c6fec3e96 100644
--- a/ishtar_common/management/commands/import_insee_comm_csv.py
+++ b/ishtar_common/management/commands/import_insee_comm_csv.py
@@ -55,6 +55,17 @@ class Command(BaseCommand):
         with open(csv_file, 'rt') as csvfile:
             reader = csv.DictReader(csvfile)
             for idx, row in enumerate(reader):
+                new_insee = row['DepComN']
+                if len(new_insee) < 5:
+                    new_insee = '0' + new_insee
+
+                if not idx:  # test if first do not exist
+                    q = Town.objects.filter(numero_insee=new_insee,
+                                            year=default_year)
+                    if q.count():
+                        print("First town already exists for this year....")
+                        return
+
                 if not quiet:
                     sys.stdout.write('Processing town %d.\r' % (idx + 1))
                     sys.stdout.flush()
@@ -74,9 +85,6 @@ class Command(BaseCommand):
                         continue
                 old_town = q.all()[0]
 
-                new_insee = row['DepComN']
-                if len(new_insee) < 5:
-                    new_insee = '0' + new_insee
                 q = Town.objects.filter(numero_insee=new_insee,
                                         year=default_year)
                 if not q.count():