Ammended csv to use a global var for default delimiter. Also removed dependency on vars
							parent
							
								
									45154c9a97
								
							
						
					
					
						commit
						14ab1c7177
					
				| 
						 | 
				
			
			@ -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 ;
 | 
			
		||||
| 
						 | 
				
			
			@ -56,24 +60,17 @@ VAR: delimiter
 | 
			
		|||
  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 ;
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
		Loading…
	
		Reference in New Issue