From 14ab1c7177f451fb944a06281d5ac40e66268241 Mon Sep 17 00:00:00 2001 From: Phil Dawes Date: Tue, 5 Aug 2008 10:36:13 +0100 Subject: [PATCH] Ammended csv to use a global var for default delimiter. Also removed dependency on vars --- extra/csv/csv.factor | 24 ++++++++++-------------- 1 file changed, 10 insertions(+), 14 deletions(-) diff --git a/extra/csv/csv.factor b/extra/csv/csv.factor index 3d1fb64492..9779c37ed4 100644 --- a/extra/csv/csv.factor +++ b/extra/csv/csv.factor @@ -4,12 +4,16 @@ ! Simple CSV Parser ! Phil Dawes phil@phildawes.net -USING: kernel sequences io namespaces combinators unicode.categories vars ; +USING: kernel sequences io namespaces combinators unicode.categories ; IN: csv -DEFER: quoted-field +SYMBOL: delimiter -VAR: delimiter +CHAR: , delimiter set-global + +: delimiter> delimiter get ; inline + +DEFER: quoted-field ( -- endchar ) ! trims whitespace from either end of string : trim-whitespace ( str -- str ) @@ -44,7 +48,7 @@ VAR: delimiter : (row) ( -- sep ) field , - dup delimiter> = [ drop (row) ] when ; + dup delimiter get = [ drop (row) ] when ; : row ( -- eof? array[string] ) [ (row) ] { } make ; @@ -55,25 +59,18 @@ VAR: delimiter : (csv) ( -- ) row append-if-row-not-empty [ (csv) ] when ; - -: init-vars ( -- ) - delimiter> [ CHAR: , >delimiter ] unless ; inline : csv-row ( stream -- row ) - init-vars [ row nip ] with-input-stream ; : csv ( stream -- rows ) - init-vars [ [ (csv) ] { } make ] with-input-stream ; : with-delimiter ( char quot -- ) delimiter swap with-variable ; inline - - : needs-escaping? ( cell -- ? ) - [ "\n\"" delimiter> suffix member? ] contains? ; inline ! " + [ [ "\n\"" member? ] [ delimiter get = ] bi or ] contains? ; inline ! " : escape-quotes ( cell -- cell' ) [ [ dup , CHAR: " = [ CHAR: " , ] when ] each ] "" make ; inline @@ -85,8 +82,7 @@ VAR: delimiter dup needs-escaping? [ escape-quotes enclose-in-quotes ] when ; inline : write-row ( row -- ) - [ delimiter> write1 ] [ escape-if-required write ] interleave nl ; inline + [ delimiter get write1 ] [ escape-if-required write ] interleave nl ; inline : write-csv ( rows outstream -- ) - init-vars [ [ write-row ] each ] with-output-stream ;