You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
62 lines
2.2 KiB
62 lines
2.2 KiB
2 months ago
|
import re
|
||
|
|
from pathlib import Path
|
||
|
|
from pathlib import WindowsPath
|
||
|
|
from typing import Optional, List
|
||
|
|
|
||
|
|
|
||
|
|
class DirectionTree:
|
||
|
|
def __init__(self,
|
||
|
|
direction_name: str = 'WorkingDirection',
|
||
|
|
direction_path: str = '.',
|
||
|
|
ignore_list: Optional[List[str]] = None):
|
||
|
|
self.owner: WindowsPath = Path(direction_path)
|
||
|
|
self.tree: str = direction_name + '/\n'
|
||
|
|
self.ignore_list = ignore_list
|
||
|
|
if ignore_list is None:
|
||
|
|
self.ignore_list = []
|
||
|
|
self.direction_ergodic(path_object=self.owner, n=0)
|
||
|
|
|
||
|
|
def tree_add(self, path_object: WindowsPath, n=0, last=False):
|
||
|
|
if n > 0:
|
||
|
|
if last:
|
||
|
|
self.tree += '│' + (' │' * (n - 1)) + ' └────' + path_object.name
|
||
|
|
else:
|
||
|
|
self.tree += '│' + (' │' * (n - 1)) + ' ├────' + path_object.name
|
||
|
|
else:
|
||
|
|
if last:
|
||
|
|
self.tree += '└' + ('──' * 2) + path_object.name
|
||
|
|
else:
|
||
|
|
self.tree += '├' + ('──' * 2) + path_object.name
|
||
|
|
if path_object.is_file():
|
||
|
|
self.tree += '\n'
|
||
|
|
return False
|
||
|
|
elif path_object.is_dir():
|
||
|
|
self.tree += '/\n'
|
||
|
|
return True
|
||
|
|
|
||
|
|
def filter_file(self, file):
|
||
|
|
for item in self.ignore_list:
|
||
|
|
if re.fullmatch(item, file.name):
|
||
|
|
return False
|
||
|
|
return True
|
||
|
|
|
||
|
|
def direction_ergodic(self, path_object: WindowsPath, n=0):
|
||
|
|
dir_file: list = list(path_object.iterdir())
|
||
|
|
dir_file.sort(key=lambda x: x.name.lower())
|
||
|
|
dir_file = [f for f in filter(self.filter_file, dir_file)]
|
||
|
|
for i, item in enumerate(dir_file):
|
||
|
|
if i + 1 == len(dir_file):
|
||
|
|
if self.tree_add(item, n, last=True):
|
||
|
|
self.direction_ergodic(item, n + 1)
|
||
|
|
else:
|
||
|
|
if self.tree_add(item, n, last=False):
|
||
|
|
self.direction_ergodic(item, n + 1)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
i_l = [
|
||
|
|
'\.git', '__pycache__', 'test.+', 'venv', '.+\.whl', '\.idea', '.+\.jpg', '.+\.png',
|
||
|
|
'image', 'css', 'admin', 'db.sqlite3'
|
||
|
|
]
|
||
|
|
tree = DirectionTree(ignore_list=i_l, direction_path='./')
|
||
|
|
print(tree.tree)
|