mirrors/limbo
1
0
Fork 0
mirror of https://github.com/tursodatabase/limbo.git synced 2025-07-07 12:35:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 9

Fix import csv failing when single quote is in string

Browse source
This commit is contained in:
psvri 2025-01-05 17:34:56 +05:30
parent 3bc86a3cfc
commit ce8600a695
3 changed files with 5 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

3
cli/import.rs
View file

@ -81,7 +81,8 @@ impl<'a> ImportFile<'a> {
for r in record.iter() {
values_string.push('\'');
values_string.push_str(r);
// The string can have a single quote which needs to be escaped
values_string.push_str(&r.replace("'", "''"));
values_string.push_str("',");
}

4
testing/shelltests.py
View file

@ -253,10 +253,10 @@ def test_import_csv(test_name: str, options: str, import_output: str, table_outp
table_output,
)
test_import_csv('no_options', '--csv', '', '1|2.0|String1\n3|4.0|String2')
test_import_csv('no_options', '--csv', '', '1|2.0|String\'1\n3|4.0|String2')
test_import_csv('verbose', '--csv -v',
'Added 2 rows with 0 errors using 2 lines of input'
,'1|2.0|String1\n3|4.0|String2')
,'1|2.0|String\'1\n3|4.0|String2')
test_import_csv('skip', '--csv --skip 1', '' ,'3|4.0|String2')

2
testing/test_files/test.csv
View file

@ -1,2 +1,2 @@
1,2.0,"String1"
1,2.0,"String'1"
3,4.0,"String2"
1 1 2.0 String1 String'1
2 3 4.0 String2 String2