QuLab 2.11.0__tar.gz → 2.11.1__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: QuLab
-Version: 2.11.0
+Version: 2.11.1
 Summary: contral instruments and manage data
 Author-email: feihoo87 <feihoo87@gmail.com>
 Maintainer-email: feihoo87 <feihoo87@gmail.com>

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: QuLab
-Version: 2.11.0
+Version: 2.11.1
 Summary: contral instruments and manage data
 Author-email: feihoo87 <feihoo87@gmail.com>
 Maintainer-email: feihoo87 <feihoo87@gmail.com>

{qulab-2.11.0 → qulab-2.11.1}/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.1}/qulab/executor/cli.py

@@ -163,6 +163,83 @@ def get(key, api):
     click.echo(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()
 @click.option('--bootstrap',
               '-b',

qulab-2.11.1/qulab/version.py

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

qulab-2.11.0/qulab/version.py

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