fixup! db: Implement basic import/export of data

sanitize table name
nextstrain · Dec 31, 2022 · 30c7bcc · 30c7bcc
1 parent 23a2c81
commit 30c7bcc
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/augur/db/import_/sqlite3.py b/augur/db/import_/sqlite3.py
@@ -6,7 +6,7 @@
 from augur.errors import AugurError
 from augur.io.metadata import Metadata
 from augur.io.sequences import Sequences
-from augur.io.sqlite3 import Sqlite3Database
+from augur.io.sqlite3 import Sqlite3Database, sanitize_identifier
 
 
 class Sqlite3ImportError(AugurError):
@@ -49,7 +49,7 @@ def import_metadata(metadata: Metadata, table_name: str, database: Sqlite3Databa
         raise Sqlite3ImportError(f'Failed to create table: {e}')
 
     insert_statement = f"""
-        INSERT INTO {table_name}
+        INSERT INTO {sanitize_identifier(table_name)}
         VALUES ({','.join(['?' for _ in metadata.names])})
     """
     rows = metadata.rows()
@@ -76,7 +76,7 @@ def import_sequences(sequences: Sequences, id_column: str, table_name: str, data
         raise Sqlite3ImportError(f'Failed to create table: {e}')
 
     insert_statement = f"""
-        INSERT INTO {table_name}
+        INSERT INTO {sanitize_identifier(table_name)}
         VALUES ({','.join(['?' for _ in columns])})
     """
     records = ((record.id, str(record.seq)) for record in sequences.get_seqrecords())