forked from mirrors/gecko-dev
d024a31639
This patch was generated via: $ pyupgrade --py36-plus $(find taskcluster/taskgraph -name "*.py" -type f) $ autoflake --remove-all-unused-imports -i $(find taskcluster/taskgraph -name "*.py" -type f) The same set of commands are being applied to standalone taskgraph as well so they remain in sync. Differential Revision: https://phabricator.services.mozilla.com/D123234
53 lines
1.5 KiB
Python
53 lines
1.5 KiB
Python
# This Source Code Form is subject to the terms of the Mozilla Public
|
|
# License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
|
|
import inspect
|
|
import os
|
|
|
|
|
|
def find_object(path):
|
|
"""
|
|
Find a Python object given a path of the form <modulepath>:<objectpath>.
|
|
Conceptually equivalent to
|
|
|
|
def find_object(modulepath, objectpath):
|
|
import <modulepath> as mod
|
|
return mod.<objectpath>
|
|
"""
|
|
if path.count(":") != 1:
|
|
raise ValueError(f'python path {path!r} does not have the form "module:object"')
|
|
|
|
modulepath, objectpath = path.split(":")
|
|
obj = __import__(modulepath)
|
|
for a in modulepath.split(".")[1:]:
|
|
obj = getattr(obj, a)
|
|
for a in objectpath.split("."):
|
|
obj = getattr(obj, a)
|
|
return obj
|
|
|
|
|
|
def import_sibling_modules(exceptions=None):
|
|
"""
|
|
Import all Python modules that are siblings of the calling module.
|
|
|
|
Args:
|
|
exceptions (list): A list of file names to exclude (caller and
|
|
__init__.py are implicitly excluded).
|
|
"""
|
|
frame = inspect.stack()[1]
|
|
mod = inspect.getmodule(frame[0])
|
|
|
|
name = os.path.basename(mod.__file__)
|
|
excs = {"__init__.py", name}
|
|
if exceptions:
|
|
excs.update(exceptions)
|
|
|
|
modpath = mod.__name__
|
|
if not name.startswith("__init__.py"):
|
|
modpath = modpath.rsplit(".", 1)[0]
|
|
|
|
for f in os.listdir(os.path.dirname(mod.__file__)):
|
|
if f.endswith(".py") and f not in excs:
|
|
__import__(modpath + "." + f[:-3])
|