|
2 | 2 | from dataclasses import asdict, dataclass, field |
3 | 3 | from enum import Enum |
4 | 4 | from pathlib import Path |
5 | | -from typing import Any, Optional, Union |
| 5 | +from typing import Any |
6 | 6 |
|
7 | 7 | from tomlkit import TOMLDocument, dumps, load, table |
8 | 8 |
|
@@ -38,55 +38,55 @@ class TwynConfiguration: |
38 | 38 | """Method for selecting similar packages.""" |
39 | 39 | allowlist: set[str] |
40 | 40 | """Set of package names to allow without checking.""" |
41 | | - pypi_source: Optional[str] |
| 41 | + pypi_source: str | None |
42 | 42 | """Alternative PyPI source URL.""" |
43 | | - npm_source: Optional[str] |
| 43 | + npm_source: str | None |
44 | 44 | """Alternative npm source URL.""" |
45 | 45 | use_cache: bool |
46 | 46 | """Whether to use cached trusted packages.""" |
47 | | - package_ecosystem: Optional[PackageEcosystems] |
| 47 | + package_ecosystem: PackageEcosystems | None |
48 | 48 | """Target package ecosystem for analysis.""" |
49 | | - recursive: Optional[bool] |
| 49 | + recursive: bool | None |
50 | 50 | """Whether to recursively search for dependency files.""" |
51 | 51 |
|
52 | 52 |
|
53 | 53 | @dataclass |
54 | 54 | class ReadTwynConfiguration: |
55 | 55 | """Configuration for twyn as set by the user. It may have None values.""" |
56 | 56 |
|
57 | | - dependency_files: Optional[set[str]] = field(default_factory=set) |
| 57 | + dependency_files: set[str] | None = field(default_factory=set) |
58 | 58 | """Optional set of dependency file paths to analyze.""" |
59 | | - selector_method: Optional[str] = None |
| 59 | + selector_method: str | None = None |
60 | 60 | """Optional method for selecting similar packages.""" |
61 | 61 | allowlist: set[str] = field(default_factory=set) |
62 | 62 | """Set of package names to allow without checking.""" |
63 | | - pypi_source: Optional[str] = None |
| 63 | + pypi_source: str | None = None |
64 | 64 | """Optional alternative PyPI source URL.""" |
65 | | - npm_source: Optional[str] = None |
| 65 | + npm_source: str | None = None |
66 | 66 | """Optional alternative npm source URL.""" |
67 | | - use_cache: Optional[bool] = None |
| 67 | + use_cache: bool | None = None |
68 | 68 | """Optional setting for using cached trusted packages.""" |
69 | | - package_ecosystem: Optional[PackageEcosystems] = None |
| 69 | + package_ecosystem: PackageEcosystems | None = None |
70 | 70 | """Optional target package ecosystem for analysis.""" |
71 | | - recursive: Optional[bool] = None |
| 71 | + recursive: bool | None = None |
72 | 72 | """Optional setting for recursive dependency file search.""" |
73 | 73 |
|
74 | 74 |
|
75 | 75 | class ConfigHandler: |
76 | 76 | """Manage reading and writing configurations for Twyn.""" |
77 | 77 |
|
78 | | - def __init__(self, file_handler: Optional[FileHandler] = None) -> None: |
| 78 | + def __init__(self, file_handler: FileHandler | None = None) -> None: |
79 | 79 | self.file_handler = file_handler |
80 | 80 |
|
81 | 81 | def resolve_config( # noqa: C901, PLR0912 |
82 | 82 | self, |
83 | | - selector_method: Optional[str] = None, |
84 | | - dependency_files: Optional[set[str]] = None, |
85 | | - use_cache: Optional[bool] = None, |
86 | | - package_ecosystem: Optional[PackageEcosystems] = None, |
87 | | - recursive: Optional[bool] = None, |
88 | | - pypi_source: Optional[str] = None, |
89 | | - npm_source: Optional[str] = None, |
| 83 | + selector_method: str | None = None, |
| 84 | + dependency_files: set[str] | None = None, |
| 85 | + use_cache: bool | None = None, |
| 86 | + package_ecosystem: PackageEcosystems | None = None, |
| 87 | + recursive: bool | None = None, |
| 88 | + pypi_source: str | None = None, |
| 89 | + npm_source: str | None = None, |
90 | 90 | ) -> TwynConfiguration: |
91 | 91 | """Resolve the configuration for Twyn. |
92 | 92 |
|
@@ -244,10 +244,10 @@ def get_default_config_file_path() -> str: |
244 | 244 | return DEFAULT_PROJECT_TOML_FILE |
245 | 245 |
|
246 | 246 |
|
247 | | -def _serialize_config(x: Any) -> Union[Any, str, list[Any]]: |
| 247 | +def _serialize_config(x: Any) -> Any | str | list[Any]: |
248 | 248 | """Serialize configuration values for TOML format.""" |
249 | 249 |
|
250 | | - def _value_to_for_config(v: Any) -> Union[str, list[Any], Any]: |
| 250 | + def _value_to_for_config(v: Any) -> str | list[Any] | Any: |
251 | 251 | if isinstance(v, Enum): |
252 | 252 | return v.name |
253 | 253 | elif isinstance(v, set): |
|
0 commit comments