modernize code with ruff
This commit is contained in:
@@ -1,13 +1,12 @@
|
|||||||
import argparse
|
import argparse
|
||||||
import json
|
import json
|
||||||
import subprocess
|
import subprocess
|
||||||
from typing import Optional
|
|
||||||
|
|
||||||
from ..errors import ClanError
|
from ..errors import ClanError
|
||||||
from ..machines.machines import Machine
|
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(
|
backup_scripts = json.loads(
|
||||||
machine.eval_nix(f"nixosConfigurations.{machine.name}.config.clanCore.backups")
|
machine.eval_nix(f"nixosConfigurations.{machine.name}.config.clanCore.backups")
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -1,13 +1,12 @@
|
|||||||
import argparse
|
import argparse
|
||||||
import pprint
|
import pprint
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
from typing import Optional
|
|
||||||
|
|
||||||
from ..errors import ClanError
|
from ..errors import ClanError
|
||||||
|
|
||||||
|
|
||||||
def list_backups(
|
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]]]]:
|
) -> dict[str, dict[str, list[dict[str, str]]]]:
|
||||||
dummy_data = {
|
dummy_data = {
|
||||||
"testhostname": {
|
"testhostname": {
|
||||||
|
|||||||
@@ -1,6 +1,5 @@
|
|||||||
import argparse
|
import argparse
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
from typing import Optional
|
|
||||||
|
|
||||||
from ..errors import ClanError
|
from ..errors import ClanError
|
||||||
|
|
||||||
@@ -10,7 +9,7 @@ def restore_backup(
|
|||||||
machine: str,
|
machine: str,
|
||||||
provider: str,
|
provider: str,
|
||||||
backup_id: str,
|
backup_id: str,
|
||||||
service: Optional[str] = None,
|
service: str | None = None,
|
||||||
) -> None:
|
) -> None:
|
||||||
if service is None:
|
if service is None:
|
||||||
print("would restore backup", machine, provider, backup_id)
|
print("would restore backup", machine, provider, backup_id)
|
||||||
|
|||||||
@@ -4,7 +4,7 @@ import urllib.parse
|
|||||||
from dataclasses import dataclass
|
from dataclasses import dataclass
|
||||||
from enum import Enum, member
|
from enum import Enum, member
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
from typing import Dict, Self
|
from typing import Self
|
||||||
|
|
||||||
from .errors import ClanError
|
from .errors import ClanError
|
||||||
|
|
||||||
@@ -52,7 +52,7 @@ class ClanURI:
|
|||||||
if uri.startswith("clan://"):
|
if uri.startswith("clan://"):
|
||||||
self._nested_uri = uri[7:]
|
self._nested_uri = uri[7:]
|
||||||
else:
|
else:
|
||||||
raise ClanError("Invalid scheme: expected clan://, got {}".format(uri))
|
raise ClanError(f"Invalid scheme: expected clan://, got {uri}")
|
||||||
|
|
||||||
# Parse the URI into components
|
# Parse the URI into components
|
||||||
# scheme://netloc/path;parameters?query#fragment
|
# scheme://netloc/path;parameters?query#fragment
|
||||||
@@ -61,14 +61,12 @@ class ClanURI:
|
|||||||
# Parse the query string into a dictionary
|
# Parse the query string into a dictionary
|
||||||
query = urllib.parse.parse_qs(self._components.query)
|
query = urllib.parse.parse_qs(self._components.query)
|
||||||
|
|
||||||
params: Dict[str, str] = {}
|
params: dict[str, str] = {}
|
||||||
for field in dataclasses.fields(ClanParameters):
|
for field in dataclasses.fields(ClanParameters):
|
||||||
if field.name in query:
|
if field.name in query:
|
||||||
values = query[field.name]
|
values = query[field.name]
|
||||||
if len(values) > 1:
|
if len(values) > 1:
|
||||||
raise ClanError(
|
raise ClanError(f"Multiple values for parameter: {field.name}")
|
||||||
"Multiple values for parameter: {}".format(field.name)
|
|
||||||
)
|
|
||||||
params[field.name] = values[0]
|
params[field.name] = values[0]
|
||||||
|
|
||||||
# Remove the field from the query dictionary
|
# Remove the field from the query dictionary
|
||||||
@@ -89,12 +87,10 @@ class ClanURI:
|
|||||||
case "file":
|
case "file":
|
||||||
self.scheme = ClanScheme.FILE.value(Path(self._components.path)) # type: ignore
|
self.scheme = ClanScheme.FILE.value(Path(self._components.path)) # type: ignore
|
||||||
case _:
|
case _:
|
||||||
raise ClanError(
|
raise ClanError(f"Unsupported scheme: {self._components.scheme}")
|
||||||
"Unsupported scheme: {}".format(self._components.scheme)
|
|
||||||
)
|
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def from_path(cls, path: Path, params: ClanParameters) -> Self: # noqa
|
def from_path(cls, path: Path, params: ClanParameters) -> Self: # noqa
|
||||||
urlparams = urllib.parse.urlencode(params.__dict__)
|
urlparams = urllib.parse.urlencode(params.__dict__)
|
||||||
|
|
||||||
return cls("clan://file://{}?{}".format(path, urlparams))
|
return cls(f"clan://file://{path}?{urlparams}")
|
||||||
|
|||||||
Reference in New Issue
Block a user