Many fix on CSV imports - allow to import only slice of rows

1 files changed, 5 insertions, 3 deletions

diff --git a/archaeological_operations/management/commands/import_operations.py b/archaeological_operations/management/commands/import_operations.py
index 3cf4a569d..acc43a591 100755
--- a/archaeological_operations/management/commands/import_operations.py
+++ b/archaeological_operations/management/commands/import_operations.py
@@ -28,7 +28,7 @@ IMPORTERS = {'csv':import_from_csv,
              'vfp':import_from_dbf}
 
 class Command(BaseCommand):
-    args = '<filename> [<update> <csv|dbf>]'
+    args = '<filename> [<update> <csv|dbf> <lines>]'
     help = "Import archaelogical operations"
 
     def handle(self, *args, **options):
@@ -36,7 +36,8 @@ class Command(BaseCommand):
             raise CommandError("No file provided.")
         filename = args[0]
         update = len(args) > 1 and args[1]
-        file_type = len(args) > 1 and args[2]
+        file_type = len(args) > 2 and args[2]
+        lines = len(args) > 3 and args[3]
         if not file_type:
             suffix = filename.split('.')[-1].lower()
             if suffix in IMPORTERS.keys():
@@ -48,7 +49,8 @@ class Command(BaseCommand):
             raise CommandError("This file type is not managed.")
         nb_ops, errors = IMPORTERS[file_type](filename,
                                               update=update,
-                                              stdout=self.stdout)
+                                              stdout=self.stdout,
+                                              lines=lines)
         self.stdout.write('\n* %d operation treated\n' % nb_ops)
         if errors:
             self.stderr.write('\n'.join(errors))