mirror of
https://github.com/hwchase17/langchain
synced 2024-11-10 01:10:59 +00:00
60 lines
1.6 KiB
Python
60 lines
1.6 KiB
Python
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
|
|
from .file import File
|
|
|
|
|
|
class Folder:
|
|
def __init__(self, name: str, *files: Folder | File) -> None:
|
|
self.name = name
|
|
self._files = files
|
|
|
|
@property
|
|
def files(self) -> list[Folder | File]:
|
|
return sorted(self._files, key=lambda f: f.name)
|
|
|
|
def create_structure(self, root: Path) -> None:
|
|
path = root / self.name
|
|
path.mkdir()
|
|
|
|
for file in self.files:
|
|
if isinstance(file, Folder):
|
|
file.create_structure(path)
|
|
else:
|
|
(path / file.name).write_text(file.content, encoding="utf-8")
|
|
|
|
@classmethod
|
|
def from_structure(cls, root: Path) -> Folder:
|
|
name = root.name
|
|
files: list[File | Folder] = []
|
|
|
|
for path in root.iterdir():
|
|
if path.is_dir():
|
|
files.append(cls.from_structure(path))
|
|
else:
|
|
files.append(
|
|
File(path.name, path.read_text(encoding="utf-8").splitlines())
|
|
)
|
|
|
|
return Folder(name, *files)
|
|
|
|
def __eq__(self, __value: object) -> bool:
|
|
if isinstance(__value, File):
|
|
return False
|
|
|
|
if not isinstance(__value, Folder):
|
|
return NotImplemented
|
|
|
|
if self.name != __value.name:
|
|
return False
|
|
|
|
if len(self.files) != len(__value.files):
|
|
return False
|
|
|
|
for self_file, other_file in zip(self.files, __value.files):
|
|
if self_file != other_file:
|
|
return False
|
|
|
|
return True
|