QuLab 2.10.3__cp311-cp311-win_amd64.whl → 2.10.5__cp311-cp311-win_amd64.whl

qulab/__init__.py

@@ -1,4 +1,5 @@
 from .executor.analyze import manual_analysis
+from .executor.registry import Registry
 from .executor.storage import find_report
 from .executor.storage import get_report_by_index as get_report
 from .executor.template import VAR

qulab/cli/decorators.py

@@ -22,7 +22,7 @@ def run(main, *args, **kwds):
         try:
             import uvloop
             uvloop.run(main(*args, **kwds))
-        except ImportError:
+        except (ImportError, ModuleNotFoundError):
             asyncio.run(main(*args, **kwds))
     else:
         main(*args, **kwds)

qulab/executor/cli.py

@@ -12,10 +12,10 @@ from ..cli.config import get_config_value, log_options
 from ..cli.decorators import async_command
 from .load import (WorkflowType, find_unreferenced_workflows, get_entries,
                    load_workflow, make_graph)
+from .registry import Registry, set_config_api
 from .schedule import CalibrationFailedError
 from .schedule import maintain as maintain_workflow
 from .schedule import run as run_workflow
-from .transform import set_config_api
 from .utils import workflow_template
 
 
@@ -95,6 +95,7 @@ def command_option(command_name):
               '-c',
               default=lambda: get_config_value("code", str, 'create'),
               help='The path of the code.')
