Added processor and reader modules which are used in the doctests

Siddharta Govindaraj · Siddharta Govindaraj · commit 480cf94e7684 · 2015-03-25T08:37:09.000+05:30
diff --git a/stock_alerter/processor.py b/stock_alerter/processor.py
@@ -0,0 +1,9 @@
+class Processor:
+    def __init__(self, reader, exchange):
+        self.reader = reader
+        self.exchange = exchange
+
+    def process(self):
+        for symbol, timestamp, price in self.reader.get_updates():
+            stock = self.exchange[symbol]
+            stock.update(timestamp, price)
diff --git a/stock_alerter/reader.py b/stock_alerter/reader.py
@@ -0,0 +1,28 @@
+from datetime import datetime
+
+
+class ListReader:
+    """Reads a series of updates from a list"""
+    def __init__(self, updates):
+        self.updates = updates
+
+    def get_updates(self):
+        for update in self.updates:
+            yield update
+
+
+class FileReader:
+    """Reads a series of stock updates from a file"""
+    def __init__(self, filename):
+        self.filename = filename
+
+    def get_updates(self):
+        """Returns the next update everytime the method is called"""
+        with open(self.filename, "r") as fp:
+            data = fp.read()
+            lines = data.split()
+            for line in lines:
+                symbol, timestamp, price = line.split(",")
+                yield (symbol,
+                       datetime.strptime(timestamp, "%Y-%m-%dT%H:%M:%S.%f"),
+                       int(price))
