83 lines
2.2 KiB
Python
83 lines
2.2 KiB
Python
#!/usr/bin/env python
|
|
|
|
from eco import ecocnf
|
|
from eco.utilities.config import Terminal
|
|
import sys
|
|
|
|
|
|
import argparse
|
|
|
|
parser = argparse.ArgumentParser(description="eco startup utility")
|
|
|
|
parser.add_argument(
|
|
"-s",
|
|
"--scope",
|
|
type=str,
|
|
default=None,
|
|
help="scope name, usually instrument or beamline",
|
|
)
|
|
parser.add_argument(
|
|
"-a",
|
|
"--scopes_available",
|
|
action="store_true",
|
|
default=False,
|
|
help="print available scopes.",
|
|
)
|
|
parser.add_argument(
|
|
"-l", "--lazy", action="store_true", default=False, help="lazy initialisation"
|
|
)
|
|
parser.add_argument(
|
|
"--shell", action="store_true", default=False, help="open eco in ipython shell"
|
|
)
|
|
parser.add_argument(
|
|
"--pylab", type=bool, default=True, help="open ipython shell in pylab mode"
|
|
)
|
|
|
|
arguments = parser.parse_args()
|
|
|
|
scope = arguments.scope
|
|
# scope = 'bernina'
|
|
|
|
if arguments.scopes_available:
|
|
print("{:<15s}{:<15s}{:<15s}".format("module", "name", "facility"))
|
|
for ts in ecocnf.scopes:
|
|
print(
|
|
" {:<14s} {:<14s} {:<14s}".format(ts["module"], ts["name"], ts["facility"])
|
|
)
|
|
|
|
|
|
print(
|
|
" ___ _______\n / -_) __/ _ \ \n Experiment Control \__/\__/\___/ \n\n"
|
|
)
|
|
|
|
term = Terminal(scope=scope)
|
|
|
|
if scope:
|
|
# import importlib
|
|
# eco = importlib.import_module('eco')
|
|
# mdl = importlib.import_module(scope,package=eco)
|
|
# mdl = importlib.import_module('eco.bernina')
|
|
if arguments.lazy:
|
|
ecocnf.startup_lazy = True
|
|
exec(f"import eco.{scope} as {scope}")
|
|
exec(f"op = {scope}.init()")
|
|
for tk, tv in op.items():
|
|
sys.modules["__main__"].__dict__[tk] = tv
|
|
# exec(f'{scope}.init(lazy=ecocnf.startup_lazy)')
|
|
# exec(f"from eco.{scope} import *")
|
|
# is there an __all__? if so respect it
|
|
# if "__all__" in mdl.__dict__:
|
|
# names = mdl.__dict__["__all__"]
|
|
# else:
|
|
# otherwise we import all names that don't begin with _
|
|
# names = [x for x in mdl.__dict__ if not x.startswith("_")]
|
|
# now drag them in
|
|
# globals().update({k: getattr(mdl, k) for k in names})
|
|
|
|
term.set_title()
|
|
from IPython import get_ipython
|
|
|
|
_ipy = get_ipython()
|
|
_ipy.Completer.use_jedi = False
|
|
print(arguments)
|