+@log_options('create')
 def create(workflow, code):
     """
     Create a new workflow file.
@@ -124,12 +125,13 @@ def create(workflow, code):
               '-a',
               default=lambda: get_config_value("api", str, 'set'),
               help='The modlule name of the api.')
+@log_options('set')
 def set(key, value, api):
     """
     Set a config.
     """
     logger.info(f'[CMD]: set {key} {value} --api {api}')
-    from . import transform
+    reg = Registry()
     if api is not None:
         api = importlib.import_module(api)
         set_config_api(api.query_config, api.update_config, api.delete_config,
@@ -138,7 +140,7 @@ def set(key, value, api):
         value = eval(value)
     except:
         pass
-    transform.update_config({key: value})
+    reg.set(key, value)
 
 
 @click.command()
@@ -147,17 +149,18 @@ def set(key, value, api):
               '-a',
               default=lambda: get_config_value("api", str, 'get'),
               help='The modlule name of the api.')
+@log_options('get')
 def get(key, api):
     """
     Get a config.
     """
     logger.info(f'[CMD]: get {key} --api {api}')
-    from . import transform
+    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)
-    click.echo(transform.query_config(key))
+    click.echo(reg.get(key))
 
 
 @click.command()
@@ -176,7 +179,7 @@ def get(key, api):
 @click.option('--veryfy-source-code',
               is_flag=True,
               help='Veryfy the source code.')
-@log_options
+@log_options('run')
 @command_option('run')
 @async_command
 async def run(workflow,
@@ -236,7 +239,6 @@ async def run(workflow,
                                 data,
                                 plot=plot,
                                 freeze=freeze,
-                                veryfy_source_code=veryfy_source_code,
                             )
                         except Exception as e:
                             if fail_fast:
@@ -251,7 +253,6 @@ async def run(workflow,
                         data,
                         plot=plot,
                         freeze=freeze,
-                        veryfy_source_code=veryfy_source_code,
                     )
             else:
                 if hasattr(wf, 'entries'):
@@ -305,7 +306,7 @@ async def run(workflow,
 @click.option('--veryfy-source-code',
               is_flag=True,
               help='Veryfy the source code.')
-@log_options
+@log_options('maintain')
 @command_option('maintain')
 @async_command
 async def maintain(workflow,
@@ -390,7 +391,7 @@ async def maintain(workflow,
 @click.command()
 @click.argument('report_id')
 @click.option('--plot', '-p', is_flag=True, help='Plot the report.')
-@log_options
+@log_options('reproduce')
 @command_option('reproduce')
 @async_command
 async def reproduce(report_id, code, data, api, plot):
@@ -417,16 +418,17 @@ async def reproduce(report_id, code, data, api, plot):
     code = Path(os.path.expanduser(code))
     data = Path(os.path.expanduser(data))
 
-    from . import transform
     from .load import load_workflow_from_source_code
     from .storage import get_report_by_index
 
+    reg = Registry()
+
     r = get_report_by_index(int(report_id), data)
 
     wf = load_workflow_from_source_code(r.workflow, r.script)
-    cfg = transform.export_config()
-    transform.clear_config()
-    transform.update_config(r.config)
+    cfg = reg.export()
+    reg.clear()
+    reg.update(r.config)
     await run_workflow(wf, code, data, plot=plot, freeze=True)
-    transform.clear_config()
-    transform.update_config(cfg)
+    reg.clear()
+    reg.update(cfg)

qulab/executor/{transform.py → registry.py}

@@ -1,3 +1,9 @@
+import importlib
+import os
+import sys
+from typing import Any
+
+from ..cli.config import get_config_value
 from .storage import Report, save_item
 
 __current_config_id = None
@@ -132,3 +138,44 @@ def set_config_api(query_method,
     clear_config = clear_method
 
     return query_config, update_config, delete_config, export_config, clear_config
+
+
+def _init():
+    code = get_config_value("code", str, default=None)
+    if code is not None:
+        code = os.path.expanduser(code)
+        if code not in sys.path:
+            sys.path.insert(0, code)
+
+    api = get_config_value('api', str, None)
+    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)
+
+
+_init()
+
+
+class Registry():
+
+    def query(self, key: str) -> Any:
+        return query_config(key)
+
+    def get(self, key: str) -> Any:
+        return self.query(key)
+
+    def export(self) -> dict[str, dict[str, Any]]:
+        return export_config()
+
+    def set(self, key: str, value: Any):
+        return update_config({key: value})
+
+    def delete(self, key: str):
+        return delete_config(key)
+
+    def clear(self):
+        return clear_config()
+
+    def update(self, parameters: dict[str, Any]):
+        return update_config(parameters)

qulab/executor/schedule.py

@@ -8,9 +8,9 @@ from pathlib import Path
 from loguru import logger
 
 from .load import WorkflowType, get_dependents
+from .registry import current_config, obey_the_oracle, update_parameters
 from .storage import (Report, find_report, get_head, get_heads, renew_report,
                       revoke_report, save_item, save_report)
-from .transform import current_config, obey_the_oracle, update_parameters
 
 __session_id = None
 __session_cache = {}

qulab/version.py

@@ -1 +1 @@
-__version__ = "2.10.3"
+__version__ = "2.10.5"

{qulab-2.10.3.dist-info → qulab-2.10.5.dist-info}/METADATA

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

{qulab-2.10.3.dist-info → qulab-2.10.5.dist-info}/RECORD

@@ -1,22 +1,22 @@
-qulab/__init__.py,sha256=ulapyzt9DsDgT68EEbzuhniofHb512pNTFc-Sxfrr8Y,375
+qulab/__init__.py,sha256=-tTI2n3lAFU3sRFEu-Z_ihgiHbyyaQeuLDWfLFHlPvc,416
 qulab/__main__.py,sha256=FL4YsGZL1jEtmcPc5WbleArzhOHLMsWl7OH3O-1d1ss,72
 qulab/dicttree.py,sha256=ZoSJVWK4VMqfzj42gPb_n5RqLlM6K1Me0WmLIfLEYf8,14195
-qulab/fun.cp311-win_amd64.pyd,sha256=k62pIEk_w4Id1J7iC8hiQqCilIA6_ra-7manzop27HE,31744
+qulab/fun.cp311-win_amd64.pyd,sha256=3Ib2duRQI4tI3P2pbRTDrF2fZaCsVsFAPRMPWG5s9m8,31744
 qulab/typing.py,sha256=PRtwbCHWY2ROKK8GHq4Bo8llXrIGo6xC73DrQf7S9os,71
 qulab/utils.py,sha256=65N2Xj7kqRsQ4epoLNY6tL-i5ts6Wk8YuJYee3Te6zI,3077
-qulab/version.py,sha256=XIYhk5PoReQgD4uSOa1eXmCSwNRY66HHGSW3_z-nibk,22
+qulab/version.py,sha256=SB3sQYH0e-iHrGpjw5MXxU2T-sxkpouM4O_fyPlJPXQ,22
 qulab/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qulab/cli/commands.py,sha256=ZTs32yQjvwPIsFjXYWNr4KqEvly0NISIqyh--96qCAY,627
 qulab/cli/config.py,sha256=A3UnyaRtiofpokyzYkJnUdwzcsYX7H-xZvBVduIOSdg,5617
-qulab/cli/decorators.py,sha256=UmJab1o-XH-8DVS_LZFGY2SPjyVgZ1QrJwkauTZDUik,626
+qulab/cli/decorators.py,sha256=mgAZiiaj2vZvsJEg70gxgB9TWkxR1nQSb2m-73RYHr0,649
 qulab/executor/__init__.py,sha256=LosPzOMaljSZY1thy_Fxtbrgq7uubJszMABEB7oM7tU,101
 qulab/executor/analyze.py,sha256=VoSthE2RTarY6Wj3QFKh4FqReMYL7djSAyuitgHs5K0,5837
-qulab/executor/cli.py,sha256=ylsKlr1qiloTVPhF2xoTxbz5DD98VQkRXQpvmiVJdxY,15047
+qulab/executor/cli.py,sha256=bmuFPpPhHiRzTjIjhvlp2tQYp0jEwXSETILX8Nu2tLE,14901
 qulab/executor/load.py,sha256=0-EtO4dkP6jB1vsgQlniv-OtPH1AZLjbtuvaWHsgbPs,20335
-qulab/executor/schedule.py,sha256=1YWz2d5YLmnEpXGX1aVroE0SOKdU28WwHGEioxBpTKQ,21411
+qulab/executor/registry.py,sha256=qqj90HVYun-DDI9HPmwyOQI87M2mQ0jyw4vWwPMfYjg,4592
+qulab/executor/schedule.py,sha256=7gAJFwj13j1niGjVa1fSzwOS22eNFEN1hdrN3dfTY6A,21410
 qulab/executor/storage.py,sha256=8K73KGLAVgchJdtd4rKHXkr1CQOJORWH-Gi57w8IYsw,21081
 qulab/executor/template.py,sha256=dKQM3IlADdTi9qp0fnOYjyycRNEl7KeSCBZhwbmP8bQ,10828
-qulab/executor/transform.py,sha256=rk4CLIKVjGRaFzi5FVSgadUxAKKVLSopEHZCaAzDwDg,3435
 qulab/executor/utils.py,sha256=3OLRMBJu-1t78BeuZs4fv4jioEXnRNygaPnSoibzfgs,6405
 qulab/monitor/__init__.py,sha256=xEVDkJF8issrsDeLqQmDsvtRmrf-UiViFcGTWuzdlFU,43
 qulab/monitor/__main__.py,sha256=k2H1H5Zf9LLXTDLISJkbikLH-z0f1e5i5i6wXXYPOrE,105
@@ -99,9 +99,9 @@ qulab/visualization/plot_seq.py,sha256=Uo1-dB1YE9IN_A9tuaOs9ZG3S5dKDQ_l98iD2Wbxp
 qulab/visualization/qdat.py,sha256=HubXFu4nfcA7iUzghJGle1C86G6221hicLR0b-GqhKQ,5887
 qulab/visualization/rot3d.py,sha256=jGHJcqj1lEWBUV-W4GUGONGacqjrYvuFoFCwPse5h1Y,757
 qulab/visualization/widgets.py,sha256=HcYwdhDtLreJiYaZuN3LfofjJmZcLwjMfP5aasebgDo,3266
-qulab-2.10.3.dist-info/licenses/LICENSE,sha256=b4NRQ-GFVpJMT7RuExW3NwhfbrYsX7AcdB7Gudok-fs,1086
-qulab-2.10.3.dist-info/METADATA,sha256=xpIJzNygtIMGd9wGlhVT99BwC_RTRsGEhn7YkWFLvdk,3860
-qulab-2.10.3.dist-info/WHEEL,sha256=_ZWIY2n7n6SpiuIFl1-RvcMp4Ty36T57FKf-7NzqZHM,101
-qulab-2.10.3.dist-info/entry_points.txt,sha256=b0v1GXOwmxY-nCCsPN_rHZZvY9CtTbWqrGj8u1m8yHo,45
-qulab-2.10.3.dist-info/top_level.txt,sha256=3T886LbAsbvjonu_TDdmgxKYUn939BVTRPxPl9r4cEg,6
-qulab-2.10.3.dist-info/RECORD,,
+qulab-2.10.5.dist-info/licenses/LICENSE,sha256=b4NRQ-GFVpJMT7RuExW3NwhfbrYsX7AcdB7Gudok-fs,1086
+qulab-2.10.5.dist-info/METADATA,sha256=qO155vhjLn82S1GuWo9IHD5JPWMQpIEpgIz7bQ6ps44,3860
+qulab-2.10.5.dist-info/WHEEL,sha256=_ZWIY2n7n6SpiuIFl1-RvcMp4Ty36T57FKf-7NzqZHM,101
+qulab-2.10.5.dist-info/entry_points.txt,sha256=b0v1GXOwmxY-nCCsPN_rHZZvY9CtTbWqrGj8u1m8yHo,45
+qulab-2.10.5.dist-info/top_level.txt,sha256=3T886LbAsbvjonu_TDdmgxKYUn939BVTRPxPl9r4cEg,6
+qulab-2.10.5.dist-info/RECORD,,