Skip to content

Commit eb2c41c

Browse files
committed
fix(lint): resolve errors from merged PR #14 (see issue #8)
1 parent 291ff56 commit eb2c41c

File tree

3 files changed

+19
-15
lines changed

3 files changed

+19
-15
lines changed

qbreader/_api_utils.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -106,7 +106,7 @@ def normalize_subcat(unnormalized_subcats: UnnormalizedCategory):
106106

107107
def category_correspondence(
108108
typed_alt_subcat: AlternateSubcategory,
109-
) -> Tuple[Category, Subcategory]:
109+
) -> Tuple[Category | None, Subcategory | None]:
110110
if typed_alt_subcat in [
111111
AlternateSubcategory.ASTRONOMY,
112112
AlternateSubcategory.COMPUTER_SCIENCE,
@@ -146,13 +146,16 @@ def category_correspondence(
146146
AlternateSubcategory.MISC_LITERATURE,
147147
]:
148148
return (Category.LITERATURE, None)
149+
150+
# Accounts for AlternateSubcategory.PRACTICES and AlternateSubcategory.BELIEFS
151+
return (None, None)
149152

150153

151154
def normalize_cats(
152155
unnormalized_cats: UnnormalizedCategory,
153156
unnormalized_subcats: UnnormalizedSubcategory,
154157
unnormalized_alt_subcats: UnnormalizedAlternateSubcategory,
155-
) -> Tuple[Category, Subcategory, AlternateSubcategory]:
158+
) -> Tuple[str, str, str]:
156159
"""
157160
Normalize a single or list of categories, subcategories, and alternate_subcategories
158161
to their corresponding comma-separated strings, taking into account categories and
@@ -183,8 +186,8 @@ def normalize_cats(
183186
elif isinstance(unnormalized_cats, str):
184187
final_cats = [Category(unnormalized_cats), *to_be_pushed_cats]
185188
elif isinstance(unnormalized_cats, Iterable):
186-
for subcat in unnormalized_cats:
187-
final_cats.append(Subcategory(subcat))
189+
for cat in unnormalized_cats:
190+
final_cats.append(Category(cat))
188191
final_cats.append(*to_be_pushed_cats)
189192

190193
final_subcats = []

qbreader/synchronous.py

Lines changed: 8 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -283,19 +283,20 @@ def random_bonus(
283283
Parameters
284284
----------
285285
difficulties : qbreader.types.UnnormalizedDifficulty, optional
286-
The difficulties to search for. Can be a single or an array of `Difficulty`
287-
enums, strings, or integers.
286+
The difficulties to search for. Can be a single or an array of
287+
`Difficulty` enums, strings, or integers.
288288
categories : qbreader.types.UnnormalizedCategory, optional
289-
The categories to search for. Can be a single or an array of `Category`
290-
enums or strings.
289+
The categories to search for. Can be a single or an array of
290+
`Category` enums or strings.
291291
subcategories : qbreader.types.UnnormalizedSubcategory, optional
292292
The subcategories to search for. Can be a single or an array of
293293
`Subcategory` enums or strings. The API does not check for consistency
294294
between categories and subcategories.
295-
alternate_subcategories: qbreaader.types.UnnormalizedAlternateSubcategory, optional
295+
alternate_subcategories : qbreaader.types.UnnormalizedAlternateSubcategory,
296+
optional
296297
The alternates subcategories to search for. Can be a single or an array of
297-
`AlternateSubcategory` enum variants or strings. The API does not check for consistency
298-
between categories, subcategories, and alternate subcategories.
298+
`AlternateSubcategory` enum variants or strings. The API does not check for
299+
consistency between categories, subcategories, and alternate subcategories.
299300
number : int, default = 1
300301
The number of bonuses to return.
301302
min_year : int, default = Year.MIN_YEAR

qbreader/types.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -290,7 +290,7 @@ def __init__(
290290
self.packet: PacketMetadata = packet
291291
self.set: SetMetadata = set
292292
self.number: int = number
293-
self.alternate_subcategory: AlternateSubcategory = alternate_subcategory
293+
self.alternate_subcategory: AlternateSubcategory | None = alternate_subcategory
294294

295295
@classmethod
296296
def from_json(cls: Type[Self], json: dict[str, Any]) -> Self:
@@ -384,7 +384,7 @@ def __init__(
384384
self.set: SetMetadata = set
385385
self.packet: PacketMetadata = packet
386386
self.number: int = number
387-
self.alternate_subcategory: AlternateSubcategory = alternate_subcategory
387+
self.alternate_subcategory: AlternateSubcategory | None = alternate_subcategory
388388
self.values: Optional[tuple[int, ...]] = tuple(values) if values else None
389389
self.difficultyModifiers: Optional[tuple[DifficultyModifier, ...]] = (
390390
tuple(difficultyModifiers) if difficultyModifiers else None
@@ -687,8 +687,8 @@ def __str__(self) -> str:
687687
UnnormalizedAlternateSubcategory: TypeAlias = Optional[
688688
Union[AlternateSubcategory, str, Iterable[Union[AlternateSubcategory, str]]]
689689
]
690-
"""Type alias for unnormalized alternate subcategories. Union of `AlternateSubcategory`, `str`, and
691-
`collections.abc.Iterable` containing either."""
690+
"""Type alias for unnormalized alternate subcategories. Union of `AlternateSubcategory`,
691+
`str`, and `collections.abc.Iterable` containing either."""
692692

693693

694694
__all__ = (

0 commit comments

Comments
 (0)