QuLab 2.7.8__cp311-cp311-macosx_10_9_universal2.whl → 2.7.10__cp311-cp311-macosx_10_9_universal2.whl

{QuLab-2.7.8.dist-info → QuLab-2.7.10.dist-info}/METADATA

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

{QuLab-2.7.8.dist-info → QuLab-2.7.10.dist-info}/RECORD

@@ -1,19 +1,19 @@
 qulab/__init__.py,sha256=KJcUcZ5qXY6wlAoirzK_B-dgtDjsLmOE671v3gcXO_c,286
 qulab/__main__.py,sha256=fjaRSL_uUjNIzBGNgjlGswb9TJ2VD5qnkZHW3hItrD4,68
 qulab/dicttree.py,sha256=tRRMpGZYVOLw0TEByE3_2Ss8FdOmzuGL9e1DWbs8qoY,13684
-qulab/fun.cpython-311-darwin.so,sha256=bti_a0L68qdw0Czc4tylTKtDlujrojghscH_bNOE_tc,126848
+qulab/fun.cpython-311-darwin.so,sha256=NU_NLJ3C44apsURL4y_seor221rmQOd3GvuwzKs9iyU,126848
 qulab/typing.py,sha256=vg62sGqxuD9CI5677ejlzAmf2fVdAESZCQjAE_xSxPg,69
 qulab/utils.py,sha256=JIXMSmZU0uYfKG_tzawpK7vRNPRir_hJE8JlqkVLX2o,1260
-qulab/version.py,sha256=89JN-VCK3x9bbDL4mH6FsYHIsGSTeoIxhxxTqfViSLE,21
+qulab/version.py,sha256=jdPSCkDOwzHQzdidl89buWuyX5dorFZluLWAvXtMdIU,22
 qulab/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qulab/cli/commands.py,sha256=tgDIkkeIoasQXAifJZ6NU8jDgpNgb2a-B0C4nF0evrE,559
 qulab/cli/config.py,sha256=SdNmWzweWAdyk8M2oKYhMxnkaJ0qIayPlsLGCNlVqME,3108
 qulab/executor/__init__.py,sha256=LosPzOMaljSZY1thy_Fxtbrgq7uubJszMABEB7oM7tU,101
 qulab/executor/cli.py,sha256=gGka2M6xccfM8facsIJ2qZ1y2Yx8C4BRhc1JG6nK9mo,8932
 qulab/executor/load.py,sha256=4FY_SwumLDUewC265v4j_ZGGpfYOgH4c8PtglYcWpBw,18077
-qulab/executor/schedule.py,sha256=OYvolWiafZbXiodeyJvr0-hscHvBt-eifmg7uWEymII,16952
-qulab/executor/storage.py,sha256=JH817-d6izBiz6wBGWvWxNEY_IPhRwRmGcB_eUpvE8U,11364
-qulab/executor/transform.py,sha256=s0kxWQx8Sr9pMIQke1BLNM6KqrSogAkjB6Zkapl8YSU,2189
+qulab/executor/schedule.py,sha256=S8zRdkM2_I9-EqMUCEGI9boe3sfc68Gn5NRK057ShAw,17997
+qulab/executor/storage.py,sha256=gI6g28BmKKEZ_Pl-hFwvpiOj3mF8Su-yjj3hfMXs1VY,11630
+qulab/executor/transform.py,sha256=4DyGXv1Iw36cCPqbqXlsn0Lw6gFjQpp_6Hcta50YQxU,2181
 qulab/executor/utils.py,sha256=cF6-2jlvlHyTjNHdxXKG04Fjfm3_3wfDQAF1G8DQphk,5686
 qulab/monitor/__init__.py,sha256=nTHelnDpxRS_fl_B38TsN0njgq8eVTEz9IAnN3NbDlM,42
 qulab/monitor/__main__.py,sha256=w3yUcqq195LzSnXTkQcuC1RSFRhy4oQ_PEBmucXguME,97
@@ -94,9 +94,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.7.8.dist-info/LICENSE,sha256=PRzIKxZtpQcH7whTG6Egvzl1A0BvnSf30tmR2X2KrpA,1065
-QuLab-2.7.8.dist-info/METADATA,sha256=ZZQMlhagxeJHogg-wXRxB2SaFziTmaBjXwy05kW4fps,3698
-QuLab-2.7.8.dist-info/WHEEL,sha256=K10eKSN6_vzvMOgXxWbVOQNR7Orfl6gBTCpCI8bcYx4,114
-QuLab-2.7.8.dist-info/entry_points.txt,sha256=b0v1GXOwmxY-nCCsPN_rHZZvY9CtTbWqrGj8u1m8yHo,45
-QuLab-2.7.8.dist-info/top_level.txt,sha256=3T886LbAsbvjonu_TDdmgxKYUn939BVTRPxPl9r4cEg,6
-QuLab-2.7.8.dist-info/RECORD,,
+QuLab-2.7.10.dist-info/LICENSE,sha256=PRzIKxZtpQcH7whTG6Egvzl1A0BvnSf30tmR2X2KrpA,1065
+QuLab-2.7.10.dist-info/METADATA,sha256=-3XSmnAFhNexNzMEufmMAAUF2JfWg2YL_7crqvgyIPg,3699
+QuLab-2.7.10.dist-info/WHEEL,sha256=K10eKSN6_vzvMOgXxWbVOQNR7Orfl6gBTCpCI8bcYx4,114
+QuLab-2.7.10.dist-info/entry_points.txt,sha256=b0v1GXOwmxY-nCCsPN_rHZZvY9CtTbWqrGj8u1m8yHo,45
+QuLab-2.7.10.dist-info/top_level.txt,sha256=3T886LbAsbvjonu_TDdmgxKYUn939BVTRPxPl9r4cEg,6
+QuLab-2.7.10.dist-info/RECORD,,

