VimPlug/jedi-fork
1
0
Fork 0
forked from VimPlug/jedi
Code Pull Requests Activity

Added FolderIO.walk and FolderIO.get_base_name

Browse Source
This commit is contained in:
Dave Halter
2020-01-20 00:36:18 +01:00
parent 26f0fa9eb0
commit 62a77dcd16
2 changed files with 40 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
jedi/file_io.py
View File

@@ -21,6 +21,9 @@ class AbstractFolderIO(object):
class FolderIO(AbstractFolderIO):
def get_base_name(self):
return os.path.basename(self.path)
def list(self):
return os.listdir(self.path)
@@ -30,6 +33,24 @@ class FolderIO(AbstractFolderIO):
def get_parent_folder(self):
return FolderIO(os.path.dirname(self.path))
def walk(self):
for root, dirs, files in os.walk(self.path):
root_folder_io = FolderIO(root)
original_folder_ios = [FolderIO(os.path.join(root, d)) for d in dirs]
modified_folder_ios = list(original_folder_ios)
yield (
root_folder_io,
modified_folder_ios,
[FileIO(os.path.join(root, f)) for f in files],
)
modified_iterator = iter(modified_folder_ios)
current = next(modified_iterator, None)
for i, folder_io in enumerate(original_folder_ios):
if current is folder_io:
current = next(modified_iterator, None)
else:
del dirs[i]
class FileIOFolderMixin(object):
def get_parent_folder(self):