QuLab 2.11.0__tar.gz → 2.11.2__tar.gz

{qulab-2.11.0 → qulab-2.11.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: QuLab
-Version: 2.11.0
+Version: 2.11.2
 Summary: contral instruments and manage data
 Author-email: feihoo87 <feihoo87@gmail.com>
 Maintainer-email: feihoo87 <feihoo87@gmail.com>
@@ -47,6 +47,7 @@ Requires-Dist: SQLAlchemy>=2.0.19
 Requires-Dist: watchdog>=4.0.0
 Requires-Dist: wath>=1.1.6
 Requires-Dist: waveforms>=1.9.4
+Requires-Dist: rich>=14.0.0
 Provides-Extra: full
 Requires-Dist: uvloop>=0.19.0; extra == "full"
 Dynamic: license-file

{qulab-2.11.0 → qulab-2.11.2}/QuLab.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: QuLab
-Version: 2.11.0
+Version: 2.11.2
 Summary: contral instruments and manage data
 Author-email: feihoo87 <feihoo87@gmail.com>
 Maintainer-email: feihoo87 <feihoo87@gmail.com>
@@ -47,6 +47,7 @@ Requires-Dist: SQLAlchemy>=2.0.19
 Requires-Dist: watchdog>=4.0.0
 Requires-Dist: wath>=1.1.6
 Requires-Dist: waveforms>=1.9.4
+Requires-Dist: rich>=14.0.0
 Provides-Extra: full
 Requires-Dist: uvloop>=0.19.0; extra == "full"
 Dynamic: license-file

{qulab-2.11.0 → qulab-2.11.2}/QuLab.egg-info/requires.txt

@@ -21,6 +21,7 @@ SQLAlchemy>=2.0.19
 watchdog>=4.0.0
 wath>=1.1.6
 waveforms>=1.9.4
+rich>=14.0.0
 
 [full]
 uvloop>=0.19.0

{qulab-2.11.0 → qulab-2.11.2}/pyproject.toml

@@ -50,7 +50,8 @@ dependencies = [
     "SQLAlchemy>=2.0.19",
     "watchdog>=4.0.0",
     "wath>=1.1.6",
-    "waveforms>=1.9.4"
+    "waveforms>=1.9.4",
+    "rich>=14.0.0",
 ]
 dynamic = ["version"]
 

{qulab-2.11.0 → qulab-2.11.2}/qulab/cli/commands.py

@@ -1,6 +1,7 @@
 import click
 
-from ..executor.cli import boot, create, get, maintain, reproduce, run, set
+from ..executor.cli import (boot, create, export, get, load, maintain,
+                            reproduce, run, set)
 from ..monitor.__main__ import main as monitor
 from ..scan.server import server
 from ..sys.net.cli import dht
@@ -18,6 +19,12 @@ def hello():
     click.echo('hello, world')
 
 
+@cli.group()
+def reg():
+    """Regestry operations."""
+    pass
+
+
 cli.add_command(monitor)
 cli.add_command(plot)
 cli.add_command(dht)
@@ -26,6 +33,8 @@ cli.add_command(maintain)
 cli.add_command(run)
 cli.add_command(reproduce)
 cli.add_command(create)
-cli.add_command(set)
-cli.add_command(get)
+reg.add_command(set)
+reg.add_command(get)
+reg.add_command(load)
+reg.add_command(export)
 cli.add_command(boot)

{qulab-2.11.0 → qulab-2.11.2}/qulab/executor/cli.py

@@ -6,6 +6,7 @@ import sys
 from pathlib import Path
 
 import click
+import rich
 from loguru import logger
 
 from ..cli.config import get_config_value, log_options
@@ -160,7 +161,84 @@ def get(key, api):
         api = importlib.import_module(api)
         set_config_api(api.query_config, api.update_config, api.delete_config,
                        api.export_config, api.clear_config)
-    click.echo(reg.get(key))
+    rich.print(reg.get(key))
+
+
+@click.command()
+@click.argument('file')
+@click.option('--api',
+              '-a',
+              default=lambda: get_config_value("api", str, 'get'),
+              help='The modlule name of the api.')
+@click.option('--format',
+              '-f',
+              default='json',
+              help='The format of the config.')
+@log_options('export')
+def export(file, api, format):
+    """
+    Export a config.
+    """
+    logger.info(f'[CMD]: export {file} --api {api}')
+    reg = Registry()
+    if api is not None:
+        api = importlib.import_module(api)
+        set_config_api(api.query_config, api.update_config, api.delete_config,
+                       api.export_config, api.clear_config)
+    cfg = reg.export()
+    if format == 'json':
+        import json
+        with open(file, 'w') as f:
+            json.dump(cfg, f, indent=4)
+    elif format == 'yaml':
+        import yaml
+        with open(file, 'w') as f:
+            yaml.dump(cfg, f)
+    elif format == 'pickle':
+        import pickle
+        with open(file, 'wb') as f:
+            pickle.dump(cfg, f)
+    else:
+        raise ValueError(f'Unknown format: {format}')
+
+
+@click.command()
+@click.argument('file')
+@click.option('--api',
+              '-a',
+              default=lambda: get_config_value("api", str, 'get'),
+              help='The modlule name of the api.')
+@click.option('--format',
+              '-f',
+              default='json',
+              help='The format of the config.')
+@log_options('load')
+def load(file, api, format):
+    """
+    Load a config.
+    """
+    logger.info(f'[CMD]: load {file} --api {api}')
+    reg = Registry()
+    if api is not None:
+        api = importlib.import_module(api)
+        set_config_api(api.query_config, api.update_config, api.delete_config,
+                       api.export_config, api.clear_config)
+    if format == 'json':
+        import json
+        with open(file, 'r') as f:
+            cfg = json.load(f)
+    elif format == 'yaml':
+        import yaml
+        with open(file, 'r') as f:
+            cfg = yaml.load(f, Loader=yaml.FullLoader)
+    elif format == 'pickle':
+        import pickle
+        with open(file, 'rb') as f:
+            cfg = pickle.load(f)
+    else:
+        raise ValueError(f'Unknown format: {format}')
+    reg.clear()
+    reg.update(cfg)
 
 
 @click.command()

qulab-2.11.2/qulab/version.py

@@ -0,0 +1 @@
+__version__ = "2.11.2"

qulab-2.11.0/qulab/version.py

@@ -1 +0,0 @@
-__version__ = "2.11.0"