QuLab 2.9.4__cp311-cp311-macosx_10_9_universal2.whl → 2.9.6__cp311-cp311-macosx_10_9_universal2.whl

qulab/cli/commands.py

@@ -1,6 +1,6 @@
 import click
 
-from ..executor.cli import create, get, maintain, run, set
+from ..executor.cli import create, get, maintain, reproduce, run, set
 from ..monitor.__main__ import main as monitor
 from ..scan.server import server
 from ..sys.net.cli import dht
@@ -24,6 +24,7 @@ cli.add_command(dht)
 cli.add_command(server)
 cli.add_command(maintain)
 cli.add_command(run)
+cli.add_command(reproduce)
 cli.add_command(create)
 cli.add_command(set)
 cli.add_command(get)

qulab/executor/cli.py

@@ -290,3 +290,40 @@ def maintain(workflow, code, data, api, retry, plot):
                 raise e
             logger.warning(f'Calibration failed, retrying ({i + 1}/{retry})')
             continue
+
+
+@click.command()
+@click.argument('report_id')
+@click.option('--plot', '-p', is_flag=True, help='Plot the report.')
+@log_options
+@command_option('reproduce')
+def reproduce(report_id, code, data, api, plot):
+    """
+    Reproduce a report.
+
+    If `--plot` is set, plot the report.
+    If `--api` is set, use the api to get and update the config table.
+    If `--code` is not set, use the current working directory.
+    If `--data` is not set, use the `logs` directory in the code path.
+    """
+    logger.info(
+        f'[CMD]: reproduce {report_id} --code {code} --data {data} --api {api}'
+        f'{" --plot" if plot else ""}')
+    if api is not None:
+        api = importlib.import_module(api)
+        set_config_api(api.query_config, api.update_config, api.export_config)
+    if code is None:
+        code = Path.cwd()
+    if data is None:
+        data = Path(code) / 'logs'
+
+    code = Path(os.path.expanduser(code))
+    data = Path(os.path.expanduser(data))
+
+    from .load import load_workflow_from_source_code
+    from .storage import get_report_by_index
+
+    r = get_report_by_index(int(report_id), data)
+
+    wf = load_workflow_from_source_code(r.workflow, r.script)
+    run_workflow(wf, code, data, plot=plot, freeze=True)

qulab/executor/load.py

@@ -1,6 +1,7 @@
 import graphlib
 import inspect
 import pickle
+import sys
 import warnings
 from importlib.util import module_from_spec, spec_from_file_location
 from pathlib import Path
