X-Git-Url: http://gitweb.michael.orlitzky.com/?a=blobdiff_plain;f=src%2FMain.hs;h=84593976ec9e3eb05300e89bc4fe42e54b7cf663;hb=d2f5d93b2b68f581d4cb4eabecc556c01762d370;hp=da27b12e9a59ebf8a1e27a4c5fd4a13cf42f85cc;hpb=0105a698fb3ff86773d08471b7506c6af4d4d673;p=dead%2Fhtsn-import.git

diff --git a/src/Main.hs b/src/Main.hs
index da27b12..8459397 100644
--- a/src/Main.hs
+++ b/src/Main.hs
@@ -45,6 +45,8 @@ import Network.Services.TSN.Report (
   report_info,
   report_error )
 import TSN.DbImport ( DbImport(..), ImportResult(..) )
+import qualified TSN.XML.AutoRacingSchedule as AutoRacingSchedule (
+  pickle_message )
 import qualified TSN.XML.Heartbeat as Heartbeat ( verify )
 import qualified TSN.XML.Injuries as Injuries ( pickle_message )
 import qualified TSN.XML.InjuriesDetail as InjuriesDetail ( pickle_message )
@@ -151,6 +153,11 @@ import_file cfg path = do
           migrate_and_import m = dbmigrate m >> dbimport m
 
           importer
+            | dtd == "Auto_Racing_Schedule_XML.dtd" = do
+               let m = unpickleDoc AutoRacingSchedule.pickle_message xml
+               let errmsg = "Could not unpickle Auto_Racing_Schedule_XML."
+               maybe (return $ ImportFailed errmsg) migrate_and_import m
+
             | dtd == "injuriesxml.dtd" = do
                let m = unpickleDoc Injuries.pickle_message xml
                let errmsg = "Could not unpickle injuriesxml."