Skip to content

Commit 2c35927

Browse files
committed
Improve range
1 parent 8888992 commit 2c35927

File tree

2 files changed

+6
-3
lines changed

2 files changed

+6
-3
lines changed

pseudo_python/ast_translator.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1636,10 +1636,11 @@ def _translate_iter(self,target, k):
16361636
raise PseudoPythonTypeCheckError('enumerate expected one arg not %d and two indices' % len(k.args))
16371637
return self._translate_enumerate(target.elts, k.args[0])
16381638
elif k.func.id == 'range':
1639-
if not isinstance(target, ast.Tuple) or len(target.elts) != 2:
1639+
if isinstance(target, ast.Name):
1640+
return self._translate_range([target], k.args)
1641+
elif not isinstance(target, (ast.Name, ast.Tuple)) or isinstance(target, ast.Tuple) and len(target.elts) != 2:
16401642
raise PseudoPythonTypeCheckError('range expected two indices')
1641-
1642-
if not k.args or len(k.args) > 3:
1643+
elif not k.args or len(k.args) > 3:
16431644
raise PseudoPythonTypeCheckError('range expected 1 to 3 args not %d' % len(k.args))
16441645
return self._translate_range(target.elts, k.args)
16451646
elif k.func.id == 'zip':

pseudo_python/main.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,8 @@
11
#!/usr/bin/env python3
22
import os
33
import sys
4+
sys.path.append("/home/alehander42/pseudo-python/")
5+
sys.path.append("/home/alehander42/pseudo/")
46
import pseudo_python
57
import pseudo_python.errors
68
import pseudo

0 commit comments

Comments
 (0)