X-Git-Url: http://git.veekun.com/zzz-pokedex.git/blobdiff_plain/6df90db0baf023ab59a21f1ca73b79fb04f9e22b..4334a1a641cdc77605d84d366ab2bc1b9b1ab484:/pokedex/db/load.py diff --git a/pokedex/db/load.py b/pokedex/db/load.py index 6a548b8..f0e4b6d 100644 --- a/pokedex/db/load.py +++ b/pokedex/db/load.py @@ -185,15 +185,15 @@ def load(session, tables=[], directory=None, drop_tables=False, verbose=False, s force_not_null = 'FORCE NOT NULL ' + ','.join('"%s"' % c for c in not_null_cols) else: force_not_null = '' - command = "COPY {table_name} ({columns}) FROM '{csvpath}' CSV HEADER {force_not_null}" + command = "COPY %(table_name)s (%(columns)s) FROM '%(csvpath)s' CSV HEADER %(force_not_null)s" session.connection().execute( - command.format( - table_name=table_name, - csvpath=csvpath, - columns=','.join('"%s"' % c for c in column_names), - force_not_null=force_not_null, - ) + command % dict( + table_name=table_name, + csvpath=csvpath, + columns=','.join('"%s"' % c for c in column_names), + force_not_null=force_not_null, ) + ) session.commit() print_done() continue