Changeset 109:7352863453bc

Timestamp:

12/20/10 12:50:42 (5 years ago)

Author:

Daniil Alexeyevsky <me.dendik@òÀæ>

Branch:

default

Message:

snake.*.load now reports the line number of error

File:

• snake.py (modified) (2 diffs)

snake.py

@@ -9 +9 @@
     line = line.rstrip()
     return line
+
+class File(object):
+    """Wrapper around file that saves the current line number."""
+    def __init__(self, file):
+        self.file = file
+        self.name = file.name
+        self.line_no = 0
+    def __iter__(self):
+        for line_no, line in enumerate(self.file, self.line_no):
+            self.line_no = line_no
+            yield line
 
 class Snake(object):
@@ -32 +43 @@
         See program design docs for file syntax.
         """
+        file = File(file)
+        try:
+            self._load(file)
+        except Exception, e:
+            raise Exception("%s:%s: %s" % (file.name, file.line_no, e))
+
+    def _load (self, file):
+        """Actually do the loading."""
         for line in file:
             magic, self.name = preprocess(line).split(' ', 1)