2017-02-13 54 views
1

对于如何将csv-conduit与自定义数据类型一起使用,我感到非常困惑。我想利用股票数据的一排像这样:使用csv-conduit将CSV文件解析为自定义数据类型

Date,Open,High,Low,Close,Volume,Adj Close 
2017-02-10,2312.27002,2319.22998,2311.100098,2316.100098,3475020000,2316.100098 

,并解析到StockInfo类型我在下面MWE声明。我从文档中收集了我需要使我的StockInfoFromNamedRecordToNamedRecordCSV ByteString的实例。我相信我已经为前两项这样做了,但我不明白如何实施CSV ByteString的必要方法。任何帮助将不胜感激。

MWE:

{-# LANGUAGE MultiParamTypeClasses #-} 
{-# LANGUAGE OverloadedStrings  #-} 

module Lib 
    (readStocks 
    ) where 

import   Data.ByteString 
import   Data.Conduit 
import   Data.Conduit.Binary 
import   Data.Conduit.List   as CL 
import   Data.CSV.Conduit 
import   Data.CSV.Conduit.Conversion 
import   Data.Text     (Text) 
import   Data.Vector 
import   System.IO 

readStocks :: FilePath -> IO (Vector StockInfo) 
readStocks fp = readCSVFile defCSVSettings fp 

data StockInfo = StockInfo 
    { date  :: !String 
    , open  :: !Double 
    , high  :: !Double 
    , low  :: !Double 
    , close :: !Double 
    , volume :: !Integer 
    , adjClose :: !Double 
    } 

instance FromNamedRecord StockInfo where 
    parseNamedRecord m = 
    StockInfo <$> 
    m .: "Date" <*> 
    m .: "Open" <*> 
    m .: "High" <*> 
    m .: "Low" <*> 
    m .: "Close" <*> 
    m .: "Volume" <*> 
    m .: "Adj Close" 

instance ToNamedRecord StockInfo where 
    toNamedRecord (StockInfo date open high low close volume adjClose) = 
    namedRecord [ "Date" .= date 
       , "Open" .= open 
       , "High" .= high 
       , "Low" .= low 
       , "Close" .= close 
       , "Volume" .= volume 
       , "Adj Close" .= adjClose 
       ] 

instance CSV ByteString StockInfo where 
    -- rowToStr = undefined 
    -- intoCSV = undefined 
    -- fromCSV = undefined 
+1

库中已定义了一个实例'(FromNamedRecord一,ToNamedRecord一个,CSV S(MapRow字节字符串))=> CSV s(命名为a)'--' Named'就是身份,而且你有'{To/From} NamedRecord'实例,所以你不需要自己写这个实例 - 只需使用'Named'实例。如果你真的想要,你可以根据库提供的通用定义你的'CSV ByteString StockInfo'实例。 – user2407038

+0

明白!我认为这没有意义,但这是我的最好理解。 – anthonybrice

回答

2

这是我应该做的:

{-# LANGUAGE OverloadedStrings  #-} 

module Lib 
    (readStocks 
    ) where 

import   Data.ByteString 
import   Data.Conduit 
import   Data.Conduit.Binary 
import   Data.CSV.Conduit 
import   Data.CSV.Conduit.Conversion 
import   Data.Text     (Text) 
import   Data.Vector 
import   System.IO 

readStocks :: FilePath -> IO (Vector (Named StockInfo)) 
readStocks fp = readCSVFile defCSVSettings fp 

data StockInfo = StockInfo 
    { date  :: !String 
    , open  :: !Double 
    , high  :: !Double 
    , low  :: !Double 
    , close :: !Double 
    , volume :: !Integer 
    , adjClose :: !Double 
    } deriving (Show, Eq, Read) 

instance FromNamedRecord StockInfo where 
    parseNamedRecord m = 
    StockInfo <$> 
    m .: "Date" <*> 
    m .: "Open" <*> 
    m .: "High" <*> 
    m .: "Low" <*> 
    m .: "Close" <*> 
    m .: "Volume" <*> 
    m .: "Adj Close" 

instance ToNamedRecord StockInfo where 
    toNamedRecord (StockInfo date open high low close volume adjClose) = 
    namedRecord [ "Date" .= date 
       , "Open" .= open 
       , "High" .= high 
       , "Low" .= low 
       , "Close" .= close 
       , "Volume" .= volume 
       , "Adj Close" .= adjClose 
       ] 
相关问题