Add function that returns information on every team

Fixes #16
panzarino · Nov 9, 2016 · 6b9a476 · 6b9a476
1 parent c73bf79
commit 6b9a476
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/mlbgame/__init__.py b/mlbgame/__init__.py
@@ -276,3 +276,9 @@ def league():
     """Return Info object that contains league information"""
     info = mlbgame.info.league_info()
     return mlbgame.info.Info(info)
+
+def teams():
+    """Return list of Info objects for each team"""
+    info = mlbgame.info.team_info()
+    output = [mlbgame.info.Info(x) for x in info]
+    return output