style: black formatter
This commit is contained in:
@@ -6,6 +6,7 @@ from typing import TYPE_CHECKING
|
|||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from .xray_manager_config import XrayManagerConfig
|
from .xray_manager_config import XrayManagerConfig
|
||||||
|
|
||||||
|
|
||||||
@dataclass(frozen=True)
|
@dataclass(frozen=True)
|
||||||
class Target:
|
class Target:
|
||||||
id: str
|
id: str
|
||||||
@@ -24,25 +25,25 @@ class TargetStorage:
|
|||||||
def load_by_id(self, target_id: str) -> Target:
|
def load_by_id(self, target_id: str) -> Target:
|
||||||
if target_id not in self.target_dict:
|
if target_id not in self.target_dict:
|
||||||
raise ValueError(f"Target not found: {target_id}")
|
raise ValueError(f"Target not found: {target_id}")
|
||||||
|
|
||||||
target_data = self.target_dict[target_id]
|
target_data = self.target_dict[target_id]
|
||||||
|
|
||||||
if "route" not in target_data:
|
if "route" not in target_data:
|
||||||
raise ValueError(
|
raise ValueError(
|
||||||
f"Missing required 'route' field for target: {target_id}"
|
f"Missing required 'route' field for target: {target_id}"
|
||||||
)
|
)
|
||||||
|
|
||||||
return Target(
|
return Target(
|
||||||
id=target_id,
|
id=target_id,
|
||||||
pretty=target_data.get("pretty_name", target_id),
|
pretty=target_data.get("pretty_name", target_id),
|
||||||
route=target_data["route"]
|
route=target_data["route"],
|
||||||
)
|
)
|
||||||
|
|
||||||
def load_by_pretty(self, target_pretty_name: str) -> Target:
|
def load_by_pretty(self, target_pretty_name: str) -> Target:
|
||||||
for target_id, target_data in self.target_dict.items():
|
for target_id, target_data in self.target_dict.items():
|
||||||
if target_data.get("pretty_name", target_id) == target_pretty_name:
|
if target_data.get("pretty_name", target_id) == target_pretty_name:
|
||||||
return self.load_by_id(target_id)
|
return self.load_by_id(target_id)
|
||||||
|
|
||||||
raise ValueError(f"Target pretty name not found: {target_pretty_name}")
|
raise ValueError(f"Target pretty name not found: {target_pretty_name}")
|
||||||
|
|
||||||
def load_all_targets(self) -> list[Target]:
|
def load_all_targets(self) -> list[Target]:
|
||||||
@@ -57,5 +58,5 @@ class TargetStorage:
|
|||||||
def delete_target(self, target: Target):
|
def delete_target(self, target: Target):
|
||||||
if target not in self.load_all_targets():
|
if target not in self.load_all_targets():
|
||||||
raise ValueError(f"Target not found: {target.id}")
|
raise ValueError(f"Target not found: {target.id}")
|
||||||
|
|
||||||
self.xrmc.delete_target(target.id)
|
self.xrmc.delete_target(target.id)
|
||||||
|
|||||||
Reference in New Issue
Block a user