Fixing minor CSV bug
parent
19eb0471bb
commit
6215c38676
|
@ -7,9 +7,6 @@ IN: csv.tests
|
||||||
: named-unit-test ( name output input -- )
|
: named-unit-test ( name output input -- )
|
||||||
unit-test drop ; inline
|
unit-test drop ; inline
|
||||||
|
|
||||||
! tests nicked from the wikipedia csv article
|
|
||||||
! http://en.wikipedia.org/wiki/Comma-separated_values
|
|
||||||
|
|
||||||
"Fields are separated by commas"
|
"Fields are separated by commas"
|
||||||
[ { { "1997" "Ford" "E350" } } ]
|
[ { { "1997" "Ford" "E350" } } ]
|
||||||
[ "1997,Ford,E350" <string-reader> csv ] named-unit-test
|
[ "1997,Ford,E350" <string-reader> csv ] named-unit-test
|
||||||
|
@ -90,3 +87,5 @@ IN: csv.tests
|
||||||
{ { "writing,some,csv,tests" } } dup "csv-test2-"
|
{ { "writing,some,csv,tests" } } dup "csv-test2-"
|
||||||
unique-file utf8 [ csv>file ] [ file>csv ] 2bi =
|
unique-file utf8 [ csv>file ] [ file>csv ] 2bi =
|
||||||
] unit-test
|
] unit-test
|
||||||
|
|
||||||
|
[ { { "hello" "" "" "" "goodbye" "" } } ] [ "hello,,\"\",,goodbye," <string-reader> csv ] unit-test
|
||||||
|
|
|
@ -46,13 +46,15 @@ DEFER: quoted-field ( -- endchar )
|
||||||
|
|
||||||
: (row) ( -- sep )
|
: (row) ( -- sep )
|
||||||
field ,
|
field ,
|
||||||
dup delimiter get = [ drop (row) ] when ;
|
dup delimiter> = [ drop (row) ] when ;
|
||||||
|
|
||||||
: row ( -- eof? array[string] )
|
: row ( -- eof? array[string] )
|
||||||
[ (row) ] { } make ;
|
[ (row) ] { } make ;
|
||||||
|
|
||||||
: (csv) ( -- )
|
: (csv) ( -- )
|
||||||
row harvest [ , ] unless-empty [ (csv) ] when ;
|
row
|
||||||
|
dup [ empty? ] all? [ drop ] [ , ] if
|
||||||
|
[ (csv) ] when ;
|
||||||
|
|
||||||
PRIVATE>
|
PRIVATE>
|
||||||
|
|
||||||
|
@ -60,7 +62,8 @@ PRIVATE>
|
||||||
[ row nip ] with-input-stream ;
|
[ row nip ] with-input-stream ;
|
||||||
|
|
||||||
: csv ( stream -- rows )
|
: csv ( stream -- rows )
|
||||||
[ [ (csv) ] { } make ] with-input-stream ;
|
[ [ (csv) ] { } make ] with-input-stream
|
||||||
|
dup peek { "" } = [ but-last ] when ;
|
||||||
|
|
||||||
: file>csv ( path encoding -- csv )
|
: file>csv ( path encoding -- csv )
|
||||||
<file-reader> csv ;
|
<file-reader> csv ;
|
||||||
|
|
Loading…
Reference in New Issue