kopia lustrzana https://github.com/bellingcat/auto-archiver
40 wiersze
1.5 KiB
Python
40 wiersze
1.5 KiB
Python
from __future__ import annotations
|
|
from dataclasses import dataclass, field
|
|
from inspect import ClassFoundException
|
|
from typing import Type
|
|
from ..core import Metadata
|
|
from abc import ABC
|
|
# from collections.abc import Iterable
|
|
|
|
|
|
@dataclass
|
|
class Step(ABC):
|
|
name: str = None
|
|
|
|
def __init__(self, config: dict) -> None:
|
|
# reads the configs into object properties
|
|
# self.config = config[self.name]
|
|
for k, v in config.get(self.name, {}).items():
|
|
self.__setattr__(k, v)
|
|
|
|
@staticmethod
|
|
def configs() -> dict: {}
|
|
|
|
def init(name: str, config: dict, child: Type[Step]) -> Step:
|
|
"""
|
|
looks into direct subclasses of child for name and returns such ab object
|
|
TODO: cannot find subclasses of child.subclasses
|
|
"""
|
|
for sub in child.__subclasses__():
|
|
if sub.name == name:
|
|
return sub(config)
|
|
raise ClassFoundException(f"Unable to initialize STEP with {name=}, check your configuration file/step names, and make sure you made the step discoverable by putting it into __init__.py")
|
|
|
|
def assert_valid_string(self, prop: str) -> None:
|
|
"""
|
|
receives a property name an ensures it exists and is a valid non-empty string, raises an exception if not
|
|
"""
|
|
assert hasattr(self, prop), f"property {prop} not found"
|
|
s = getattr(self, prop)
|
|
assert s is not None and type(s) == str and len(s) > 0, f"invalid property {prop} value '{s}', it should be a valid string"
|