fix remaining lint and type problems

Author: geoffrey-wu

qbreader/_api_utils.py

@@ -4,17 +4,17 @@
 
 import warnings
 from enum import Enum, EnumType
-from typing import Iterable, Optional, Union, Tuple
+from typing import Iterable, Optional, Tuple, Union
 
 from qbreader.types import (
-    Difficulty,
+    AlternateSubcategory,
     Category,
+    Difficulty,
     Subcategory,
-    AlternateSubcategory,
+    UnnormalizedAlternateSubcategory,
     UnnormalizedCategory,
     UnnormalizedDifficulty,
     UnnormalizedSubcategory,
-    UnnormalizedAlternateSubcategory,
 )
 
 
@@ -107,6 +107,7 @@ def normalize_subcat(unnormalized_subcats: UnnormalizedCategory):
 def category_correspondence(
     typed_alt_subcat: AlternateSubcategory,
 ) -> Tuple[Category | None, Subcategory | None]:
+    """Return the corresponding category/subcategory for a alternate_subcategory."""
     if typed_alt_subcat in [
         AlternateSubcategory.ASTRONOMY,
         AlternateSubcategory.COMPUTER_SCIENCE,
@@ -146,7 +147,7 @@ def category_correspondence(
         AlternateSubcategory.MISC_LITERATURE,
     ]:
         return (Category.LITERATURE, None)
-    
+
     # Accounts for AlternateSubcategory.PRACTICES and AlternateSubcategory.BELIEFS
     return (None, None)
 
@@ -156,12 +157,10 @@ def normalize_cats(
     unnormalized_subcats: UnnormalizedSubcategory,
     unnormalized_alt_subcats: UnnormalizedAlternateSubcategory,
 ) -> Tuple[str, str, str]:
-    """
-    Normalize a single or list of categories, subcategories, and alternate_subcategories
-    to their corresponding comma-separated strings, taking into account categories and
-    subcategories that must be added for the alternate_subcategories to work.
-    """
-
+    """Normalize a single or list of categories, subcategories, and\
+    alternate_subcategories to their corresponding comma-separated strings, taking into\
+    account categories and subcategories that must be added for the\
+    alternate_subcategories to work."""
     typed_alt_subcats: list[AlternateSubcategory] = []
 
     if isinstance(unnormalized_alt_subcats, str):
@@ -186,8 +185,8 @@ def normalize_cats(
     elif isinstance(unnormalized_cats, str):
         final_cats = [Category(unnormalized_cats), *to_be_pushed_cats]
     elif isinstance(unnormalized_cats, Iterable):
-        for cat in unnormalized_cats:
-            final_cats.append(Category(cat))
+        for unnormalized_cat in unnormalized_cats:
+            final_cats.append(Category(unnormalized_cat))
         final_cats.append(*to_be_pushed_cats)
 
     final_subcats = []
@@ -196,8 +195,8 @@ def normalize_cats(
     elif isinstance(unnormalized_subcats, str):
         final_subcats = [Subcategory(unnormalized_subcats), *to_be_pushed_subcats]
     elif isinstance(unnormalized_subcats, Iterable):
-        for subcat in unnormalized_subcats:
-            final_subcats.append(Subcategory(subcat))
+        for unnormalized_subcat in unnormalized_subcats:
+            final_subcats.append(Subcategory(unnormalized_subcat))
         final_subcats.append(*to_be_pushed_subcats)
 
     return (

qbreader/asynchronous.py

@@ -16,10 +16,10 @@
     QuestionType,
     SearchType,
     Tossup,
+    UnnormalizedAlternateSubcategory,
     UnnormalizedCategory,
     UnnormalizedDifficulty,
     UnnormalizedSubcategory,
-    UnnormalizedAlternateSubcategory,
     Year,
 )
 
@@ -116,7 +116,8 @@ class type.
             The subcategories to search for. Can be a single or an array of
             `Subcategory` enums or strings. The API does not check for consistency
             between categories and subcategories.
-        alternate_subcategories : qbreader.types.UnnormalizedAlternateSubcategory, optional
+        alternate_subcategories : qbreader.types.UnnormalizedAlternateSubcategory,\
+            optional
             The alternate subcategories to search for. Can be a single or an array of
             `AlternateSubcategory` enums or strings. The API does not check for
             consistency between categories and subcategories
@@ -244,7 +245,8 @@ async def random_tossup(
             The subcategories to search for. Can be a single or an array of
             `Subcategory` enums or strings. The API does not check for consistency
             between categories and subcategories.
-        alternate_subcategories : qbreader.types.UnnormalizedAlternateSubcategory, optional
+        alternate_subcategories : qbreader.types.UnnormalizedAlternateSubcategory,\
+            optional
             The alternate subcategories to search for. Can be a single or an array of
             `AlternateSubcategory` enums or strings. The API does not check for
             consistency between categories and subcategories
@@ -327,10 +329,11 @@ async def random_bonus(
             The subcategories to search for. Can be a single or an array of
             `Subcategory` enums or strings. The API does not check for consistency
             between categories and subcategories.
-        alternate_subcategories: qbreaader.types.UnnormalizedAlternateSubcategory, optional
+        alternate_subcategories: qbreader.types.UnnormalizedAlternateSubcategory, \
+            optional
             The alternates subcategories to search for. Can be a single or an array of
-            `AlternateSubcategory` enum variants or strings. The API does not check for consistency
-            between categories, subcategories, and alternate subcategories.
+            `AlternateSubcategory` enum variants or strings. The API does not check for
+            consistency between categories, subcategories, and alternate subcategories.
         number : int, default = 1
             The number of bonuses to return.
         min_year : int, default = Year.MIN_YEAR

qbreader/synchronous.py

@@ -16,10 +16,10 @@
     QuestionType,
     SearchType,
     Tossup,
-    UnnormalizedDifficulty,
+    UnnormalizedAlternateSubcategory,
     UnnormalizedCategory,
+    UnnormalizedDifficulty,
     UnnormalizedSubcategory,
-    UnnormalizedAlternateSubcategory,
     Year,
 )
 
@@ -81,10 +81,11 @@ class type.
             The subcategories to search for. Can be a single or an array of
             `Subcategory` enums or strings. The API does not check for consistency
             between categories and subcategories.
-        alternate_subcategories: qbreaader.types.UnnormalizedAlternateSubcategory, optional
+        alternate_subcategories: qbreaader.types.UnnormalizedAlternateSubcategory,\
+            optional
             The alternates subcategories to search for. Can be a single or an array of
-            `AlternateSubcategory` enum variants or strings. The API does not check for consistency
-            between categories, subcategories, and alternate subcategories.
+            `AlternateSubcategory` enum variants or strings. The API does not check for
+            consistency between categories, subcategories, and alternate subcategories.
         maxReturnLength : int, default = 25
             The maximum number of questions to return.
         tossupPagination : int, default = 1
@@ -209,10 +210,11 @@ def random_tossup(
             The subcategories to search for. Can be a single or an array of
             `Subcategory` enums or strings. The API does not check for consistency
             between categories and subcategories.
-        alternate_subcategories: qbreaader.types.UnnormalizedAlternateSubcategory, optional
+        alternate_subcategories: qbreaader.types.UnnormalizedAlternateSubcategory,
+            optional
             The alternates subcategories to search for. Can be a single or an array of
-            `AlternateSubcategory` enum variants or strings. The API does not check for consistency
-            between categories, subcategories, and alternate subcategories.
+            `AlternateSubcategory` enum variants or strings. The API does not check for
+            consistency between categories, subcategories, and alternate subcategories.
         number : int, default = 1
             The number of tossups to return.
         min_year : int, default = Year.MIN_YEAR

qbreader/types.py

@@ -688,7 +688,7 @@ def __str__(self) -> str:
 UnnormalizedAlternateSubcategory: TypeAlias = Optional[
     Union[AlternateSubcategory, str, Iterable[Union[AlternateSubcategory, str]]]
 ]
-"""Type alias for unnormalized alternate subcategories. Union of `AlternateSubcategory`, 
+"""Type alias for unnormalized alternate subcategories. Union of `AlternateSubcategory`,
 `str`, and `collections.abc.Iterable` containing either."""