-
Notifications
You must be signed in to change notification settings - Fork 0
/
classes.py
32 lines (25 loc) · 880 Bytes
/
classes.py
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
from abc import ABC, abstractmethod
from datetime import datetime
from functools import reduce
class Entry(ABC):
def __init__(self, name: str, size: int, created: datetime, modified: datetime):
self.name = name
self.size = size
self.created = created
self.modified = modified
pass
class File(Entry):
pass
class Directory(Entry):
def __init__(self, name: str, size: int, created: datetime, modified: datetime):
super().__init__(name, size, created, modified)
self.files = []
self.directories = []
@property
def total_size(self) -> int:
return (
reduce(lambda acc, file: acc + file.size, self.files, 0) +
reduce(lambda acc, dir: acc + dir.total_size, self.directories, 0)
)
def items(self) -> list[Entry]:
return self.files + self.directories