From d229360a8d4071c2f150558897f37e13eb09f430 Mon Sep 17 00:00:00 2001 From: Mike Bayer Date: Thu, 3 Jan 2019 22:28:09 -0500 Subject: Prep for flake8 refactoring a few code changes ahead of time to handle some __all__ issues better. also include new flake8 rules, since the existing flake8 doesn't pass in any case. Change-Id: I1efdf75124ae7bcac719c22e505bb5b13db06c04 --- lib/sqlalchemy/util/langhelpers.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'lib/sqlalchemy/util') diff --git a/lib/sqlalchemy/util/langhelpers.py b/lib/sqlalchemy/util/langhelpers.py index 6a1db7a98..7e387f4f2 100644 --- a/lib/sqlalchemy/util/langhelpers.py +++ b/lib/sqlalchemy/util/langhelpers.py @@ -853,7 +853,7 @@ class MemoizedSlots(object): return self._fallback_getattr(key) -def dependency_for(modulename): +def dependency_for(modulename, add_to_all=False): def decorate(obj): # TODO: would be nice to improve on this import silliness, # unfortunately importlib doesn't work that great either @@ -862,6 +862,8 @@ def dependency_for(modulename): ".".join(tokens[0:-1]), globals(), locals(), [tokens[-1]]) mod = getattr(mod, tokens[-1]) setattr(mod, obj.__name__, obj) + if add_to_all and hasattr(mod, "__all__"): + mod.__all__.append(obj.__name__) return obj return decorate -- cgit v1.2.1