Skip to content

Core functionality for the sophys-cli project, a commandline interface for Bluesky orchestration at SIRIUS

License

Notifications You must be signed in to change notification settings

cnpem/sophys-cli-core

Repository files navigation

sophys-cli-core

A command-line client for the sophys project group.

This is the core project of the client, providing basic and common functionality for all extensions.

Installation

To use it, you'll have to be in a valid Python environment (consider using micromamba). In there, you'll need to do the following:

Normal installation (TODO: Create pre-built packages):

$ pip install git+https://github.com/cnpem/sophys-cli-core.git

Developer installation:

$ cd <path where you will clone the sophys-cli package>
$ git clone https://github.com/cnpem/sophys-cli-core.git
$ pip install -e sophys-cli-core

With that, you'll have access to the sophys-cli command in the environment you installed it. Furthermore, to use sophys-cli with a particular beamline configuration, you must also install the sophys-<beamline> package in that environment. After that, to use that configuration, see the Usage section.

Usage

With the package installed, you can launch it via a terminal configured in the proper environment, like so:

$ sophys-cli <arguments>

The allowed arguments are printed with the -h/--help flags, like so:

$ sophys-cli -h
usage: sophys-cli [-h] [--debug] [--local] [--test] [--nocolor] extension

positional arguments:
  extension    The extension to load the configuration from.

options:
  -h, --help  show this help message and exit
  --debug     Configure debug mode, with more verbose logging and error messgaes.
  --local     Use a local RunEngine instead of communicating with HTTPServer.
  --test      Setup testing configurations to test the tool without interfering with production configured parameters.
  --nocolor   Remove color codes from rich output.

Running commands and scripts automatically

When using pure IPython, we have arguments like -c, -m and -i that allow us to run pre-created routines and code, enabling some form of higher-level automation. In sophys-cli, we currently support the -c and -i flags, working in the same way they would do in IPython (it's actually exactly passing it on for IPython to deal with it!).

You can use -c directly to run a single line of prompt, or you can use the %run magic to run an entire script.

As an example of that second case, we could have a file called my_routine.ipy in /home/cool_user/, with the following:

print("Starting linear scans...")

for i in range(10, 100, 10):
  %scan -d SIM_det -m SIM_motor -2 2 --num $i

print("Performing grid scan...")

%grid_scan -d SIM_det4 -m SIM_motor1 -2 2 10 SIM_motor2 -2 2 10

print("Scans completed successfully. Have a nice day!")

Then, running that whole procedure non-interactively is a matter of calling:

sophys-cli <extension> -c "%run /home/cool_user/my_routine.ipy"

Development

Creating your own extension

To create your own extension and make it acessible via the application, it is necessary that the created package have an IPython entrypoint (a function called load_ipython_extension) in the python import prefix sophys.cli.extensions.<extension name>. This is to ensure backwards-compatibility with the monorepo era.

After doing so, and having the package installed in your environment, running sophys-cli <extension name> [args] ought to work as intended.

Inside that entrypoint, you can do whatever you want, but generally you'll want to configure variables in the user namespace for usage during the program lifetime, and set up magics for user convenience.

Communicating with httpserver

One of the main features of this package in the option of transparently communicating with httpserver instead of using a local RunEngine. To do so, we can use the RemoteSessionHandler class from the http_utils module, with automatically handles authentication and session management for us.

Using it should be as simple as importing setup_remote_session_handler from the sophys.cli.core.magics module, and calling it on your extension entrypoint with the ipython object and httpserver address as arguments.

Besides the session management bits, we also have many pre-assembled magics for interacting with the remote server. These are located in the sophys.cli.extensions.tools_magics module, under the HTTPMagics class.

To use that, we must register the class magics, like one would normally do in IPython (ipython.register_magics(HTTPMagics)), and we can also configure a class property, pertaining to the reload_plans specifically, which can use a plan whitelist object to filter out plans available on the server, based on the extension configuration, like so:

from sophys.cli.core.magics.plan_magics import PlanInformation, PlanWhitelist
from sophys.cli.core.magics.plan_magics import PlanMV, PlanReadMany, PlanCount

whitelisted_plan_list = [
    PlanInformation("mov", "mov", PlanMV, has_detectors=False),
    PlanInformation("read_many", "read", PlanReadMany, has_detectors=False),
    PlanInformation("count", "count", PlanCount),
    ...
]

plan_whitelist = PlanWhitelist(*whitelisted_plan_list)

ipython.register_magics(HTTPMagics)
ipython.magics_manager.registry["HTTPMagics"].plan_whitelist = plan_whitelist

Code architecture: Startup flow

pyproject.toml

All begins in the generated entrypoint generated by setuptools at installation-time.

__main__.py

From there, we go to the __main__ file, where we find the entrypoint function. There, the first thing done is establishing the available flags and arguments for the program, using argparse.

Right after that, we make a call to the create_kernel function, whose purpose is to configure the IPython kernel we'll be using. Here is where we'll hook up the code to be executed later to properly configure our environment and extensions, via configuration of the extensions property.

This is a separate function because we customize the kernel in testing in order to have an easier time with IPython internals.

Then we start the IPython kernel.

base_configuration.py

From here, the core extension for sophys-cli is executed first. This will be defined in the base_configuration.py file, responsible for triggering the configuration of all basic functionality of sophys-cli applications. The executed function, load_ipython_extension, will call execute_at_start, which will instantiate callbacks, monitors, devices, and other useful bits, depending on the mode of operation configured via the CLI interface (local or remote (default)).

The instantiated functions and objects are kept in the global IPython scope for later use and retrieval. This is done via the helper functions get_from_namespace and add_to_namespace, to make keeping track of these items easier. The global scope is later propagated to extensions via the ipython object.

Extensions

With that execution concluded, the other extensions configured earlier in the extensions IPython property will be loaded. This means they'll execute their load_ipython_extension function too. See the Creating your own extension section for more details on that.

This concludes our little tour! ^.^

About

Core functionality for the sophys-cli project, a commandline interface for Bluesky orchestration at SIRIUS

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages