summaryrefslogtreecommitdiff
path: root/lib/sqlalchemy/util
diff options
context:
space:
mode:
authorMike Bayer <mike_mp@zzzcomputing.com>2019-01-03 22:28:09 -0500
committerMike Bayer <mike_mp@zzzcomputing.com>2019-01-03 22:28:09 -0500
commitd229360a8d4071c2f150558897f37e13eb09f430 (patch)
tree99f639d804f760a9d4936bfec26a323cc4c41187 /lib/sqlalchemy/util
parent1eaf9dc7776b9833a9fb62fe630b4b9ac63a31f4 (diff)
downloadsqlalchemy-d229360a8d4071c2f150558897f37e13eb09f430.tar.gz
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
Diffstat (limited to 'lib/sqlalchemy/util')
-rw-r--r--lib/sqlalchemy/util/langhelpers.py4
1 files changed, 3 insertions, 1 deletions
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