X-Git-Url: http://gitweb.michael.orlitzky.com/?a=blobdiff_plain;f=src%2FMain.hs;h=90f5f91d9682b5366bbed33d2b00b334b567f0dd;hb=a9a8667246a544705d85698ec967437e4770be2c;hp=d5f6612611a29f090411e47e1815860f9417dd23;hpb=b3a3f27b592e389d69babaca741e8553bb05fa92;p=dead%2Fhtsn-import.git diff --git a/src/Main.hs b/src/Main.hs index d5f6612..90f5f91 100644 --- a/src/Main.hs +++ b/src/Main.hs @@ -46,7 +46,8 @@ import Network.Services.TSN.Report ( report_error ) import TSN.DbImport ( DbImport(..), ImportResult(..) ) import qualified TSN.XML.AutoRacingResults as AutoRacingResults ( - dtd ) + dtd, + pickle_message ) import qualified TSN.XML.AutoRacingSchedule as AutoRacingSchedule ( dtd, pickle_message ) @@ -56,8 +57,12 @@ import qualified TSN.XML.Injuries as Injuries ( dtd, pickle_message ) import qualified TSN.XML.InjuriesDetail as InjuriesDetail ( dtd, pickle_message ) +import qualified TSN.XML.JFile as JFile ( dtd, pickle_message ) import qualified TSN.XML.News as News ( dtd, pickle_message ) import qualified TSN.XML.Odds as Odds ( dtd, pickle_message ) +import qualified TSN.XML.ScheduleChanges as ScheduleChanges ( + dtd, + pickle_message ) import qualified TSN.XML.Scores as Scores ( dtd, pickle_message ) import qualified TSN.XML.SportInfo as SportInfo ( dtds, parse_xml ) import qualified TSN.XML.Weather as Weather ( dtd, pickle_message ) @@ -162,39 +167,43 @@ import_file cfg path = do -- migrate_and_import m = dbmigrate m >> dbimport m + -- | The error message we return if unpickling fails. + -- + errmsg = "Could not unpickle " ++ dtd ++ "." + + -- | Try to migrate and import using the given pickler @f@; + -- if it works, return the result. Otherwise, return an + -- 'ImportFailed' along with our error message. + -- + go f = maybe + (return $ ImportFailed errmsg) + migrate_and_import + (unpickleDoc f xml) + importer - | dtd == AutoRacingResults.dtd = undefined + | dtd == AutoRacingResults.dtd = + go AutoRacingResults.pickle_message + + | dtd == AutoRacingSchedule.dtd = + go AutoRacingSchedule.pickle_message - | dtd == AutoRacingSchedule.dtd = do - let m = unpickleDoc AutoRacingSchedule.pickle_message xml - maybe (return $ ImportFailed errmsg) migrate_and_import m + -- GameInfo and SportInfo appear last in the guards + | dtd == Injuries.dtd = go Injuries.pickle_message - -- GameInfo and SportInfo appear least in the guards - | dtd == Injuries.dtd = do - let m = unpickleDoc Injuries.pickle_message xml - maybe (return $ ImportFailed errmsg) migrate_and_import m + | dtd == InjuriesDetail.dtd = go InjuriesDetail.pickle_message - | dtd == InjuriesDetail.dtd = do - let m = unpickleDoc InjuriesDetail.pickle_message xml - maybe (return $ ImportFailed errmsg) migrate_and_import m + | dtd == JFile.dtd = go JFile.pickle_message + | dtd == News.dtd = go News.pickle_message - | dtd == News.dtd = do - let m = unpickleDoc News.pickle_message xml - maybe (return $ ImportFailed errmsg) migrate_and_import m + | dtd == Odds.dtd = go Odds.pickle_message - | dtd == Odds.dtd = do - let m = unpickleDoc Odds.pickle_message xml - maybe (return $ ImportFailed errmsg) migrate_and_import m + | dtd == ScheduleChanges.dtd = go ScheduleChanges.pickle_message - | dtd == Scores.dtd = do - let m = unpickleDoc Scores.pickle_message xml - maybe (return $ ImportFailed errmsg) migrate_and_import m + | dtd == Scores.dtd = go Scores.pickle_message - -- SportInfo and GameInfo appear least in the guards - | dtd == Weather.dtd = do - let m = unpickleDoc Weather.pickle_message xml - maybe (return $ ImportFailed errmsg) migrate_and_import m + -- SportInfo and GameInfo appear last in the guards + | dtd == Weather.dtd = go Weather.pickle_message | dtd `elem` GameInfo.dtds = do let either_m = GameInfo.parse_xml dtd xml @@ -217,8 +226,6 @@ import_file cfg path = do "Unrecognized DTD in " ++ path ++ ": " ++ dtd ++ "." return $ ImportUnsupported infomsg - where - errmsg = "Could not unpickle " ++ dtd ++ "." -- | Entry point of the program. It twiddles some knobs for