Data importation: add an option to select the default choice (for testing)

1 files changed, 12 insertions, 1 deletions

diff --git a/archaeological_operations/management/commands/ishtar_imports.py b/archaeological_operations/management/commands/ishtar_imports.py
index 23397204b..3f4d9e2e8 100644
--- a/archaeological_operations/management/commands/ishtar_imports.py
+++ b/archaeological_operations/management/commands/ishtar_imports.py
@@ -18,6 +18,7 @@
 # See the file COPYING for details.
 
 import datetime, unicodecsv
+from optparse import make_option
 
 from django.conf import settings
 from django.core.management.base import BaseCommand, CommandError
@@ -54,6 +55,14 @@ except ImportError:
 class Command(BaseCommand):
     args = '<filename> <importer_name> [<nb lines skipped>]'
     help = "Import archaeological operations"
+    option_list = BaseCommand.option_list + (
+        make_option('--choose-default',
+            action='store_true',
+            dest='choose_default',
+            default=False,
+            help='When a choice is requested choose the first one available. '\
+                 'For testing purpose'),
+        )
 
     def handle(self, *args, **options):
         if not args or not args[0]:
@@ -68,6 +77,7 @@ class Command(BaseCommand):
             skip_lines = int(args[2])
         except:
             skip_lines = 0
+        choose_default = options.get('choose_default')
         filename = args[0]
         importer = IMPORTERS[args[1]](skip_lines=skip_lines, output='cli')
         sys.stdout.write("*" * 72 + "\n")
@@ -82,7 +92,8 @@ class Command(BaseCommand):
             for encoding in encodings:
                 try:
                     importer.importation([line for line in
-                            unicodecsv.reader(csv_file, encoding='utf-8')])
+                            unicodecsv.reader(csv_file, encoding='utf-8')],
+                            choose_default=choose_default)
                     errors = importer.get_csv_errors()
                     sys.stdout.write("\n")
                     if errors: