File size: 1,588 Bytes
a80f6e6 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 | 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
|