Enabling explicit relative imports in main modules the easy way.
For enabling relative import in __main__
module PEP 366 presents a workaround like:
if __name__ == "__main__" and __package__ is None:
__package__ = "my_pkg"
from .foo import bar
With rel_imp
you can rewrite it as:
import rel_imp; rel_imp.init()
from .foo import bar
Cleaner, faster and less coupled. (you don't need to specify the package manually)
Note: In order to use rel_imp
the module you are coding must be inside a package or a sub-package.
Python 2 still supports implicit relative import and will be deprecated in Python 3, so you will want to migrate those scripts using implicit relative import to explicit relative import. Check more on the PEP 404.
Explicit relative imports makes your code less coupled. As PEP 328 says:
Several use cases were presented, the most important of which is being able to rearrange the structure of large packages without having to edit sub-packages. In addition, a module inside a package can't easily import itself without relative imports.
On the other hand PEP 8 says:
Absolute imports are recommended, as they are usually more readable and tend to be better behaved (or at least give better error messages) if the import system is incorrectly configured.
If you find a point to use explicit relative imports, then you can use rel_imp to allow running packages as main.
Some reasons:
- A submodule can become a command line tool if called as main.
- You can have unit test or smoke test within the module.
- You simply want to run it without any explicit test to see if at least it imports everything it needs.
pip install rel_imp
Remove it with:
pip uninstall rel_imp
Download the rel_imp.py
in one your Python's search path.
wget https://raw.githubusercontent.com/joaduo/rel_imp/master/rel_imp.py
Imagine you have 2 modules inside a package called "my_pkg".
That would be:
my_pkg/__init__.py
my_pkg/math_lib.py
my_pkg/test.py
So in test.py we could have
from .math_lib import factorize
def factorize_and_print(number):
num = factorize(number)
print num
if __name__ == '__main__':
#Small smoke test
factorize_and_print(10)
factorize_and_print(0)
factorize_and_print(-10)
If you do python my_pkg/test.py
it will throw an exception because of the relative import at the first line.
So you can use rel_imp
to make your code look nicer. Simply do:
import rel_imp; rel_imp.init()
from .math_lib import factorize
It is equivalent as the prior solution but you don't have to worry about keeping in sync __package__
's value.
You can also use a shorter solution with implicit init()
using the relative_import
module:
import relative_import
from .math_lib import factorize
It uses the same technique in PEP 366 but __package__
's value is set through dynamic inspection of the stack. To solve the value of __package__
it compares the current __main__
's file with search paths in sys.path - or, optionally, a list of paths given in the settings -.
For example, for a file in /home/user/projects/python/math/my_pkg/test.py
given the following paths in sys.path:
[
'/home/user/projects/python/',
'/home/user/projects/python/math/',
'/home/user/projects/python/math/my_pkg/',
]
It will pick the closest path to the __main__
's file that is not the __main__
's file's directory.
Then the base path use to solve __package__
variable will be /home/user/projects/python/math/
This process is only done once and for the __main__
. Succesive calls to rel_imp.init()
does nothing.
You can simply define a environment variable PYTHON_DISABLE_REL_IMP
with any value.
File an issue through github's issue tracking system.
You can optionally contact me at joaduo gmail com.