X-Git-Url: http://gitweb.michael.orlitzky.com/?a=blobdiff_plain;f=src%2FTSN%2FXML%2FEarlyLine.hs;h=4ad9ae2ca0a5f1bc59c9d0b3bb60b71f13f8fa5d;hb=16d86e7a3c1eda08b91752f92510a1de0b952a17;hp=3870eb8d139b59a009fcc58c0cceb746cbaa54e7;hpb=c024d97dd3686c140de40a039b1298d2a7882506;p=dead%2Fhtsn-import.git

diff --git a/src/TSN/XML/EarlyLine.hs b/src/TSN/XML/EarlyLine.hs
index 3870eb8..4ad9ae2 100644
--- a/src/TSN/XML/EarlyLine.hs
+++ b/src/TSN/XML/EarlyLine.hs
@@ -5,6 +5,11 @@
 {-# LANGUAGE TemplateHaskell #-}
 {-# LANGUAGE TypeFamilies #-}
 
+-- | Parse TSN XML for the DTD \"earlylineXML.dtd\". Each \<message\>
+--   element contains a bunch of \<date\>s, and those \<date\>s
+--   contain a single \<game\>. In the database, we merge the date
+--   info into the games, and key the games to the messages.
+--
 module TSN.XML.EarlyLine (
   dtd,
   pickle_message,
@@ -100,7 +105,7 @@ data Message =
     xml_category :: String,
     xml_sport :: String,
     xml_title :: String,
-    xml_dates :: [EarlyLineDateXml],
+    xml_dates :: [EarlyLineDate],
     xml_time_stamp :: UTCTime }
   deriving (Eq, Show)
 
@@ -135,18 +140,18 @@ instance XmlImport Message
 
 
 
--- * EarlyLineDateXml
+-- * EarlyLineDate
 
 -- | XML representation of a \<date\>. It has a \"value\" attribute
 --   containing the actual date string. As children it contains a
 --   (non-optional) note, and a game. The note and date value are
 --   properties of the game as far as I can tell.
 --
-data EarlyLineDateXml =
-  EarlyLineDateXml {
-    xml_date_value :: UTCTime,
-    xml_note :: String,
-    xml_game :: EarlyLineGameXml }
+data EarlyLineDate =
+  EarlyLineDate {
+    date_value :: UTCTime,
+    date_note :: String,
+    date_game :: EarlyLineGameXml }
   deriving (Eq, Show)
 
 
@@ -182,18 +187,27 @@ data EarlyLineGameTeam =
   deriving (Eq, Show)
 
 
-date_to_game :: (DefaultKey EarlyLine) -> EarlyLineDateXml -> EarlyLineGame
+-- | Convert an 'EarlyLineDate' into an 'EarlyLineGame'. Each date has
+--   exactly one game, and the fields that belong to the date should
+--   really be in the game anyway. So the database representation of a
+--   game has the combined fields of the XML date/game.
+--
+--   This function gets the game out of a date, and then sticks the
+--   date value and note inside the game. It also adds the foreign key
+--   reference to the game's parent message, and returns the result.
+--
+date_to_game :: (DefaultKey EarlyLine) -> EarlyLineDate -> EarlyLineGame
 date_to_game fk date =
   EarlyLineGame {
     db_early_lines_id = fk,
     db_game_time = combined_date_time,
-    db_note = (xml_note date),
-    db_away_team = xml_away_team (xml_game date),
-    db_home_team = xml_home_team (xml_game date),
-    db_over_under = xml_over_under (xml_game date) }
+    db_note = (date_note date),
+    db_away_team = xml_away_team (date_game date),
+    db_home_team = xml_home_team (date_game date),
+    db_over_under = xml_over_under (date_game date) }
   where
-    date_part = xml_date_value date
-    time_part = xml_game_time (xml_game date)
+    date_part = date_value date
+    time_part = xml_game_time (date_game date)
     combined_date_time = UTCTime (utctDay date_part) (utctDayTime time_part)
 
 --
@@ -268,6 +282,10 @@ mkPersist tsn_codegen_config [groundhog|
 --
 -- * Pickling
 --
+
+
+-- | Pickler for the top-level 'Message'.
+--
 pickle_message :: PU Message
 pickle_message =
   xpElem "message" $
@@ -289,7 +307,10 @@ pickle_message =
                   xml_dates m,
                   xml_time_stamp m)
 
-pickle_date :: PU EarlyLineDateXml
+
+-- | Pickler for the \<date\> elements within each \<message\>.
+--
+pickle_date :: PU EarlyLineDate
 pickle_date =
   xpElem "date" $
     xpWrap (from_tuple, to_tuple) $
@@ -297,10 +318,13 @@ pickle_date =
              (xpElem "note" xpText)
              pickle_game
   where
-    from_tuple = uncurryN EarlyLineDateXml
-    to_tuple m = (xml_date_value m, xml_note m, xml_game m)
+    from_tuple = uncurryN EarlyLineDate
+    to_tuple m = (date_value m, date_note m, date_game m)
 
 
+
+-- | Pickler for the \<game\> element within each \<date\>.
+--
 pickle_game :: PU EarlyLineGameXml
 pickle_game =
   xpElem "game" $
@@ -318,12 +342,26 @@ pickle_game =
 
 
 
+-- | Pickle an away team (\<teamA\>) element within a \<game\>. Most
+--   of the work (common with the home team pickler) is done by
+--   'pickle_team'.
+--
 pickle_away_team :: PU EarlyLineGameTeam
 pickle_away_team = xpElem "teamA" pickle_team
 
+
+-- | Pickle a home team (\<teamH\>) element within a \<game\>. Most
+--   of the work (common with theaway team pickler) is done by
+--   'pickle_team'.
+--
 pickle_home_team :: PU EarlyLineGameTeam
 pickle_home_team = xpElem "teamH" pickle_team
 
+
+-- | Team pickling common to both 'pickle_away_team' and
+--   'pickle_home_team'. Handles everything inside the \<teamA\> and
+--   \<teamH\> elements.
+--
 pickle_team :: PU EarlyLineGameTeam
 pickle_team =
   xpWrap (from_tuple, to_tuple) $