Handle more edge cases with if and try block imports.

* Test mypy against Python 3.8 not 3.6

flake8_dunder_all/__init__.py

@@ -32,7 +32,7 @@
 # stdlib
 import ast
 import sys
-from typing import Any, Generator, List, Set, Tuple, Type, Union
+from typing import Any, Generator, Iterator, List, Set, Tuple, Type, Union
 
 # 3rd party
 from consolekit.terminal_colours import Fore
@@ -189,16 +189,59 @@ def visit_If(self, node: ast.If) -> None:
 
 		:param node: The node being visited.
 		"""
-		# Check if the condition is checking for TYPE_CHECKING
-		if isinstance(node.test, ast.Name) and node.test.id == "TYPE_CHECKING":
-			# Process the body of the if statement for any import statements
-			for body_node in node.body:
-				if isinstance(body_node, (ast.Import, ast.ImportFrom)):
-					self.handle_import(body_node)
-		# Continue visiting other nodes
+
+		if _is_type_checking(node.test):
+			if self.use_endlineno and node.end_lineno is not None:
+				self.last_import = max(self.last_import, node.end_lineno)
+			else:
+				self.last_import = max(self.last_import, max(_descend_node(node)))
+
+		self.generic_visit(node)
+
+	def visit_Try(self, node: ast.Try) -> None:
+		"""
+		Visit a Try statement.
+
+		:param node: The node being visited.
+		"""
+
+		if any(isinstance(n, (ast.Import, ast.ImportFrom)) for n in node.body):
+			if self.use_endlineno and node.end_lineno is not None and sys.implementation.name != "pypy":  # pragma: no cover (pypy)
+				self.last_import = max(self.last_import, node.end_lineno)
+			else:  # pragma: no cover (!pypy)
+				end_lineno = max(
+						*_descend_node(node),
+						*_descend_node(node, "handlers"),
+						*_descend_node(node, "orelse"),
+						*_descend_node(node, "finalbody"),
+						)
+				self.last_import = max(self.last_import, end_lineno)
+
 		self.generic_visit(node)
 
 
+def _descend_node(node, attr: str = "body") -> Iterator[int]:
+	for child in getattr(node, attr, []):
+		yield child.lineno
+		yield from _descend_node(child)
+
+
+_nameconstant = ast.Constant if sys.version_info >= (3, 8) else ast.NameConstant
+
+
+def _is_type_checking(node: ast.AST):
+	"""
+	Does the given ``if`` node indicate a `TYPE_CHECKING` block?
+	"""
+
+	if isinstance(node, ast.Name) and node.id == "TYPE_CHECKING":
+		return True
+	elif isinstance(node, _nameconstant) and node.value is False:
+		return True
+	elif isinstance(node, ast.BoolOp):
+		return any(_is_type_checking(value) for value in node.values)
+
+
 class Plugin:
 	"""
 	A Flake8 plugin which checks to ensure modules have defined ``__all__``.

tests/common.py

@@ -179,3 +179,39 @@ def a_function():
 
 def a_function(): ...
 """
+
+if_type_checking_else_source = """
+if False or sys.version_info < (3, 7):
+    import foo
+else:
+    import bar
+
+def a_function(): ...
+"""
+
+if_type_checking_try_source = """
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+	try:
+		from x import y
+	except ImportError:
+		pass
+
+def a_function(): ...
+"""
+
+if_type_checking_try_finally_source = """
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+	try:
+		from x import y
+	except ImportError:
+		pass
+	finally:
+		pass
+
+
+def a_function(): ...
+"""

tests/test_flake8_dunder_all.py

@@ -8,7 +8,10 @@
 import pytest
 from coincidence.regressions import AdvancedFileRegressionFixture
 from common import (
+		if_type_checking_else_source,
 		if_type_checking_source,
+		if_type_checking_try_finally_source,
+		if_type_checking_try_source,
 		mangled_source,
 		results,
 		testing_source_a,
@@ -125,6 +128,14 @@ def test_visitor(source: str, members: List[str], found_all: bool, last_import:
 				pytest.param(testing_source_j, ["a_function"], False, 14, id="multiline import"),
 				pytest.param(testing_source_m, ["a_function"], False, 7, id="if False"),
 				pytest.param(if_type_checking_source, ["a_function"], False, 5, id="if TYPE_CHECKING:"),
+				pytest.param(if_type_checking_else_source, ["a_function"], False, 5, id="if TYPE_CHECKING else"),
+				pytest.param(if_type_checking_try_source, ["a_function"], False, 8, id="if TYPE_CHECKING try"),
+				pytest.param(
+						if_type_checking_try_finally_source, ["a_function"],
+						False,
+						10,
+						id="if TYPE_CHECKING try finally"
+						),
 				]
 		)
 def test_visitor_endlineno(source: str, members: List[str], found_all: bool, last_import: int):
@@ -160,6 +171,9 @@ def test_visitor_endlineno(source: str, members: List[str], found_all: bool, las
 				pytest.param(testing_source_l, [], 0, id="typing.overload"),
 				pytest.param(testing_source_m, [], 1, id="if False"),
 				pytest.param(testing_source_n, [], 1, id="if TYPE_CHECKING"),
+				pytest.param(if_type_checking_else_source, [], 1, id="if TYPE_CHECKING else"),
+				pytest.param(if_type_checking_try_source, [], 1, id="if TYPE_CHECKING try"),
+				pytest.param(if_type_checking_try_finally_source, [], 1, id="if TYPE_CHECKING try finally"),
 				]
 		)
 def test_check_and_add_all(

tests/test_flake8_dunder_all_/test_check_and_add_all_if_TYPE_CHECKING_else_._py_

@@ -0,0 +1,10 @@
+
+if False or sys.version_info < (3, 7):
+    import foo
+else:
+    import bar
+
+__all__ = ["a_function"]
+
+
+def a_function(): ...

tests/test_flake8_dunder_all_/test_check_and_add_all_if_TYPE_CHECKING_try_._py_

@@ -0,0 +1,13 @@
+
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+	try:
+		from x import y
+	except ImportError:
+		pass
+
+__all__ = ["a_function"]
+
+
+def a_function(): ...

tests/test_flake8_dunder_all_/test_check_and_add_all_if_TYPE_CHECKING_try_finally_._py_

@@ -0,0 +1,15 @@
+
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+	try:
+		from x import y
+	except ImportError:
+		pass
+	finally:
+		pass
+
+__all__ = ["a_function"]
+
+
+def a_function(): ...