Add SportInfo support for mlbrunsleadersxml.dtd.

[dead/htsn-import.git] / src / TSN / Picklers.hs

diff --git a/src/TSN/Picklers.hs b/src/TSN/Picklers.hs
index 387f4cf4fe4926d0c436678ca0e2ed509b075049..ea335c81f2894e8de38a44bddea3c5b2938f82c4 100644 (file)
--- a/src/TSN/Picklers.hs
+++ b/src/TSN/Picklers.hs
@@ -13,6 +13,8 @@ module TSN.Picklers (
 where
 
 -- System imports.
+import Data.List ( intercalate )
+import Data.List.Split ( chunksOf )
 import Data.String.Utils ( replace )
 import Data.Time.Clock ( NominalDiffTime, UTCTime, addUTCTime )
 import Data.Time.Format ( formatTime, parseTime )
@@ -33,8 +35,9 @@ import TSN.Parse (
 -- | The format string for a base date in m/d/yyyy format. The
 --   day/month are not padded at all. This will match for example,
 --
---   * 2/15/1983
---   * 1/1/0000
+--   * 2\/15\/1983
+--
+--   * 1\/1\/0000
 --
 date_format :: String
 date_format = "%-m/%-d/%Y"
@@ -44,8 +47,9 @@ date_format = "%-m/%-d/%Y"
 --   day/month are padded to two characters with zeros. This will
 --   match for example,
 --
---   * 02/15/1983
---   * 01/01/0000
+--   * 02\/15\/1983
+--
+--   * 01\/01\/0000
 --
 date_format_padded :: String
 date_format_padded = "%0m/%0d/%Y"
@@ -78,12 +82,39 @@ xp_date_padded =
     from_date = formatTime defaultTimeLocale date_format_padded
 
 
+
+-- | Format a number as a string using a comma as the thousands
+--   separator.
+--
+--   Examples:
+--
+--   >>> format_commas 0
+--   "0"
+--   >>> format_commas 10
+--   "10"
+--   >>> format_commas 100
+--   "100"
+--   >>> format_commas 1000
+--   "1,000"
+--   >>> format_commas 10000
+--   "10,000"
+--   >>> format_commas 100000
+--   "100,000"
+--   >>> format_commas 1000000
+--   "1,000,000"
+--
+format_commas :: Int -> String
+format_commas x =
+  reverse (intercalate "," $ chunksOf 3 $ reverse $ show x)
+
 -- | Parse \<Earnings\> from an 'AutoRaceResultsListing'. These are
 --   essentially 'Int's, but they look like,
 --
---   * <Earnings>336,826</Earnings>
---   * <Earnings>1,000,191</Earnings>
---   * <Earnings>TBA</Earnings>
+--   * \<Earnings\>336,826\</Earnings\>
+--
+--   * \<Earnings\>1,000,191\</Earnings\>
+--
+--   * \<Earnings\>TBA\</Earnings\>
 --
 xp_earnings :: PU (Maybe Int)
 xp_earnings =
@@ -95,20 +126,21 @@ xp_earnings =
     to_earnings :: String -> Maybe Int
     to_earnings s
       | s == "TBA" = Nothing
-      | otherwise = (read . strip_commas . show) s
+      | otherwise = Just  $ (read . strip_commas) s
 
     from_earnings :: Maybe Int -> String
-    from_earnings Nothing = ""
-    from_earnings (Just i) = show i
+    from_earnings Nothing = "TBA"
+    from_earnings (Just i) = format_commas i
 
 
 -- | (Un)pickle a 'UTCTime' from a \<RaceDate\> element in an
 --   'AutoRaceResults' message.
 --
---   Example:
+--   Examples:
+--
+--   * \<RaceDate\>6/1/2014 1:00:00 PM\</RaceDate\>
 --
---   <RaceDate>6/1/2014 1:00:00 PM</RaceDate>
---   <RaceDate>5/24/2014 2:45:00 PM</RaceDate>
+--   * \<RaceDate\>5/24/2014 2:45:00 PM\</RaceDate\>
 --
 xp_racedate :: PU UTCTime
 xp_racedate =
@@ -200,7 +232,7 @@ xp_tba_time =
       | otherwise = parseTime defaultTimeLocale time_format s
 
     from_time :: Maybe UTCTime -> String
-    from_time Nothing = ""
+    from_time Nothing = "TBA"
     from_time (Just t) = formatTime defaultTimeLocale time_format t