@@ -329,6 +330,34 @@ def load_workflow_from_file(file_name: str,
     return module
 
 
+def load_workflow_from_source_code(workflow_id: str,
+                                   source_code: str,
+                                   package='workflows') -> WorkflowType:
+    path = Path(workflow_id)
+    module_name = f"{package}.{'.'.join([*path.parts[:-1], path.stem])}"
+
+    # 创建一个新的模块对象
+    module = ModuleType(module_name)
+    # 将模块注册到sys.modules中
+    sys.modules[module_name] = module
+    # 将源代码编译成字节码
+    code = compile(source_code, '<string>', 'exec')
+    # 在模块的命名空间中执行字节码
+    exec(code, module.__dict__)
+
+    module.__source__ = source_code
+    module.__mtime__ = 0
+    module.__workflow_id__ = workflow_id
+
+    if not hasattr(module, '__timeout__'):
+        module.__timeout__ = None
+
+    if not hasattr(module, 'depends'):
+        module.depends = lambda: []
+
+    return module
+
+
 def _generate_target_file_path(template_path: str | Path, hash_str: str,
                                content: str, base_path: str | Path) -> Path:
     path = Path(template_path)
@@ -341,7 +370,7 @@ def _generate_target_file_path(template_path: str | Path, hash_str: str,
         path = path.parent / f'{path.stem}_tmp{hash_str}.py'
 
     if 'templates' in path.parts:
-        path = Path(*['run' if p == 'templates' else p for p in path.parts])
+        return Path(*['run' if p == 'templates' else p for p in path.parts])
     else:
         return Path('run') / path
 

qulab/version.py

@@ -1 +1 @@
-__version__ = "2.9.4"
+__version__ = "2.9.6"

{qulab-2.9.4.dist-info → qulab-2.9.6.dist-info}/METADATA

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

{qulab-2.9.4.dist-info → qulab-2.9.6.dist-info}/RECORD

@@ -1,16 +1,16 @@
 qulab/__init__.py,sha256=JZ3bn_kTVlnY-P8JXQ5xEdViieFXqfxX8ReLuiiXIpo,321
 qulab/__main__.py,sha256=fjaRSL_uUjNIzBGNgjlGswb9TJ2VD5qnkZHW3hItrD4,68
 qulab/dicttree.py,sha256=tRRMpGZYVOLw0TEByE3_2Ss8FdOmzuGL9e1DWbs8qoY,13684
-qulab/fun.cpython-311-darwin.so,sha256=F5B0uC6GQTJ2z7uuRqiQAi5w7Id1YTPGMHBE8S_gSoo,126848
+qulab/fun.cpython-311-darwin.so,sha256=CzWrA4sqB0mgHXeLvvNru0O2zjSiKoL9WfOBCQeyO8g,126848
 qulab/typing.py,sha256=vg62sGqxuD9CI5677ejlzAmf2fVdAESZCQjAE_xSxPg,69
 qulab/utils.py,sha256=BdLdlfjpe6m6gSeONYmpAKTTqxDaYHNk4exlz8kZxTg,2982
-qulab/version.py,sha256=CNO-J_I4xO7wq79g0PqvHvoxcy0QHZ0CMdgoYP0rkOI,21
+qulab/version.py,sha256=xbldOz0XM3Jg2OQP4pA5dIy-zqYfu1fHZ9muPBbegtA,21
 qulab/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-qulab/cli/commands.py,sha256=tgDIkkeIoasQXAifJZ6NU8jDgpNgb2a-B0C4nF0evrE,559
+qulab/cli/commands.py,sha256=F1LATmSC9lOAdnrOTMK7DRjETCEcOmMsocovWRyjWTc,597
 qulab/cli/config.py,sha256=Ei7eSYnbwPPlluDnm8YmWONYiI4g7WtvlZGQdr1Z6vo,3688
 qulab/executor/__init__.py,sha256=LosPzOMaljSZY1thy_Fxtbrgq7uubJszMABEB7oM7tU,101
-qulab/executor/cli.py,sha256=8d-8bRWZ5lmsMtjASsl1zu1rV-syeAESMNVthvIQxlo,10018
-qulab/executor/load.py,sha256=cSmP4xICJDNjtHdRh3Hs1nHqfw4OTi7adPaMnrml_yg,17736
+qulab/executor/cli.py,sha256=OXFmCo2loOCgUYXZpvJz5AnzDag5m_iPetsFM1J11uQ,11291
+qulab/executor/load.py,sha256=qI-3b3xvsjZMCthRO3hlzHA0BZI-tH9Y_L0XLBcyWV8,18657
 qulab/executor/schedule.py,sha256=XRimchHYCgnMAOtCvNLjwMv9IkcTAyBAUCHKQs5RBRw,18745
 qulab/executor/storage.py,sha256=PWQIDYjQaoyLGgAKh0X1tlNQTgDWR8bI-HVie4hSkyA,21075
 qulab/executor/template.py,sha256=_HEtsUQ5_jSujCw8FBDAK1PRTMRCa4iD4DduHIpjo3c,10569
@@ -97,9 +97,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.9.4.dist-info/licenses/LICENSE,sha256=PRzIKxZtpQcH7whTG6Egvzl1A0BvnSf30tmR2X2KrpA,1065
-qulab-2.9.4.dist-info/METADATA,sha256=SDufMOCyA22SmE5Ka8zWoV1CWD6zilk2XtTZbsBdT5k,3720
-qulab-2.9.4.dist-info/WHEEL,sha256=RbtvOFtP--plS2sXkxxM8xYfkibPkCKcYIqEe_GHrhY,114
-qulab-2.9.4.dist-info/entry_points.txt,sha256=b0v1GXOwmxY-nCCsPN_rHZZvY9CtTbWqrGj8u1m8yHo,45
-qulab-2.9.4.dist-info/top_level.txt,sha256=3T886LbAsbvjonu_TDdmgxKYUn939BVTRPxPl9r4cEg,6
-qulab-2.9.4.dist-info/RECORD,,
+qulab-2.9.6.dist-info/licenses/LICENSE,sha256=PRzIKxZtpQcH7whTG6Egvzl1A0BvnSf30tmR2X2KrpA,1065
+qulab-2.9.6.dist-info/METADATA,sha256=EUA0I36KYGMd3X8yTwXUgOFsCvO4bQx5gZyKnsqfjcU,3720
+qulab-2.9.6.dist-info/WHEEL,sha256=RbtvOFtP--plS2sXkxxM8xYfkibPkCKcYIqEe_GHrhY,114
+qulab-2.9.6.dist-info/entry_points.txt,sha256=b0v1GXOwmxY-nCCsPN_rHZZvY9CtTbWqrGj8u1m8yHo,45
+qulab-2.9.6.dist-info/top_level.txt,sha256=3T886LbAsbvjonu_TDdmgxKYUn939BVTRPxPl9r4cEg,6
+qulab-2.9.6.dist-info/RECORD,,