Add configuration construction tooling
This commit is contained in:
parent
4cf26b861b
commit
891009014d
105
drone.py
105
drone.py
@ -19,7 +19,7 @@ class Config:
|
||||
repo_init: bool = True
|
||||
repo_init_append_only: bool = False
|
||||
repo_init_quota: Optional[str] = None
|
||||
repo_init_encryption: str = ""
|
||||
repo_init_encryption: Optional[str] = None
|
||||
repo_init_overwrite_key: bool = False
|
||||
repo_key_file: pathlib.Path = pathlib.Path("/keys")
|
||||
repo_passphrase: Optional[str] = None
|
||||
@ -42,14 +42,103 @@ class Config:
|
||||
|
||||
@classmethod
|
||||
def build(cls):
|
||||
parameters = {}
|
||||
for conf, field in cls.__dataclass_fields__.items():
|
||||
env = f"BORGDRONE_{conf.upper()}"
|
||||
if env in os.environ:
|
||||
if field.type == bool:
|
||||
parameters[conf] = cls._parse_bool(env, os.environ[env])
|
||||
prefix = "BORGDRONE_"
|
||||
|
||||
return cls(**parameters)
|
||||
try:
|
||||
return cls(
|
||||
repo_path=pathlib.Path(os.environ[f"{prefix}REPO_PATH"]),
|
||||
archive_path=[
|
||||
pathlib.Path(item.strip())
|
||||
for item in os.environ[f"{prefix}ARCHIVE_PATH"].split(",")
|
||||
if item
|
||||
],
|
||||
compact=os.environ.get(f"{prefix}COMPACT", str(cls.compact)).lower()
|
||||
== "true",
|
||||
check=os.environ.get(f"{prefix}CHECK", str(cls.check)).lower()
|
||||
== "true",
|
||||
repo_init=os.environ.get(
|
||||
f"{prefix}REPO_INIT", str(cls.repo_init)
|
||||
).lower()
|
||||
== "true",
|
||||
repo_init_append_only=os.environ.get(
|
||||
f"{prefix}REPO_INIT_APPEND_ONLY", str(cls.repo_init_append_only)
|
||||
).lower()
|
||||
== "true",
|
||||
repo_init_quota=os.environ[f"{prefix}REPO_INIT_QUOTA"]
|
||||
if f"{prefix}REPO_INIT_QUOTA" in os.environ
|
||||
else cls.repo_init_quota,
|
||||
repo_init_encryption=os.environ[f"{prefix}REPO_INIT_ENCRYPTION"]
|
||||
if f"{prefix}REPO_INIT_ENCRYPTION" in os.environ
|
||||
else cls.repo_init_encryption,
|
||||
repo_init_overwrite_key=os.environ.get(
|
||||
f"{prefix}REPO_INIT_OVERWRITE_KEY", str(cls.repo_init_overwrite_key)
|
||||
).lower()
|
||||
== "true",
|
||||
repo_key_file=pathlib.Path(os.environ[f"{prefix}REPO_KEY_FILE"])
|
||||
if f"{prefix}REPO_KEY_FILE" in os.environ
|
||||
else cls.repo_key_file,
|
||||
repo_passphrase=os.environ[f"{prefix}REPO_PASSPHRASE"]
|
||||
if f"{prefix}REPO_PASSPHRASE" in os.environ
|
||||
else cls.repo_passphrase,
|
||||
repo_passphrase_file=pathlib.Path(
|
||||
os.environ[f"{prefix}REPO_PASSPHRASE_FILE"]
|
||||
)
|
||||
if f"{prefix}REPO_PASSPHRASE_FILE" in os.environ
|
||||
else cls.repo_passphrase_file,
|
||||
prune=os.environ.get(f"{prefix}PRUNE", str(cls.prune)).lower()
|
||||
== "true",
|
||||
prune_interval=datetime.timedelta(
|
||||
seconds=int(os.environ[f"{prefix}PRUNE_INTERVAL"])
|
||||
)
|
||||
if f"{prefix}PRUNE_INTERVAL" in os.environ
|
||||
else cls.prune_interval,
|
||||
prune_secondly=int(
|
||||
os.environ.get(f"{prefix}PRUNE_SECONDLY", cls.prune_secondly)
|
||||
),
|
||||
prune_minutely=int(
|
||||
os.environ.get(f"{prefix}PRUNE_MINUTELY", cls.prune_minutely)
|
||||
),
|
||||
prune_hourly=int(
|
||||
os.environ.get(f"{prefix}PRUNE_HOURLY", cls.prune_hourly)
|
||||
),
|
||||
prune_daily=int(
|
||||
os.environ.get(f"{prefix}PRUNE_DAILY", cls.prune_daily)
|
||||
),
|
||||
prune_weekly=int(
|
||||
os.environ.get(f"{prefix}PRUNE_WEEKLY", cls.prune_weekly)
|
||||
),
|
||||
prune_monthly=int(
|
||||
os.environ.get(f"{prefix}PRUNE_MONTHLY", cls.prune_monthly)
|
||||
),
|
||||
prune_yearly=int(
|
||||
os.environ.get(f"{prefix}PRUNE_YEARLY", cls.prune_yearly)
|
||||
),
|
||||
archive_name=os.environ[f"{prefix}ARCHIVE_NAME"]
|
||||
if f"{prefix}ARCHIVE_NAME" in os.environ
|
||||
else cls.archive_name,
|
||||
archive_comment=os.environ[f"{prefix}ARCHIVE_COMMENT"]
|
||||
if f"{prefix}ARCHIVE_COMMENT" in os.environ
|
||||
else cls.archive_comment,
|
||||
archive_exclude_pattern=os.environ[f"{prefix}ARCHIVE_EXCLUDE_PATTERN"]
|
||||
if f"{prefix}ARCHIVE_EXCLUDE_PATTERN" in os.environ
|
||||
else cls.archive_exclude_pattern,
|
||||
archive_exclude_caches=os.environ.get(
|
||||
f"{prefix}ARCHIVE_EXCLUDE_CACHES", str(cls.archive_exclude_caches)
|
||||
).lower()
|
||||
== "true",
|
||||
archive_exclude_if_present=os.environ[
|
||||
f"{prefix}ARCHIVE_EXCLUDE_IF_PRESENT"
|
||||
]
|
||||
if f"{prefix}ARCHIVE_EXCLUDE_IF_PRESENT" in os.environ
|
||||
else cls.archive_exclude_if_present,
|
||||
archive_keep_exclude_tags=os.environ.get(
|
||||
f"{prefix}ARCHIVE_KEEP_EXCLUDE_TAGS",
|
||||
str(cls.archive_keep_exclude_tags),
|
||||
).lower()
|
||||
== "true",
|
||||
)
|
||||
except KeyError as err:
|
||||
raise RuntimeError(f"Required configuration parameter {err} not provided")
|
||||
|
||||
@staticmethod
|
||||
def _parse_bool(env: str, value: str) -> bool:
|
||||
|
Loading…
Reference in New Issue
Block a user