From a8413e1cc2d20c5a41a815f517b48b17c89954ac Mon Sep 17 00:00:00 2001
From: Eddie Hebert <eddie@eddiehebert.com>
Date: Thu, 27 Dec 2012 16:58:35 -0500
Subject: [PATCH] Adds reprs for PerformanceTracker and TradingEnvironment.

For debugging in the REPL.
---
 zipline/finance/performance.py | 5 +++++
 zipline/finance/trading.py     | 7 +++++++
 2 files changed, 12 insertions(+)

diff --git a/zipline/finance/performance.py b/zipline/finance/performance.py
index 8c766d20..6f11a62b 100644
--- a/zipline/finance/performance.py
+++ b/zipline/finance/performance.py
@@ -206,6 +206,11 @@ class PerformanceTracker(object):
             keep_transactions=True
         )
 
+    def __repr__(self):
+        return "%s(%r)" % (
+            self.__class__.__name__,
+            {'trading_environment': self.trading_environment})
+
     def transform(self, stream_in):
         """
         Main generator work loop.
diff --git a/zipline/finance/trading.py b/zipline/finance/trading.py
index 7dc97458..0f60b742 100644
--- a/zipline/finance/trading.py
+++ b/zipline/finance/trading.py
@@ -94,6 +94,13 @@ class TradingEnvironment(object):
 
         self.prior_day_open = self.calculate_prior_day_open()
 
+    def __repr__(self):
+        return "%s(%r)" % (
+            self.__class__.__name__,
+            {'first_open': self.first_open,
+             'last_close': self.last_close
+             })
+
     def calculate_first_open(self):
         """
         Finds the first trading day on or after self.period_start.