X-Git-Url: http://git.veekun.com/zzz-pokedex.git/blobdiff_plain/45c94cc0640ae7af1308d7546a09cbce8e37f589..2b3139dd945a4e1be0ae4d1131abd2bea496c1d8:/pokedex/__init__.py

diff --git a/pokedex/__init__.py b/pokedex/__init__.py
index 2be0f80..e4397af 100644
--- a/pokedex/__init__.py
+++ b/pokedex/__init__.py
@@ -122,6 +122,8 @@ def command_load(*args):
     parser = get_parser(verbose=True)
     parser.add_option('-d', '--directory', dest='directory', default=None)
     parser.add_option('-D', '--drop-tables', dest='drop_tables', default=False, action='store_true')
+    parser.add_option('-S', '--safe', dest='safe', default=False, action='store_true',
+        help="Do not use backend-specific optimalizations.")
     options, tables = parser.parse_args(list(args))
 
     if not options.engine_uri:
@@ -137,7 +139,8 @@ def command_load(*args):
     pokedex.db.load.load(session, directory=options.directory,
                                   drop_tables=options.drop_tables,
                                   tables=tables,
-                                  verbose=options.verbose)
+                                  verbose=options.verbose,
+                                  safe=options.safe)
 
 def command_reindex(*args):
     parser = get_parser(verbose=True)
@@ -158,7 +161,8 @@ def command_setup(*args):
     session = get_session(options)
     get_csv_directory(options)
     pokedex.db.load.load(session, directory=None, drop_tables=True,
-                                  verbose=options.verbose)
+                                  verbose=options.verbose,
+                                  safe=False)
 
     lookup = get_lookup(options, session=session, recreate=True)