VimPlug/jedi
1
0
Fork 1
mirror of https://github.com/davidhalter/jedi.git synced 2025-12-10 07:41:51 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Make sith.py random work with files

Browse Source
This commit is contained in:
Aaron Meurer
2013-08-10 10:39:19 -06:00
parent f2ffa037da
commit b6b510693b
1 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
sith.py
View File

@@ -61,7 +61,9 @@ 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.' if not os.path.isdir(file_path):
yield file_path
return
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'):
@@ -208,9 +210,6 @@ 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)