+{-# LANGUAGE DeriveGeneric #-}
{-# LANGUAGE FlexibleInstances #-}
{-# LANGUAGE GADTs #-}
{-# LANGUAGE QuasiQuotes #-}
{-# LANGUAGE RecordWildCards #-}
-{-# LANGUAGE StandaloneDeriving #-}
{-# LANGUAGE TemplateHaskell #-}
{-# LANGUAGE TypeFamilies #-}
import Control.Monad ( forM_ )
import Data.Time ( UTCTime )
import Data.Tuple.Curry ( uncurryN )
+import qualified Data.Vector.HFixed as H ( HVector, cons, convert, tail )
+import qualified Data.Vector.HFixed.Cont as H (ContVec)
import Database.Groundhog (
DefaultKey,
countAll,
- executeRaw,
+ deleteAll,
migrate,
runMigration,
silentMigrationLogger )
import Database.Groundhog.TH (
groundhog,
mkPersist )
+import qualified GHC.Generics as GHC ( Generic )
import Test.Tasty ( TestTree, testGroup )
import Test.Tasty.HUnit ( (@?=), testCase )
import Text.XML.HXT.Core (
import TSN.Picklers( xp_date, xp_time_stamp )
import TSN.XmlImport ( XmlImport(..), XmlImportFk(..) )
import Xml (
+ Child(..),
FromXml(..),
FromXmlFk(..),
ToDb(..),
xml_sport :: String,
xml_listings :: [InjuriesDetailListingXml],
xml_time_stamp :: UTCTime }
- deriving (Eq, Show)
+ deriving (Eq, GHC.Generic, Show)
+
+
+-- | For 'H.convert'.
+--
+instance H.HVector Message
+
-- | Database representation of a 'Message'.
--
xml_full_name :: String, -- ^ Team full name
xml_player_listings :: [InjuriesDetailListingPlayerListingXml] }
- deriving (Eq, Show)
+ deriving (Eq, GHC.Generic, Show)
+
+
+-- | For 'H.convert'.
+--
+instance H.HVector InjuriesDetailListingXml
+
instance ToDb InjuriesDetailListingXml where
-- | The database analogue of an 'InjuriesDetailListingXml' is a
-- 'InjuriesDetailListing'.
type Db InjuriesDetailListingXml = InjuriesDetailListing
-instance FromXmlFk InjuriesDetailListingXml where
+
+instance Child InjuriesDetailListingXml where
-- | Each 'InjuriesDetailListingXml' is contained in an
-- 'InjuriesDetail'.
type Parent InjuriesDetailListingXml = InjuriesDetail
+
+instance FromXmlFk InjuriesDetailListingXml where
-- | Construct a 'InjuriesDetailListing' from a
-- 'InjuriesDetailListingXml' and a foreign key to a
-- 'InjuriesDetail'.
-- * InjuriesDetailListingPlayerListing
-- | XML representation of a \<PlayerListing\>, the main type of
--- element contains in Injuries_Detail_XML messages.
+-- element contains in Injuries_Detail_XML messages. The leading
+-- underscores prevent unused field warnings.
--
data InjuriesDetailListingPlayerListingXml =
InjuriesDetailListingPlayerListingXml {
- xml_player_team_id :: String, -- ^ TeamIDs are (apparently) three
+ _xml_player_team_id :: String, -- ^ TeamIDs are (apparently) three
-- characters long and not
-- necessarily numeric. Postgres
-- imposes no performance penalty
-- bound of three characters.
-- We add the \"player\" to avoid conflict
-- with 'InjuriesDetailListingXml'.
- xml_player_id :: Int,
- xml_date :: UTCTime,
- xml_pos :: String,
- xml_name :: String,
- xml_injury :: String,
- xml_status :: String,
- xml_fantasy :: Maybe String, -- ^ Nobody knows what this is.
- xml_injured :: Bool,
- xml_type :: String }
- deriving (Eq, Show)
-
+ _xml_player_id :: Int,
+ _xml_date :: UTCTime,
+ _xml_pos :: String,
+ _xml_name :: String,
+ _xml_injury :: String,
+ _xml_status :: String,
+ _xml_fantasy :: Maybe String, -- ^ Nobody knows what this is.
+ _xml_injured :: Bool,
+ _xml_type :: String }
+ deriving (Eq, GHC.Generic, Show)
+
+
+-- | For 'H.convert'.
+--
+instance H.HVector InjuriesDetailListingPlayerListingXml
-- | Database representation of a
-- 'InjuriesDetailListingPlayerListingXml'. We drop the team_id
--- because it's redundant.
+-- because it's redundant. The leading underscores prevent unused
+-- field warnings.
--
data InjuriesDetailListingPlayerListing =
InjuriesDetailListingPlayerListing {
- db_injuries_detail_listings_id :: DefaultKey InjuriesDetailListing,
- db_player_id :: Int,
- db_date :: UTCTime,
- db_pos :: String,
- db_name :: String,
- db_injury :: String,
- db_status :: String,
- db_fantasy :: Maybe String, -- ^ Nobody knows what this is.
- db_injured :: Bool,
- db_type :: String }
+ _db_injuries_detail_listings_id :: DefaultKey InjuriesDetailListing,
+ _db_player_id :: Int,
+ _db_date :: UTCTime,
+ _db_pos :: String,
+ _db_name :: String,
+ _db_injury :: String,
+ _db_status :: String,
+ _db_fantasy :: Maybe String, -- ^ Nobody knows what this is.
+ _db_injured :: Bool,
+ _db_type :: String }
+ deriving ( GHC.Generic )
+
+-- | For 'H.cons', 'H.tail', etc.
+--
+instance H.HVector InjuriesDetailListingPlayerListing
instance ToDb InjuriesDetailListingPlayerListingXml where
type Db InjuriesDetailListingPlayerListingXml =
InjuriesDetailListingPlayerListing
-instance FromXmlFk InjuriesDetailListingPlayerListingXml where
+
+instance Child InjuriesDetailListingPlayerListingXml where
-- | Each 'InjuriesDetailListingPlayerListingXml' is contained in an
-- 'InjuriesDetailListing'.
--
type Parent InjuriesDetailListingPlayerListingXml = InjuriesDetailListing
+
+instance FromXmlFk InjuriesDetailListingPlayerListingXml where
-- | To construct a 'InjuriesDetailListingPlayerListing' from a
-- 'InjuriesDetailListingPlayerListingXml' we need to supply a
-- foreign key to an 'InjuriesDetailListing'.
--
- from_xml_fk fk InjuriesDetailListingPlayerListingXml{..} =
- InjuriesDetailListingPlayerListing {
- db_injuries_detail_listings_id = fk,
- db_player_id = xml_player_id,
- db_date = xml_date,
- db_pos = xml_pos,
- db_name = xml_name,
- db_injury = xml_injury,
- db_status = xml_status,
- db_fantasy = xml_fantasy,
- db_injured = xml_injured,
- db_type = xml_type }
+ from_xml_fk fk = (H.cons fk) . asCont . H.tail
+ where
+ -- Should be in the library soon.
+ asCont :: H.ContVec a -> H.ContVec a
+ asCont = id
+
-- | This lets us insert the XML representation
-- 'InjuriesDetailListingPlayerListingXml' directly.
constructors:
- name: InjuriesDetailListingPlayerListing
fields:
- - name: db_injuries_detail_listings_id
+ - name: _db_injuries_detail_listings_id
reference:
onDelete: cascade
|]
pickle_player_listing :: PU InjuriesDetailListingPlayerListingXml
pickle_player_listing =
xpElem "PlayerListing" $
- xpWrap (from_tuple, to_tuple) $
+ xpWrap (from_tuple, H.convert) $
xp10Tuple (xpElem "TeamID" xpText)
(xpElem "PlayerID" xpInt)
(xpElem "Date" xp_date)
(xpElem "Type" xpText)
where
from_tuple = uncurryN InjuriesDetailListingPlayerListingXml
- to_tuple pl = (xml_player_team_id pl,
- xml_player_id pl,
- xml_date pl,
- xml_pos pl,
- xml_name pl,
- xml_injury pl,
- xml_status pl,
- xml_fantasy pl,
- xml_injured pl,
- xml_type pl)
-- | Convert 'Listing's to/from XML.
pickle_listing :: PU InjuriesDetailListingXml
pickle_listing =
xpElem "Listing" $
- xpWrap (from_tuple, to_tuple) $
+ xpWrap (from_tuple, H.convert) $
xpTriple (xpElem "TeamID" xpText)
(xpElem "FullName" xpText)
(xpList pickle_player_listing)
where
from_tuple = uncurryN InjuriesDetailListingXml
- to_tuple l = (xml_team_id l,
- xml_full_name l,
- xml_player_listings l)
-- | Convert 'Message's to/from XML.
pickle_message :: PU Message
pickle_message =
xpElem "message" $
- xpWrap (from_tuple, to_tuple) $
+ xpWrap (from_tuple, H.convert) $
xp6Tuple (xpElem "XML_File_ID" xpInt)
(xpElem "heading" xpText)
(xpElem "category" xpText)
(xpElem "time_stamp" xp_time_stamp)
where
from_tuple = uncurryN Message
- to_tuple m = (xml_xml_file_id m,
- xml_heading m,
- xml_category m,
- xml_sport m,
- xml_listings m,
- xml_time_stamp m)
--
migrate b
migrate c
_ <- dbimport inj
- -- No idea how 'delete' works, so do this instead.
- executeRaw False "DELETE FROM injuries_detail;" []
+ deleteAll a
count_a <- countAll a
count_b <- countAll b
count_c <- countAll c
projects / dead / htsn-import.git / blobdiff