[py-svn] r8053 - py/dist/py/code

[hpk at codespeak.net]

Author: hpk
Date: Mon Jan  3 18:12:29 2005
New Revision: 8053

Modified:
   py/dist/py/code/excinfo.py
   py/dist/py/code/source.py
Log:
python2.2 fixes 


Modified: py/dist/py/code/excinfo.py
==============================================================================
--- py/dist/py/code/excinfo.py	(original)
+++ py/dist/py/code/excinfo.py	Mon Jan  3 18:12:29 2005
@@ -1,3 +1,4 @@
+from __future__ import generators 
 import sys
 import py
 

Modified: py/dist/py/code/source.py
==============================================================================
--- py/dist/py/code/source.py	(original)
+++ py/dist/py/code/source.py	Mon Jan  3 18:12:29 2005
@@ -23,12 +23,15 @@
             lines.extend(partlines) 
 
     def __getitem__(self, key): 
-        if isinstance(key, slice): 
-            newsource = Source()
-            newsource.lines = self.lines[key] 
-            return newsource 
-        else:
+        if isinstance(key, int): 
             return self.lines[key]
+        else: 
+            return self.__getslice__(key) 
+
+    def __getslice__(self, start, end):
+        newsource = Source()
+        newsource.lines = self.lines[start:end]
+        return newsource 
 
     def strip(self): 
         """ return new source object with trailing 
@@ -127,7 +130,7 @@
         else: 
             source = str(self) 
         try:
-            parser.suite(source) 
+            parser.suite(source+'\n') 
         except (parser.ParserError, SyntaxError):
             return False
         else: