diff options
Diffstat (limited to 'ishtar_common/management/commands/import_insee_comm_csv.py')
| -rw-r--r-- | ishtar_common/management/commands/import_insee_comm_csv.py | 14 |
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(): |