--
module TSN.Picklers (
xp_date,
- xp_team_id )
+ xp_gamedate,
+ xp_time )
where
+-- System imports.
import Data.Time.Clock ( UTCTime )
import Data.Time.Format ( formatTime, parseTime )
import System.Locale ( defaultTimeLocale )
-import Text.Printf ( printf )
-import Text.Read ( readMaybe )
import Text.XML.HXT.Arrow.Pickle (
xpText,
xpWrapMaybe )
from_date = formatTime defaultTimeLocale format
--- | Parse a team_id. This *should* just be an 'Int', but TSN is doing
--- something weird. First of all, player IDs do look like normal
--- 'Int's. But the team IDs are all stuck in the triple digits, and
--- double-digit team IDs appear to be padded to three characters
--- with a leading '0'. So maybe they're treating these as text?
+-- | (Un)pickle a UTCTime from a weather forecast's gamedate. Example
+-- input looks like,
--
--- In any case, we do the simplest thing that is correct for all the
--- XML we've got: pad it to (only) three digits on pickling.
+-- \<forecast gamedate=\"Monday, December 30th\"\>
--
-xp_team_id :: PU Int
-xp_team_id =
- (to_team_id, from_team_id) `xpWrapMaybe` xpText
+-- When unpickling we get rid of the suffixes \"st\", \"nd\", \"rd\", and
+-- \"th\". During pickling, we add them back based on the last digit
+-- of the date.
+--
+xp_gamedate :: PU UTCTime
+xp_gamedate =
+ (to_gamedate, from_gamedate) `xpWrapMaybe` xpText
where
- to_team_id :: String -> Maybe Int
- to_team_id = readMaybe
+ format = "%A, %B %-d"
+
+ to_gamedate :: String -> Maybe UTCTime
+ to_gamedate s =
+ parseTime defaultTimeLocale format s'
+ where
+ s' = case (reverse s) of
+ (c2:c1:cs) -> let suffix = c1:(c2:[])
+ in
+ case suffix of
+ "st" -> reverse cs
+ "nd" -> reverse cs
+ "rd" -> reverse cs
+ "th" -> reverse cs
+ _ -> s -- Unknown suffix, leave it alone.
+ _ -> s -- The String is less than two characters long,
+ -- leave it alone.
+
+
+ from_gamedate :: UTCTime -> String
+ from_gamedate d = s ++ (suffix s)
+ where
+ s = formatTime defaultTimeLocale format d
+
+ suffix :: String -> String
+ suffix cs =
+ case (reverse cs) of
+ [] -> []
+ ('1':_) -> "st"
+ ('2':_) -> "nd"
+ ('3':_) -> "rd"
+ _ -> "th"
+
+
+-- | (Un)pickle a UTCTime without the date portion.
+--
+xp_time :: PU UTCTime
+xp_time =
+ (to_time, from_time) `xpWrapMaybe` xpText
+ where
+ format = "%I:%M %p"
+
+ to_time :: String -> Maybe UTCTime
+ to_time = parseTime defaultTimeLocale format
+
+ from_time :: UTCTime -> String
+ from_time = formatTime defaultTimeLocale format
- from_team_id :: Int -> String
- from_team_id = printf "%03d"