Merge pull request #1199 from dufferzafar/pathod-generator-hotfix

Handle the slice object case in TransformGenerator.__getitem__
mitmproxy · Jun 2, 2016 · b9ad7be · b9ad7be
2 parents 66ec194 + 8e60a9b
commit b9ad7be
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pathod/language/generators.py b/pathod/language/generators.py
@@ -37,6 +37,8 @@ def __len__(self):
 
     def __getitem__(self, x):
         d = self.gen.__getitem__(x)
+        if isinstance(x, slice):
+            return self.transform(x.start, d)
         return self.transform(x, d)
 
     def __repr__(self):