mirror of
https://github.com/Garmelon/PFERD.git
synced 2023-12-21 10:23:01 +01:00
43 lines
1.0 KiB
Python
43 lines
1.0 KiB
Python
"""
|
|
Transforms let the user define functions to decide where the downloaded files
|
|
should be placed locally. They let the user do more advanced things like moving
|
|
only files whose names match a regex, or renaming files from one numbering
|
|
scheme to another.
|
|
"""
|
|
|
|
from dataclasses import dataclass
|
|
from pathlib import Path
|
|
from typing import Callable, List, Optional, TypeVar
|
|
|
|
|
|
Transform = Callable[[Path], Optional[Path]]
|
|
|
|
|
|
@dataclass
|
|
class Transformable:
|
|
"""
|
|
An object that can be transformed by a Transform.
|
|
"""
|
|
|
|
path: Path
|
|
|
|
|
|
TF = TypeVar("TF", bound=Transformable)
|
|
|
|
|
|
def apply_transform(
|
|
transform: Callable[[Path], Optional[Path]],
|
|
transformables: List[TF],
|
|
) -> List[TF]:
|
|
"""
|
|
Apply a Transform to multiple Transformables, discarding those that were
|
|
not transformed by the Transform.
|
|
"""
|
|
|
|
result: List[TF] = []
|
|
for transformable in transformables:
|
|
if new_path := transform(transformable.path):
|
|
transformable.path = new_path
|
|
result.append(transformable)
|
|
return result
|