qulab/executor/schedule.py

@@ -8,7 +8,7 @@ from loguru import logger
 
 from .load import WorkflowType, get_dependents
 from .storage import (Report, find_report, get_head, get_heads, renew_report,
-                      revoke_report, save_report)
+                      revoke_report, save_item, save_report)
 from .transform import current_config, obey_the_oracle, update_parameters
 
 __session_id = None
@@ -69,6 +69,16 @@ def veryfy_analyzed_report(report: Report, script: str, method: str):
         )
 
 
+def get_source(workflow: WorkflowType, code_path: str | Path) -> str:
+    if isinstance(code_path, str):
+        code_path = Path(code_path)
+    try:
+        with open(code_path / workflow.__workflow_id__, 'r') as f:
+            return f.read()
+    except:
+        return ''
+
+
 def check_state(workflow: WorkflowType, code_path: str | Path,
                 state_path: str | Path) -> bool:
     """
@@ -91,6 +101,11 @@ def check_state(workflow: WorkflowType, code_path: str | Path,
             f'check_state: "{workflow.__workflow_id__}" has custom check_state method'
         )
         return workflow.check_state(report)
+    if datetime.fromtimestamp(workflow.__mtime__) > report.checked_time:
+        logger.debug(
+            f'check_state failed: "{workflow.__workflow_id__}" has been modified after last calibration'
+        )
+        return False
     if workflow.__timeout__ is not None and datetime.now(
     ) > report.checked_time + timedelta(seconds=workflow.__timeout__):
         logger.debug(
@@ -143,7 +158,9 @@ def call_check(workflow: WorkflowType, session_id: str, state_path: Path):
                     base_path=state_path,
                     heads=get_heads(state_path),
                     previous_path=get_head(workflow.__workflow_id__,
-                                           state_path))
+                                           state_path),
+                    script_path=save_item(get_source(workflow, state_path),
+                                          state_path))
 
     save_report(workflow.__workflow_id__, report, state_path)
 
@@ -168,7 +185,9 @@ def call_calibrate(workflow: WorkflowType, session_id: str, state_path: Path):
                     base_path=state_path,
                     heads=get_heads(state_path),
                     previous_path=get_head(workflow.__workflow_id__,
-                                           state_path))
+                                           state_path),
+                    script_path=save_item(get_source(workflow, state_path),
+                                          state_path))
 
     save_report(workflow.__workflow_id__, report, state_path)
 
@@ -187,11 +206,17 @@ def call_check_analyzer(node: WorkflowType,
     if report.in_spec:
         logger.debug(
             f'"{node.__workflow_id__}": checked in spec, renewing report')
-        renew_report(node.__workflow_id__, report.previous, state_path)
+        if report.previous is not None:
+            renew_report(node.__workflow_id__, report.previous, state_path)
+        else:
+            renew_report(node.__workflow_id__, report, state_path)
     else:
         logger.debug(
             f'"{node.__workflow_id__}": checked out of spec, revoking report')
-        revoke_report(node.__workflow_id__, report.previous, state_path)
+        if report.previous is not None:
+            revoke_report(node.__workflow_id__, report.previous, state_path)
+        else:
+            revoke_report(node.__workflow_id__, report, state_path)
     return report
 
 
@@ -244,7 +269,9 @@ def check_data(workflow: WorkflowType, state_path: str | Path, plot: bool,
                         base_path=state_path,
                         heads=get_heads(state_path),
                         previous_path=get_head(workflow.__workflow_id__,
-                                               state_path))
+                                               state_path),
+                        script_path=save_item(get_source(workflow, state_path),
+                                              state_path))
         report.in_spec = False
         report.bad_data = False
         return report
@@ -300,12 +327,7 @@ def calibrate(workflow: WorkflowType, state_path: str | Path, plot: bool,
 
     logger.debug(f'Calibrated "{workflow.__workflow_id__}" !')
 
-    report = call_analyzer(workflow,
-                           report,
-                           history,
-                           state_path,
-                           check=False,
-                           plot=plot)
+    report = call_analyzer(workflow, report, history, state_path, plot=plot)
     return report
 
 

qulab/executor/storage.py

@@ -34,6 +34,7 @@ class Report():
     base_path: Path | None = field(default=None, repr=False)
     path: Path | None = field(default=None, repr=False)
     config_path: Path | None = field(default=None, repr=False)
+    script_path: Path | None = field(default=None, repr=False)
 
     @property
     def previous(self):
@@ -54,13 +55,6 @@ class Report():
                 state = 'Outdated'
         return state
 
-    @property
-    def config(self):
-        if self.config_path is not None and self.base_path is not None:
-            return load_config(self.config_path, self.base_path)
-        else:
-            return None
-
     @state.setter
     def state(self, state: Literal['OK', 'Bad', 'Outdated', 'In spec',
                                    'Out of spec', 'Bad data']):
@@ -80,6 +74,20 @@ class Report():
             self.bad_data = False
             self.in_spec = False
 
+    @property
+    def config(self):
+        if self.config_path is not None and self.base_path is not None:
+            return load_item(self.config_path, self.base_path)
+        else:
+            return None
+
+    @property
+    def script(self):
+        if self.script_path is not None and self.base_path is not None:
+            return load_item(self.script_path, self.base_path)
+        else:
+            return None
+
 
 def random_path(base: Path) -> Path:
     while True:
@@ -157,19 +165,19 @@ def save_report(workflow: str,
         path = report.path
         if path is None:
             raise ValueError("Report path is None, can't overwrite.")
-        with open(base_path / 'objects' / path, "rb") as f:
+        with open(base_path / 'reports' / path, "rb") as f:
             index = int.from_bytes(f.read(8), 'big')
         report.index = index
     else:
-        path = random_path(base_path / 'objects')
-        (base_path / 'objects' / path).parent.mkdir(parents=True,
+        path = random_path(base_path / 'reports')
+        (base_path / 'reports' / path).parent.mkdir(parents=True,
                                                     exist_ok=True)
         report.path = path
         report.index = create_index("report",
                                     base_path,
                                     context=str(path),
                                     width=35)
-    with open(base_path / 'objects' / path, "wb") as f:
+    with open(base_path / 'reports' / path, "wb") as f:
         f.write(report.index.to_bytes(8, 'big'))
         f.write(buf)
     if refresh_heads:
@@ -179,9 +187,9 @@ def save_report(workflow: str,
 
 def load_report(path: str | Path, base_path: str | Path) -> Report | None:
     base_path = Path(base_path)
-    path = base_path / 'objects' / path
+    path = base_path / 'reports' / path
 
-    with open(base_path / 'objects' / path, "rb") as f:
+    with open(base_path / 'reports' / path, "rb") as f:
         index = int.from_bytes(f.read(8), 'big')
         report = pickle.loads(lzma.decompress(f.read()))
         report.base_path = base_path
@@ -315,18 +323,17 @@ def get_report_by_index(
         return None
 
 
-def save_config(cfg, data_path):
-    i = 0
-    buf = pickle.dumps(cfg)
+def save_item(item, data_path):
+    salt = 0
+    buf = pickle.dumps(item)
     buf = lzma.compress(buf)
     h = hashlib.md5(buf)
 
     while True:
-        salt = f"{i}".encode()
-        h.update(salt)
+        h.update(f"{salt}".encode())
         hashstr = h.hexdigest()
-        cfg_id = Path(hashstr[:2]) / hashstr[2:4] / hashstr[4:]
-        path = Path(data_path) / 'config' / cfg_id
+        item_id = Path(hashstr[:2]) / hashstr[2:4] / hashstr[4:]
+        path = Path(data_path) / 'items' / item_id
         if not path.exists():
             path.parent.mkdir(parents=True, exist_ok=True)
             with open(path, 'wb') as f:
@@ -334,13 +341,13 @@ def save_config(cfg, data_path):
             break
         elif path.read_bytes() == buf:
             break
-        i += 1
-    return str(cfg_id)
+        salt += 1
+    return str(item_id)
 
 
-@lru_cache(maxsize=1024)
-def load_config(id, data_path):
-    path = Path(data_path) / 'config' / id
+@lru_cache(maxsize=4096)
+def load_item(id, data_path):
+    path = Path(data_path) / 'items' / id
     with open(path, 'rb') as f:
         buf = f.read()
     cfg = pickle.loads(lzma.decompress(buf))

qulab/executor/transform.py

@@ -1,4 +1,4 @@
-from .storage import Report, save_config
+from .storage import Report, save_item
 
 __current_config_id = None
 
@@ -47,21 +47,21 @@ def obey_the_oracle(report: Report, data_path):
     global __current_config_id
     update_config(report.oracle)
     cfg = export_config()
-    __current_config_id = save_config(cfg, data_path)
+    __current_config_id = save_item(cfg, data_path)
 
 
 def update_parameters(report: Report, data_path):
     global __current_config_id
     update_config(report.parameters)
     cfg = export_config()
-    __current_config_id = save_config(cfg, data_path)
+    __current_config_id = save_item(cfg, data_path)
 
 
 def current_config(data_path):
     global __current_config_id
     if __current_config_id is None:
         cfg = export_config()
-        __current_config_id = save_config(cfg, data_path)
+        __current_config_id = save_item(cfg, data_path)
     return __current_config_id
 
 

qulab/version.py

@@ -1 +1 @@
-__version__ = "2.7.8"
+__version__ = "2.7.10"