add functions to fancify items

1 files changed, 11 insertions, 5 deletions

diff --git a/gamelocker/datatypes.py b/gamelocker/datatypes.py
index cf89fad..74ce965 100644
--- a/gamelocker/datatypes.py
+++ b/gamelocker/datatypes.py
@@ -1,9 +1,15 @@
 #!/usr/bin/python
 # -*- coding: utf-8 -*-
 # pylint: disable=missing-docstring
+"""
+gamelocker.datatypes
+
+Classes and utility functions to map API responses to objects.
+"""
 
 import inspect
 import sys
+import gamelocker.strings
 from gamelocker.janus import DataMessage, Attribute
 
 
@@ -38,15 +44,15 @@ class Player(DataMessage):
     key_id = attr(str, "id")
 
     name = attr(str, "name")
-    stats = attr(dict, "stats")
+    stats = attr(gamelocker.strings.Stats, "stats")
 
 
 class Participant(DataMessage):
     type_name = "participant"
     key_id = attr(str, "id")
 
-    actor = attr(str, "actor")
-    stats = attr(dict, "stats")
+    actor = attr(gamelocker.strings.Hero, "actor")
+    stats = attr(gamelocker.strings.Stats, "stats")
 
     player = rel(Player, "player")
 
@@ -62,7 +68,7 @@ class Roster(DataMessage):
     type_name = "roster"
     key_id = attr(str, "id")
 
-    stats = attr(dict, "stats")
+    stats = attr(gamelocker.strings.Stats, "stats")
 
     participants = rel(Participant, "participants")
     team = rel(Team, "team")
@@ -77,7 +83,7 @@ class Match(DataMessage):
     gameMode = attr(str, "gameMode")
     patchVersion = attr(str, "patchVersion")
     region = attr(str, "region")
-    stats = attr(dict, "stats")
+    stats = attr(gamelocker.strings.Stats, "stats")
 
     rosters = rel(Roster, "rosters")