Verify that path in "random" operation is a directory

This commit is contained in:
Danilo Bargen
2013-08-09 13:32:43 +02:00
parent 1762cb2ef8
commit 7806119911

View File

@@ -61,6 +61,7 @@ class SourceFinder(object):
@staticmethod @staticmethod
def fetch(file_path): def fetch(file_path):
assert os.path.isdir(file_path), 'Path must be a directory, not a file.'
for root, dirnames, filenames in os.walk(file_path): for root, dirnames, filenames in os.walk(file_path):
for name in filenames: for name in filenames:
if name.endswith('.py'): if name.endswith('.py'):
@@ -202,6 +203,9 @@ def main(arguments):
int(arguments['<line>']), int(arguments['<column>']) int(arguments['<line>']), int(arguments['<column>'])
).run(debugger, print_result=True) ).run(debugger, print_result=True)
else: else:
if not os.path.isdir(arguments['<path>']):
print('Error: Path must be a directory, not a file.')
sys.exit(1)
for _ in range(int(arguments['--maxtries'])): for _ in range(int(arguments['--maxtries'])):
t = TestCase.generate(arguments['<path>'] or '.') t = TestCase.generate(arguments['<path>'] or '.')
t.run(debugger, record) t.run(debugger, record)