Compare commits
12
Commits
d27eb91952
..
dev
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
af4066d3db | ||
|
|
3bc1be9eb9 | ||
|
|
d99862b3f4 | ||
|
|
22b6942900 | ||
|
|
c1e9fb7b8e | ||
|
|
82085eaaf9 | ||
|
|
493b7dd9d4 | ||
|
|
723f046ea4 | ||
|
|
bb970e4f42 | ||
|
|
1b20188dfc | ||
|
|
886f991ee3 | ||
|
|
659b817252 |
@@ -1,3 +1,5 @@
|
|||||||
# Changelog for csv-sip
|
# Changelog for csv-sip
|
||||||
|
|
||||||
## Unreleased changes
|
## Unreleased changes
|
||||||
|
- changed the types of encodeCSV and encodeRawCSV to make them more generic
|
||||||
|
- slight re-structuring of documentation
|
||||||
|
|||||||
@@ -14,5 +14,8 @@ General Public License for more details.
|
|||||||
You should have received a copy of the GNU General Public License
|
You should have received a copy of the GNU General Public License
|
||||||
along with this program. If not, see <https://www.gnu.org/licenses/>.
|
along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
## Important Note
|
## Executive Summary
|
||||||
This library is not yet ready for release. As such, all code should be considered to be unstable and subject to change at any time.
|
This library allows for reading and writing to and from CSV files in a streaming manner. Files can be read and written to on a row-by-row basis allowing larger files to be worked with, since the whole file doesn't have to be loaded to manipulate it. It is based on the [conduit](https://hackage.haskell.org/package/conduit] library.
|
||||||
|
|
||||||
|
## Pull Requests
|
||||||
|
Please make pull requests to the `dev` branch.
|
||||||
|
|||||||
+5
-3
@@ -5,10 +5,12 @@ cabal-version: 2.2
|
|||||||
-- see: https://github.com/sol/hpack
|
-- see: https://github.com/sol/hpack
|
||||||
|
|
||||||
name: csv-sip
|
name: csv-sip
|
||||||
version: 0.0.0
|
version: 0.1.0
|
||||||
synopsis: extracts data from a CSV file
|
synopsis: CSV streaming library
|
||||||
description: extracts data from a CSV file - see README.md for more details
|
description: CSV streaming library - see README.md for more details
|
||||||
category: Data
|
category: Data
|
||||||
|
homepage: https://codeberg.org/jlamothe/csv-sip
|
||||||
|
bug-reports: https://codeberg.org/jlamothe/csv-sip/issues
|
||||||
author: Jonathan Lamothe
|
author: Jonathan Lamothe
|
||||||
maintainer: jonathan@jlamothe.net
|
maintainer: jonathan@jlamothe.net
|
||||||
copyright: (C) 2022 Jonathan Lamothe
|
copyright: (C) 2022 Jonathan Lamothe
|
||||||
|
|||||||
+5
-3
@@ -1,5 +1,5 @@
|
|||||||
name: csv-sip
|
name: csv-sip
|
||||||
version: 0.0.0
|
version: 0.1.0
|
||||||
license: GPL-3.0-or-later
|
license: GPL-3.0-or-later
|
||||||
author: "Jonathan Lamothe"
|
author: "Jonathan Lamothe"
|
||||||
maintainer: "jonathan@jlamothe.net"
|
maintainer: "jonathan@jlamothe.net"
|
||||||
@@ -10,13 +10,15 @@ extra-source-files:
|
|||||||
- ChangeLog.md
|
- ChangeLog.md
|
||||||
|
|
||||||
# Metadata used when publishing your package
|
# Metadata used when publishing your package
|
||||||
synopsis: extracts data from a CSV file
|
synopsis: CSV streaming library
|
||||||
category: Data
|
category: Data
|
||||||
|
|
||||||
# To avoid duplicated efforts in documentation and dealing with the
|
# To avoid duplicated efforts in documentation and dealing with the
|
||||||
# complications of embedding Haddock markup inside cabal files, it is
|
# complications of embedding Haddock markup inside cabal files, it is
|
||||||
# common to point users to the README.md file.
|
# common to point users to the README.md file.
|
||||||
description: extracts data from a CSV file - see README.md for more details
|
description: CSV streaming library - see README.md for more details
|
||||||
|
homepage: https://codeberg.org/jlamothe/csv-sip
|
||||||
|
bug-reports: https://codeberg.org/jlamothe/csv-sip/issues
|
||||||
|
|
||||||
ghc-options:
|
ghc-options:
|
||||||
- -Wall
|
- -Wall
|
||||||
|
|||||||
+46
-30
@@ -35,20 +35,26 @@ module Data.CSV.Sip (
|
|||||||
-- ** Write an entire CSV file
|
-- ** Write an entire CSV file
|
||||||
writeCSV,
|
writeCSV,
|
||||||
writeRawCSV,
|
writeRawCSV,
|
||||||
writeCSVFromStream,
|
|
||||||
writeRawCSVFromStream,
|
|
||||||
-- * Conduits
|
-- * Conduits
|
||||||
-- ** Encoding
|
-- ** Producers
|
||||||
|
readFromCSV,
|
||||||
|
readFromCSVRaw,
|
||||||
encodeCSV,
|
encodeCSV,
|
||||||
encodeRawCSV,
|
encodeRawCSV,
|
||||||
|
-- ** Consumers
|
||||||
|
writeToCSV,
|
||||||
|
writeToCSVRaw,
|
||||||
|
-- ** Transformers
|
||||||
|
-- *** Encoding
|
||||||
encodeRows,
|
encodeRows,
|
||||||
encodeRawRows,
|
encodeRawRows,
|
||||||
-- ** Decoding
|
-- *** Decoding
|
||||||
labelFields,
|
labelFields,
|
||||||
decodeRows,
|
decodeRows,
|
||||||
decodeRawRows,
|
decodeRawRows,
|
||||||
decodeUTF8,
|
|
||||||
toBytes,
|
toBytes,
|
||||||
|
-- * Helper Functions
|
||||||
|
decodeUTF8,
|
||||||
) where
|
) where
|
||||||
|
|
||||||
import Conduit
|
import Conduit
|
||||||
@@ -127,34 +133,28 @@ writeRawCSV
|
|||||||
-> m ()
|
-> m ()
|
||||||
writeRawCSV file csv = runConduit $ encodeRawCSV csv .| sinkFile file
|
writeRawCSV file csv = runConduit $ encodeRawCSV csv .| sinkFile file
|
||||||
|
|
||||||
-- | Write a CSV file from a stream of Text-based rows
|
-- | reads a stream of Text-based rows from a CSV file
|
||||||
writeCSVFromStream
|
readFromCSV
|
||||||
:: MonadResource m
|
:: MonadResource m
|
||||||
=> FilePath
|
=> FilePath
|
||||||
-- ^ the path to the file to write to
|
-- ^ the path to the CSV file to read from
|
||||||
-> ConduitT () [T.Text] m ()
|
-> ConduitT i [T.Text] m ()
|
||||||
-- ^ the source conduit
|
readFromCSV file = sourceFile file .| decodeRows
|
||||||
-> m ()
|
|
||||||
writeCSVFromStream file src = runConduit $
|
|
||||||
src .| encodeRows .| sinkFile file
|
|
||||||
|
|
||||||
-- | Write a CSV file from a stream of ByteString-based rows
|
-- | reads a stream of ByteString-based rows from a CSV file
|
||||||
writeRawCSVFromStream
|
readFromCSVRaw
|
||||||
:: MonadResource m
|
:: MonadResource m
|
||||||
=> FilePath
|
=> FilePath
|
||||||
-- ^ the path to the file to write to
|
-- ^ the path to the CSV file to read from
|
||||||
-> ConduitT () [BS.ByteString] m ()
|
-> ConduitT i [BS.ByteString] m ()
|
||||||
-- ^ the source conduit
|
readFromCSVRaw file = sourceFile file .| decodeRawRows
|
||||||
-> m ()
|
|
||||||
writeRawCSVFromStream file src = runConduit $
|
|
||||||
src .| encodeRawRows .| sinkFile file
|
|
||||||
|
|
||||||
-- | encode an entire CSV file
|
-- | encode an entire CSV file
|
||||||
encodeCSV
|
encodeCSV
|
||||||
:: Monad m
|
:: Monad m
|
||||||
=> [[T.Text]]
|
=> [[T.Text]]
|
||||||
-- ^ the data being encoded, organized into rows and fields
|
-- ^ the data being encoded, organized into rows and fields
|
||||||
-> ConduitT () BS.ByteString m ()
|
-> ConduitT o BS.ByteString m ()
|
||||||
encodeCSV csv = sourceList csv .| encodeRows
|
encodeCSV csv = sourceList csv .| encodeRows
|
||||||
|
|
||||||
-- | encode an entire CSV file
|
-- | encode an entire CSV file
|
||||||
@@ -162,9 +162,25 @@ encodeRawCSV
|
|||||||
:: Monad m
|
:: Monad m
|
||||||
=> [[BS.ByteString]]
|
=> [[BS.ByteString]]
|
||||||
-- ^ the data being encoded, organized into rows and fields
|
-- ^ the data being encoded, organized into rows and fields
|
||||||
-> ConduitT () BS.ByteString m ()
|
-> ConduitT o BS.ByteString m ()
|
||||||
encodeRawCSV csv = sourceList csv .| encodeRawRows
|
encodeRawCSV csv = sourceList csv .| encodeRawRows
|
||||||
|
|
||||||
|
-- | Writes a stream of Text-based rows to a CSV file
|
||||||
|
writeToCSV
|
||||||
|
:: MonadResource m
|
||||||
|
=> FilePath
|
||||||
|
-- ^ the path to the CSV file to write to
|
||||||
|
-> ConduitT [T.Text] o m ()
|
||||||
|
writeToCSV file = encodeRows .| sinkFile file
|
||||||
|
|
||||||
|
-- | Writes a stream of ByteString-based rows to a CSV file
|
||||||
|
writeToCSVRaw
|
||||||
|
:: MonadResource m
|
||||||
|
=> FilePath
|
||||||
|
-- ^ the path to the CSV file to write to
|
||||||
|
-> ConduitT [BS.ByteString] o m ()
|
||||||
|
writeToCSVRaw file = encodeRawRows .| sinkFile file
|
||||||
|
|
||||||
-- | encode a CSV stream row by row, each element in the list read
|
-- | encode a CSV stream row by row, each element in the list read
|
||||||
-- represents a field, with the entire list representing a row
|
-- represents a field, with the entire list representing a row
|
||||||
encodeRows :: Monad m => ConduitT [T.Text] BS.ByteString m ()
|
encodeRows :: Monad m => ConduitT [T.Text] BS.ByteString m ()
|
||||||
@@ -196,13 +212,7 @@ decodeRows = decodeRawRows .| mapC (map $ fromMaybe "" . decodeUTF8)
|
|||||||
decodeRawRows :: Monad m => ConduitT BS.ByteString [BS.ByteString] m ()
|
decodeRawRows :: Monad m => ConduitT BS.ByteString [BS.ByteString] m ()
|
||||||
decodeRawRows = toBytes .| evalStateT decodeLoop newDecodeState
|
decodeRawRows = toBytes .| evalStateT decodeLoop newDecodeState
|
||||||
|
|
||||||
-- | decode a raw ByteString into Text (if possible)
|
-- | convert a stream to ByteStrings to a stream of bytes
|
||||||
decodeUTF8 :: BS.ByteString -> Maybe T.Text
|
|
||||||
decodeUTF8 bs = case decodeUtf8' bs of
|
|
||||||
Left _ -> Nothing
|
|
||||||
Right txt -> Just txt
|
|
||||||
|
|
||||||
-- | convert a stream to ByteStrings to a string of bytes
|
|
||||||
toBytes :: Monad m => ConduitT BS.ByteString Word8 m ()
|
toBytes :: Monad m => ConduitT BS.ByteString Word8 m ()
|
||||||
toBytes = await >>= \case
|
toBytes = await >>= \case
|
||||||
Just bs -> do
|
Just bs -> do
|
||||||
@@ -211,6 +221,12 @@ toBytes = await >>= \case
|
|||||||
toBytes
|
toBytes
|
||||||
Nothing -> return ()
|
Nothing -> return ()
|
||||||
|
|
||||||
|
-- | decode a raw ByteString into Text (if possible)
|
||||||
|
decodeUTF8 :: BS.ByteString -> Maybe T.Text
|
||||||
|
decodeUTF8 bs = case decodeUtf8' bs of
|
||||||
|
Left _ -> Nothing
|
||||||
|
Right txt -> Just txt
|
||||||
|
|
||||||
-- Internal
|
-- Internal
|
||||||
|
|
||||||
data DecodeState = DecodeState
|
data DecodeState = DecodeState
|
||||||
|
|||||||
@@ -40,8 +40,8 @@ spec = describe "Data.CSV.Sip" $ do
|
|||||||
labelFieldsSpec
|
labelFieldsSpec
|
||||||
decodeRowsSpec
|
decodeRowsSpec
|
||||||
decodeRawRowsSpec
|
decodeRawRowsSpec
|
||||||
decodeUTF8Spec
|
|
||||||
toBytesSpec
|
toBytesSpec
|
||||||
|
decodeUTF8Spec
|
||||||
|
|
||||||
encodeCSVSpec :: Spec
|
encodeCSVSpec :: Spec
|
||||||
encodeCSVSpec = describe "encodeCSV" $ do
|
encodeCSVSpec = describe "encodeCSV" $ do
|
||||||
@@ -327,6 +327,14 @@ decodeRawRowsSpec = describe "decodeRawRows" $ mapM_
|
|||||||
, ["baz", "quux"]
|
, ["baz", "quux"]
|
||||||
]
|
]
|
||||||
|
|
||||||
|
toBytesSpec :: Spec
|
||||||
|
toBytesSpec = describe "toBytes" $ let
|
||||||
|
input = ["ab", "cd"]
|
||||||
|
expected = map (fromIntegral . ord) "abcd"
|
||||||
|
in it ("should be " ++ show expected) $ do
|
||||||
|
result <- runConduit $ sourceList input .| toBytes .| consume
|
||||||
|
result `shouldBe` expected
|
||||||
|
|
||||||
decodeUTF8Spec :: Spec
|
decodeUTF8Spec :: Spec
|
||||||
decodeUTF8Spec = describe "decodeUTF8" $ mapM_
|
decodeUTF8Spec = describe "decodeUTF8" $ mapM_
|
||||||
( \(label, input, expected) -> context label $
|
( \(label, input, expected) -> context label $
|
||||||
@@ -341,12 +349,4 @@ decodeUTF8Spec = describe "decodeUTF8" $ mapM_
|
|||||||
, ( "blank", "", Just "" )
|
, ( "blank", "", Just "" )
|
||||||
]
|
]
|
||||||
|
|
||||||
toBytesSpec :: Spec
|
|
||||||
toBytesSpec = describe "toBytes" $ let
|
|
||||||
input = ["ab", "cd"]
|
|
||||||
expected = map (fromIntegral . ord) "abcd"
|
|
||||||
in it ("should be " ++ show expected) $ do
|
|
||||||
result <- runConduit $ sourceList input .| toBytes .| consume
|
|
||||||
result `shouldBe` expected
|
|
||||||
|
|
||||||
--jl
|
--jl
|
||||||
|
|||||||
Reference in New Issue
Block a user