Move imports required for typing under the TYPE_CHECKING block

beets/autotag/hooks.py

@@ -17,18 +17,20 @@
 from __future__ import annotations
 
 import re
-from collections.abc import Iterable, Iterator
 from functools import total_ordering
-from typing import Any, Callable, NamedTuple, TypeVar, cast
+from typing import Any, Callable, NamedTuple, TypeVar, cast, TYPE_CHECKING
 
 from jellyfish import levenshtein_distance
 from unidecode import unidecode
 
 from beets import config, logging, plugins
 from beets.autotag import mb
-from beets.library import Item
 from beets.util import as_string, cached_classproperty
 
+if TYPE_CHECKING:
+    from beets.library import Item
+    from collections.abc import Iterable, Iterator
+
 log = logging.getLogger("beets")
 
 V = TypeVar("V")

beets/autotag/match.py

@@ -22,7 +22,7 @@
 import re
 from collections.abc import Iterable, Sequence
 from enum import IntEnum
-from typing import Any, NamedTuple, TypeVar, Union, cast
+from typing import Any, NamedTuple, TypeVar, Union, cast, TYPE_CHECKING
 
 from munkres import Munkres
 
@@ -35,9 +35,11 @@
     TrackMatch,
     hooks,
 )
-from beets.library import Item
 from beets.util import plurality
 
+if TYPE_CHECKING:
+    from beets.library import Item
+
 # Artist signals that indicate "various artists". These are used at the
 # album level to determine whether a given release is likely a VA
 # release and also on the track level to to remove the penalty for

beets/dbcore/db.py

@@ -26,8 +26,7 @@
 from collections import defaultdict
 from collections.abc import Generator, Iterable, Iterator, Mapping, Sequence
 from sqlite3 import Connection
-from types import TracebackType
-from typing import Any, AnyStr, Callable, Generic, TypeVar, cast
+from typing import Any, AnyStr, Callable, Generic, TypeVar, cast, TYPE_CHECKING
 
 from unidecode import unidecode
 
@@ -45,6 +44,9 @@
     TrueQuery,
 )
 
+if TYPE_CHECKING:
+    from types import TracebackType
+
 
 class DBAccessError(Exception):
     """The SQLite database became inaccessible.

beets/dbcore/queryparse.py

@@ -18,10 +18,13 @@
 
 import itertools
 import re
-from collections.abc import Collection, Sequence
 
 from . import Model, query
-from .query import Sort
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from .query import Sort
+    from collections.abc import Collection, Sequence
 
 PARSE_QUERY_PART_REGEX = re.compile(
     # Non-capturing optional segment for the keyword.

beets/util/__init__.py

@@ -28,7 +28,6 @@
 import tempfile
 import traceback
 from collections import Counter
-from collections.abc import Iterator, Sequence
 from contextlib import suppress
 from enum import Enum
 from importlib import import_module
@@ -50,6 +49,7 @@
 from beets.util import hidden
 
 if TYPE_CHECKING:
+    from collections.abc import Iterator, Sequence
     from logging import Logger
 
 if sys.version_info >= (3, 10):

beetsplug/autobpm.py

@@ -19,10 +19,13 @@
 
 import librosa
 
-from beets.importer import ImportTask
-from beets.library import Item, Library
 from beets.plugins import BeetsPlugin
 from beets.ui import Subcommand, should_write
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from beets.library import Item, Library
+    from beets.importer import ImportTask
 
 
 class AutoBPMPlugin(BeetsPlugin):

beetsplug/replaygain.py

@@ -18,29 +18,31 @@
 import collections
 import enum
 import math
-import optparse
 import os
 import queue
 import signal
 import subprocess
 import sys
 import warnings
 from abc import ABC, abstractmethod
-from collections.abc import Sequence
 from dataclasses import dataclass
-from logging import Logger
 from multiprocessing.pool import ThreadPool
 from threading import Event, Thread
-from typing import Any, Callable, TypeVar, cast
+from typing import Any, Callable, TypeVar, cast, TYPE_CHECKING
 
-from confuse import ConfigView
 
 from beets import ui
-from beets.importer import ImportSession, ImportTask
-from beets.library import Album, Item, Library
 from beets.plugins import BeetsPlugin
 from beets.util import command_output, displayable_path, syspath
 
+if TYPE_CHECKING:
+    from confuse import ConfigView
+    from beets.importer import ImportSession, ImportTask
+    from beets.library import Album, Item, Library
+    from logging import Logger
+    from collections.abc import Sequence
+    import optparse
+
 # Utilities.
 
 

pyproject.toml

@@ -238,6 +238,7 @@ select = [
     "PT", # flake8-pytest-style
     # "RUF", # ruff
     # "UP", # pyupgrade
+    "TCH", # flake8-type-checking
     "W", # pycodestyle
 ]
 [tool.ruff.lint.per-file-ignores]

test/plugins/test_hook.py

@@ -18,13 +18,15 @@
 import os.path
 import sys
 import unittest
-from collections.abc import Iterator
 from contextlib import contextmanager
-from typing import Callable
+from typing import Callable, TYPE_CHECKING
 
 from beets import plugins
 from beets.test.helper import PluginTestCase, capture_log
 
+if TYPE_CHECKING:
+    from collections.abc import Iterator
+
 
 class HookTestCase(PluginTestCase):
     plugin = "hook"