#!/usr/bin/env python

## pylab activity >>>>
import numpy
import matplotlib
from matplotlib import pylab, mlab, pyplot

np = numpy
plt = pyplot

from IPython.core.pylabtools import figsize, getfigs

from pylab import *
from numpy import *

plt.ion()
## pylab activity <<<<


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"from eco.{scope} import *")
    # 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 = True
# print(arguments)
# _ipy.magic("load_ext rich")