Accessing submodules

Is there any reason not to allow access to Python submodules from the orbit module?

import orbit

bunch = orbit.core.Bunch()
lattice = orbit.teapot.TEAPOT_Lattice()
...

Currently we have to write:

from orbit.core.bunch import Bunch
from orbit.teapot import TEAPOT_Lattice
...

bunch = Bunch()
lattice = TEAPOT_Lattice()
...

Related issue #16

All this requires is the following lines in py/orbit/__init__.py

from . import core
from . import teapot

And maybe py/orbit/core/__init__.py

from .bunch import Bunch

Thinking about it, it would kind of make sense to me to export Bunch and maybe other commonly used objects directly from the orbit module:

from .core.bunch import Bunch

Then you would be able to

bunch = orbit.Bunch()

I made PR #158.

I agree that it would make sense to haveBunch accessible directly from orbit import, instead of orbit.core.bunch.Bunch.

Navigation Menu