X-Git-Url: http://gitweb.michael.orlitzky.com/?a=blobdiff_plain;f=src%2FTSN%2FXML%2FNews.hs;h=31587a35b0435ca6665c82b8e3a905ccbaa07855;hb=79cba02a6f520256991b345e9dffdaedf68ddde0;hp=550f26c44b77b52b61d36528b142d01fea02891e;hpb=0e37f70a58d512858b38e1458c6d83bc1727269c;p=dead%2Fhtsn-import.git diff --git a/src/TSN/XML/News.hs b/src/TSN/XML/News.hs index 550f26c..31587a3 100644 --- a/src/TSN/XML/News.hs +++ b/src/TSN/XML/News.hs @@ -17,7 +17,6 @@ module TSN.XML.News ( news_tests ) where -import Control.Monad.IO.Class ( MonadIO, liftIO ) import Data.Data ( Data, constrFields, dataTypeConstrs, dataTypeOf ) import Data.List.Utils ( join, split ) import Data.Tuple.Curry ( uncurryN ) @@ -40,20 +39,18 @@ import Text.XML.HXT.Core ( xp12Tuple, xpAttr, xpElem, + xpInt, xpList, xpOption, xpPair, - xpPrim, xpText, xpTriple, xpWrap ) -import Network.Services.TSN.Report ( report_error ) import TSN.Codegen ( tsn_codegen_config, - tsn_db_field_namer -- Used in a test. - ) -import TSN.DbImport ( DbImport(..) ) + tsn_db_field_namer ) -- Used in a test +import TSN.DbImport ( DbImport(..), ImportResult(..) ) import Xml ( ToFromXml(..), pickle_unpickle ) @@ -235,7 +232,7 @@ pickle_msg_id :: PU MsgId pickle_msg_id = xpElem "msg_id" $ xpWrap (from_tuple, to_tuple) $ - xpPair xpPrim (xpAttr "EventId" (xpOption xpPrim)) + xpPair xpInt (xpAttr "EventId" (xpOption xpInt)) where from_tuple = uncurryN MsgId to_tuple m = (db_msg_id m, db_event_id m) @@ -263,7 +260,7 @@ pickle_message :: PU MessageXml pickle_message = xpElem "message" $ xpWrap (from_tuple, to_tuple) $ - xp12Tuple (xpElem "XML_File_ID" xpPrim) + xp12Tuple (xpElem "XML_File_ID" xpInt) (xpElem "heading" xpText) pickle_msg_id (xpElem "category" xpText) @@ -317,8 +314,7 @@ instance DbImport Message where case root_element of Nothing -> do let errmsg = "Could not unpickle News message in dbimport." - liftIO $ report_error errmsg - return Nothing + return $ ImportFailed errmsg Just message -> do news_id <- insert (from_xml message :: Message) let nts :: [NewsTeam] = map (from_xml_fk news_id) @@ -328,7 +324,7 @@ instance DbImport Message where nt_ids <- mapM insert nts loc_ids <- mapM insert nlocs - return $ Just (1 + (length nt_ids) + (length loc_ids)) + return $ ImportSucceeded (1 + (length nt_ids) + (length loc_ids)) -- * Tasty Tests