diff options
| author | Mike Bayer <mike_mp@zzzcomputing.com> | 2019-01-03 22:28:09 -0500 |
|---|---|---|
| committer | Mike Bayer <mike_mp@zzzcomputing.com> | 2019-01-03 22:28:09 -0500 |
| commit | d229360a8d4071c2f150558897f37e13eb09f430 (patch) | |
| tree | 99f639d804f760a9d4936bfec26a323cc4c41187 /lib/sqlalchemy/util | |
| parent | 1eaf9dc7776b9833a9fb62fe630b4b9ac63a31f4 (diff) | |
| download | sqlalchemy-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.py | 4 |
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 |
