QuLab 2.10.12__py3-none-any.whl → 2.11.1__py3-none-any.whl

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/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/executor/load.py

@@ -3,6 +3,7 @@ import inspect
 import os
 import pickle
 import shutil
+import sys
 import tempfile
 import time
 import warnings
@@ -322,6 +323,8 @@ def load_workflow_from_file(file_name: str,
     source_code = (base_path / path).read_text()
     module.__source__ = source_code
 
+    sys.modules[module_name] = module
+
     if hasattr(module, 'entries'):
         if veryfy_source_code:
             verify_entries(module, base_path)
@@ -357,6 +360,8 @@ def load_workflow_from_source_code(workflow_id: str,
     module = module_from_spec(spec)
     spec.loader.exec_module(module)
 
+    sys.modules[module_name] = module
+
     module.__source__ = source_code
     module.__mtime__ = 0
     module.__workflow_id__ = workflow_id

qulab/version.py

@@ -1 +1 @@
-__version__ = "2.10.12"
+__version__ = "2.11.1"

{qulab-2.10.12.dist-info → qulab-2.11.1.dist-info}/METADATA

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

{qulab-2.10.12.dist-info → qulab-2.11.1.dist-info}/RECORD

@@ -2,15 +2,15 @@ qulab/__init__.py,sha256=hmf6R3jiM5NtJY1mSptogYnH5DMTR2dTzlXykqLxQzg,2027
 qulab/__main__.py,sha256=fjaRSL_uUjNIzBGNgjlGswb9TJ2VD5qnkZHW3hItrD4,68
 qulab/typing.py,sha256=vg62sGqxuD9CI5677ejlzAmf2fVdAESZCQjAE_xSxPg,69
 qulab/utils.py,sha256=BdLdlfjpe6m6gSeONYmpAKTTqxDaYHNk4exlz8kZxTg,2982
-qulab/version.py,sha256=i-CgswLqTZhhvAcCXnsvgdjjdAKnV-G_nzW-eW0UIMM,23
+qulab/version.py,sha256=L1NLbAiqOvyeykqIP0sEpfnHOltZ_nm9qsrW3nkEmTk,22
 qulab/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-qulab/cli/commands.py,sha256=hCeyZ4tjTu3aUbex9x_czkU6t9Ti_3F0u3bA8qs6BVQ,625
+qulab/cli/commands.py,sha256=Ezop5CEOvvEMDaPmfGXF0JH-CifKuDMk9yLglpUK9Zs,781
 qulab/cli/config.py,sha256=tZPSBLbf2x_Brb2UBuA1bsni8nC8WOPPGyWIi8m7j1I,5459
 qulab/cli/decorators.py,sha256=oImteZVnDPPWdyhJ4kzf2KYGJLON7VsKGBvZadWLQZo,621
 qulab/executor/__init__.py,sha256=LosPzOMaljSZY1thy_Fxtbrgq7uubJszMABEB7oM7tU,101
 qulab/executor/analyze.py,sha256=4Hau5LrKUdpweh7W94tcG4ahgxucHOevbM0hm57T7zE,5649
-qulab/executor/cli.py,sha256=3N4_AodGP1BKm3dZ1mNrRn7EsAiHV0Npfp_iwOW6A7s,15235
-qulab/executor/load.py,sha256=C5qepxC9QALMgfG_twxvGts6i0bCsnVg9lS05PXZZRM,20291
+qulab/executor/cli.py,sha256=FZQY249VVSoMZm3lZGaVTAvgULKjMeC97YUe4hqi5Y0,17608
+qulab/executor/load.py,sha256=eMlzOrrn8GpbP3J3uY5JJ8iO7tL5B1DWP1z2qiGyNhU,20385
 qulab/executor/registry.py,sha256=gym9F5FIDY5eV-cSCZsP99wC4l-6jkx9VMjJMaPOLaQ,4730
 qulab/executor/schedule.py,sha256=7gAJFwj13j1niGjVa1fSzwOS22eNFEN1hdrN3dfTY6A,21410
 qulab/executor/storage.py,sha256=8K73KGLAVgchJdtd4rKHXkr1CQOJORWH-Gi57w8IYsw,21081
@@ -96,9 +96,9 @@ qulab/visualization/plot_seq.py,sha256=UWTS6p9nfX_7B8ehcYo6UnSTUCjkBsNU9jiOeW2ca
 qulab/visualization/qdat.py,sha256=ZeevBYWkzbww4xZnsjHhw7wRorJCBzbG0iEu-XQB4EA,5735
 qulab/visualization/rot3d.py,sha256=lMrEJlRLwYe6NMBlGkKYpp_V9CTipOAuDy6QW_cQK00,734
 qulab/visualization/widgets.py,sha256=6KkiTyQ8J-ei70LbPQZAK35wjktY47w2IveOa682ftA,3180
-qulab-2.10.12.dist-info/licenses/LICENSE,sha256=PRzIKxZtpQcH7whTG6Egvzl1A0BvnSf30tmR2X2KrpA,1065
-qulab-2.10.12.dist-info/METADATA,sha256=qKjhGW1dhx9gDHCznSEHevQTdGoXmg62uTfdEx3yfB4,3869
-qulab-2.10.12.dist-info/WHEEL,sha256=Nw36Djuh_5VDukK0H78QzOX-_FQEo6V37m3nkm96gtU,91
-qulab-2.10.12.dist-info/entry_points.txt,sha256=b0v1GXOwmxY-nCCsPN_rHZZvY9CtTbWqrGj8u1m8yHo,45
-qulab-2.10.12.dist-info/top_level.txt,sha256=3T886LbAsbvjonu_TDdmgxKYUn939BVTRPxPl9r4cEg,6
-qulab-2.10.12.dist-info/RECORD,,
+qulab-2.11.1.dist-info/licenses/LICENSE,sha256=PRzIKxZtpQcH7whTG6Egvzl1A0BvnSf30tmR2X2KrpA,1065
+qulab-2.11.1.dist-info/METADATA,sha256=NhaeCv-fsilTHypKCMhtX0vuBVH3nCCtE-4k4Ri8zKk,3868
+qulab-2.11.1.dist-info/WHEEL,sha256=Nw36Djuh_5VDukK0H78QzOX-_FQEo6V37m3nkm96gtU,91
+qulab-2.11.1.dist-info/entry_points.txt,sha256=b0v1GXOwmxY-nCCsPN_rHZZvY9CtTbWqrGj8u1m8yHo,45
+qulab-2.11.1.dist-info/top_level.txt,sha256=3T886LbAsbvjonu_TDdmgxKYUn939BVTRPxPl9r4cEg,6
+qulab-2.11.1.dist-info/RECORD,,