modernize code with ruff
All checks were successful
checks-impure / test (pull_request) Successful in 1m27s
checks / test (pull_request) Successful in 2m10s

This commit is contained in:
Jörg Thalheim 2023-12-06 17:13:32 +01:00
parent 8ea149de07
commit 807b53c71b
4 changed files with 9 additions and 16 deletions

View File

@ -1,13 +1,12 @@
import argparse
import json
import subprocess
from typing import Optional
from ..errors import ClanError
from ..machines.machines import Machine
def create_backup(machine: Machine, provider: Optional[str] = None) -> None:
def create_backup(machine: Machine, provider: str | None = None) -> None:
backup_scripts = json.loads(
machine.eval_nix(f"nixosConfigurations.{machine.name}.config.clanCore.backups")
)

View File

@ -1,13 +1,12 @@
import argparse
import pprint
from pathlib import Path
from typing import Optional
from ..errors import ClanError
def list_backups(
flake_dir: Path, machine: str, provider: Optional[str] = None
flake_dir: Path, machine: str, provider: str | None = None
) -> dict[str, dict[str, list[dict[str, str]]]]:
dummy_data = {
"testhostname": {

View File

@ -1,6 +1,5 @@
import argparse
from pathlib import Path
from typing import Optional
from ..errors import ClanError
@ -10,7 +9,7 @@ def restore_backup(
machine: str,
provider: str,
backup_id: str,
service: Optional[str] = None,
service: str | None = None,
) -> None:
if service is None:
print("would restore backup", machine, provider, backup_id)

View File

@ -4,7 +4,7 @@ import urllib.parse
from dataclasses import dataclass
from enum import Enum, member
from pathlib import Path
from typing import Dict, Self
from typing import Self
from .errors import ClanError
@ -52,7 +52,7 @@ class ClanURI:
if uri.startswith("clan://"):
self._nested_uri = uri[7:]
else:
raise ClanError("Invalid scheme: expected clan://, got {}".format(uri))
raise ClanError(f"Invalid scheme: expected clan://, got {uri}")
# Parse the URI into components
# scheme://netloc/path;parameters?query#fragment
@ -61,14 +61,12 @@ class ClanURI:
# Parse the query string into a dictionary
query = urllib.parse.parse_qs(self._components.query)
params: Dict[str, str] = {}
params: dict[str, str] = {}
for field in dataclasses.fields(ClanParameters):
if field.name in query:
values = query[field.name]
if len(values) > 1:
raise ClanError(
"Multiple values for parameter: {}".format(field.name)
)
raise ClanError(f"Multiple values for parameter: {field.name}")
params[field.name] = values[0]
# Remove the field from the query dictionary
@ -89,12 +87,10 @@ class ClanURI:
case "file":
self.scheme = ClanScheme.FILE.value(Path(self._components.path)) # type: ignore
case _:
raise ClanError(
"Unsupported scheme: {}".format(self._components.scheme)
)
raise ClanError(f"Unsupported scheme: {self._components.scheme}")
@classmethod
def from_path(cls, path: Path, params: ClanParameters) -> Self: # noqa
urlparams = urllib.parse.urlencode(params.__dict__)
return cls("clan://file://{}?{}".format(path, urlparams))
return cls(f"clan://file://{path}?{urlparams}")