From ad75d1938cacead9b1f9906eb2c8495fa1ac1e14 Mon Sep 17 00:00:00 2001 From: Étienne Loks Date: Mon, 5 Feb 2024 10:32:32 +0100 Subject: ✨ Update IGN/INSEE import script MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ishtar_common/management/commands/import_insee_comm_csv.py | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) (limited to 'ishtar_common/management/commands/import_insee_comm_csv.py') 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(): -- cgit v1.2.3