Skip to content

Commit

Permalink
cast for mypy
Browse files Browse the repository at this point in the history
  • Loading branch information
chaen committed Jan 24, 2024
1 parent 421bacf commit 6ac8ff5
Showing 1 changed file with 10 additions and 4 deletions.
14 changes: 10 additions & 4 deletions diracx-cli/src/diracx/cli/internal/legacy.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,12 +3,16 @@
import json
import os
from pathlib import Path
from typing import cast
from typing import TYPE_CHECKING, cast
from urllib.parse import urljoin, urlparse

import diraccfg
import typer
import yaml

if TYPE_CHECKING:
from diraccfg.cfg import CFGAsDict

from pydantic import BaseModel
from typer import Option

Expand Down Expand Up @@ -59,12 +63,14 @@ def cs_sync(old_file: Path, new_file: Path):
cfg = diraccfg.CFG().loadFromBuffer(old_data)
raw = cfg.getAsDict()

diracx_section = raw["DiracX"]
diracx_section = cast("CFGAsDict", raw["DiracX"])
# DisabledVOs cannot be set if any Legacy clients are enabled
disabled_vos = diracx_section.get("DisabledVOs")
enabled_clients = []
for _, client_status in diracx_section.get("LegacyClientEnabled", {}).items():
for _, str_status in client_status.items():
for _, client_status in cast(
"CFGAsDict", diracx_section.get("LegacyClientEnabled", {})
).items():
for _, str_status in cast("CFGAsDict", client_status).items():
enabled_clients.append(str_status == "True")
if disabled_vos and any(enabled_clients):
raise RuntimeError(
Expand Down

0 comments on commit 6ac8ff5

Please sign in to comment.