QuLab 2.11.13__py3-none-any.whl → 2.11.15__py3-none-any.whl

qulab/cli/config.py

@@ -3,6 +3,7 @@ import functools
 import os
 import sys
 from pathlib import Path
+from typing import Any
 
 import click
 from loguru import logger
@@ -11,6 +12,34 @@ DEFAULT_CONFIG_PATH = Path(os.path.expanduser("~/.qulab.ini"))
 ENV_PREFIX = "QULAB_"
 
 
+def find_config_file() -> Path | None:
+    """
+    查找配置文件
+    """
+    # 环境变量
+    if config_path := os.getenv("QULAB_CONFIG_PATH"):
+        return Path(config_path)
+
+    # 先查找当前目录下的配置文件
+    config_path = Path.cwd() / "qulab.ini"
+    if config_path.exists():
+        return config_path
+
+    # 向上级目录查找，直到找到 uv 项目根目录
+    current_dir = Path.cwd()
+    while not (current_dir / "uv.lock").exists():
+        current_dir = current_dir.parent
+        if current_dir == Path("/"):
+            break
+    config_path = current_dir / "qulab.ini"
+    if config_path.exists():
+        return config_path
+
+    if DEFAULT_CONFIG_PATH.exists():
+        return DEFAULT_CONFIG_PATH
+    return None
+
+
 def _get_config_value(option_name,
                       type_cast=str,
                       command_name=None,
@@ -37,11 +66,9 @@ def _get_config_value(option_name,
     # 先加载默认配置防止段不存在
     config.read_dict({config_section: {}})
 
-    config_path = Path.cwd() / "qulab.ini"
-    if config_path.exists():
+    config_path = find_config_file()
+    if config_path is not None:
         config.read(config_path)
-    elif DEFAULT_CONFIG_PATH.exists():
-        config.read(DEFAULT_CONFIG_PATH)
 
     # 从对应配置段读取
     if config.has_section(config_section):
@@ -57,10 +84,10 @@ def _get_config_value(option_name,
     return default  # 交给 Click 处理默认值
 
 
-def get_config_value(option_name,
-                     type_cast=str,
-                     command_name=None,
-                     default=None):
+def get_config_value(option_name: str,
+                     type_cast: type = str,
+                     command_name: str | None = None,
+                     default: Any = None):
     """
     获取配置值，支持命令专属配置的优先级获取
 

qulab/executor/registry.py

@@ -1,3 +1,4 @@
+import copy
 import importlib
 import os
 import sys
@@ -157,6 +158,128 @@ def _init():
 _init()
 
 
+class RegistrySnapshot:
+
+    def __init__(self, data: dict[str, Any]):
+        self.data = copy.deepcopy(data)
+
+    def query(self, key: str, default=...) -> Any:
+        """
+        Query a value from the nested dictionary using dot-separated keys.
+        
+        Args:
+            key: Dot-separated key path (e.g., 'level1.level2.level3')
+            default: Default value to return if key not found.
+                     If not provided and key is missing, raises KeyError.
+            
+        Returns:
+            The value at the specified path, default value, or raises KeyError
+        """
+        keys = key.split('.')
+        current = self.data
+
+        # Track which level we're at for error reporting
+        for i, k in enumerate(keys):
+            if isinstance(current, dict) and k in current:
+                current = current[k]
+            else:
+                # Key not found at this level
+                missing_path = '.'.join(keys[:i + 1])
+                if default is ...:
+                    raise KeyError(
+                        f"Key '{missing_path}' not found in registry (failed at level {i+1})"
+                    )
+                return default
+
+        return current
+
+    def get(self, key: str, default=...) -> Any:
+        """
+        Query a value from the nested dictionary using dot-separated keys.
+        
+        Args:
+            key: Dot-separated key path (e.g., 'level1.level2.level3')
+            default: Default value to return if key not found.
+                     If not provided and key is missing, raises KeyError.
+            
+        Returns:
+            The value at the specified path, default value, or raises KeyError
+        """
+        return self.query(key, default)
+
+    def export(self) -> dict[str, Any]:
+        return self.data
+
+    def set(self, key: str, value: Any):
+        """
+        Set a value in the nested dictionary using dot-separated keys.
+        Creates intermediate dictionaries if they don't exist.
+        
+        Args:
+            key: Dot-separated key path (e.g., 'level1.level2.level3')
+            value: Value to set
+        """
+        keys = key.split('.')
+        current = self.data
+
+        # Navigate to the parent of the target key, creating dicts as needed
+        for k in keys[:-1]:
+            if k not in current:
+                current[k] = {}
+            elif not isinstance(current[k], dict):
+                # If existing value is not a dict, replace it with a dict
+                current[k] = {}
+            current = current[k]
+
+        # Set the final value
+        current[keys[-1]] = value
+
+    def delete(self, key: str):
+        """
+        Delete a value from the nested dictionary using dot-separated keys.
+        Also cleans up empty parent dictionaries after deletion.
+        
+        Args:
+            key: Dot-separated key path (e.g., 'level1.level2.level3')
+        """
+        keys = key.split('.')
+
+        # First check if the path exists
+        try:
+            self.query(key)
+        except KeyError:
+            return  # Path doesn't exist, nothing to delete
+
+        # Navigate and collect references to all parent dictionaries
+        path_refs = []
+        current = self.data
+
+        for k in keys[:-1]:
+            path_refs.append((current, k))
+            current = current[k]
+
+        # Delete the final key
+        if isinstance(current, dict) and keys[-1] in current:
+            del current[keys[-1]]
+
+            # Clean up empty dictionaries from leaf to root
+            for parent_dict, parent_key in reversed(path_refs):
+                # Check if the current dictionary is empty
+                if isinstance(current, dict) and len(current) == 0:
+                    del parent_dict[parent_key]
+                    current = parent_dict
+                else:
+                    # Stop if we encounter a non-empty dictionary
+                    break
+
+    def clear(self):
+        self.data.clear()
+
+    def update(self, parameters: dict[str, Any]):
+        for k, v in parameters.items():
+            self.set(k, v)
+
+
 class Registry():
 
     def __init__(self):
@@ -183,3 +306,6 @@ class Registry():
 
     def update(self, parameters: dict[str, Any]):
         return self.api[1](parameters)
+
+    def snapshot(self) -> RegistrySnapshot:
+        return RegistrySnapshot(self.export())

qulab/executor/template.py

@@ -236,6 +236,8 @@ def inject_mapping(source: str, mapping: dict[str, Any],
                 if default_value is not None:
                     # 清除 default 值前后的空格
                     clean_value = default_value.strip()
+                    if var_name not in mapping:
+                        return clean_value
                     return f'{base}.get({quote}{var_name}{quote}, {clean_value})'
                 else:
                     return f'{base}[{quote}{var_name}{quote}]'

qulab/version.py

@@ -1 +1 @@
-__version__ = "2.11.13"
+__version__ = "2.11.15"

{qulab-2.11.13.dist-info → qulab-2.11.15.dist-info}/METADATA

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

{qulab-2.11.13.dist-info → qulab-2.11.15.dist-info}/RECORD

@@ -2,19 +2,19 @@ qulab/__init__.py,sha256=hmf6R3jiM5NtJY1mSptogYnH5DMTR2dTzlXykqLxQzg,2027
 qulab/__main__.py,sha256=fjaRSL_uUjNIzBGNgjlGswb9TJ2VD5qnkZHW3hItrD4,68
 qulab/typing.py,sha256=vg62sGqxuD9CI5677ejlzAmf2fVdAESZCQjAE_xSxPg,69
 qulab/utils.py,sha256=B_8QdY9OMY7WU-F200v93BDJXJpQcKAHihnOXeEvv_w,3966
-qulab/version.py,sha256=zzz5acK8diCvnDna0wJqi0GGAyMkaKL0pR-gfoeXbEc,23
+qulab/version.py,sha256=Xwyp7A0xHh2GDh3MUxhzaH2tfp0EQ4tK8L1vQV9D1ow,23
 qulab/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qulab/cli/commands.py,sha256=ywKmwbGNBCVASx8OsgrRttaLz9ogwY38ZdKj7I-tdJ4,813
-qulab/cli/config.py,sha256=RIQlCTViDOg_QA0J7Y0jNJ8TEXG7ATmGwbK2Ief0NPY,5591
+qulab/cli/config.py,sha256=Z-lePcDMeeQmwFpdfQHqZuaXTPhWUHN5KrlSqrya73Y,6324
 qulab/cli/decorators.py,sha256=oImteZVnDPPWdyhJ4kzf2KYGJLON7VsKGBvZadWLQZo,621
 qulab/executor/__init__.py,sha256=LosPzOMaljSZY1thy_Fxtbrgq7uubJszMABEB7oM7tU,101
 qulab/executor/analyze.py,sha256=VYTfiOYJe5gDbtGA4wOkrQhhu55iIrRVByx-gBLUUm4,5672
 qulab/executor/cli.py,sha256=fTB7V8xNCvpU8EMa__ysBvRCzruP-QFbXOyLpeTPx0E,25711
 qulab/executor/load.py,sha256=eMlzOrrn8GpbP3J3uY5JJ8iO7tL5B1DWP1z2qiGyNhU,20385
-qulab/executor/registry.py,sha256=gym9F5FIDY5eV-cSCZsP99wC4l-6jkx9VMjJMaPOLaQ,4730
+qulab/executor/registry.py,sha256=CPY5PNTbE4Rw13Ucsq1wBk3fgvoSB3_J-iyRFdNNl0g,9044
 qulab/executor/schedule.py,sha256=cDFM_tIMvxHLxb-tBA6fEPF9u-4LcgnQDxbaquZIQKc,21727
 qulab/executor/storage.py,sha256=8K73KGLAVgchJdtd4rKHXkr1CQOJORWH-Gi57w8IYsw,21081
-qulab/executor/template.py,sha256=D8D7_B__R8wh1oOXk7CKZveRqDix4dNeUjefbfN6m9w,10720
+qulab/executor/template.py,sha256=Fxou5lglaCnEhJTKDotFaRrWT4csAZIgmHz9EFbPnYs,10811
 qulab/executor/utils.py,sha256=YSUEYjyYTJTjiJmd-l5GTEnSyD6bqEdNoWhivEAfGZA,7112
 qulab/monitor/__init__.py,sha256=FNYMIUr8ycRM4XmsAGpRK-A-SQsJZlcRWghXcGTgAi0,44
 qulab/monitor/__main__.py,sha256=Txy4d7vgVYUWwiHxX3uwtsi1ewde-MaiSC_K0ePoPiM,1372
@@ -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=CuLfG1jw1032HNNZxzC0OWtNzKsbj2e2WRYhMWDgvMQ,1321
 qulab/visualization/widgets.py,sha256=6KkiTyQ8J-ei70LbPQZAK35wjktY47w2IveOa682ftA,3180
-qulab-2.11.13.dist-info/licenses/LICENSE,sha256=PRzIKxZtpQcH7whTG6Egvzl1A0BvnSf30tmR2X2KrpA,1065
-qulab-2.11.13.dist-info/METADATA,sha256=Bo0lvlENxUP4QCQsnn0VzTpNuJAagUPHwP3DZXWiif4,3945
-qulab-2.11.13.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-qulab-2.11.13.dist-info/entry_points.txt,sha256=b0v1GXOwmxY-nCCsPN_rHZZvY9CtTbWqrGj8u1m8yHo,45
-qulab-2.11.13.dist-info/top_level.txt,sha256=3T886LbAsbvjonu_TDdmgxKYUn939BVTRPxPl9r4cEg,6
-qulab-2.11.13.dist-info/RECORD,,
+qulab-2.11.15.dist-info/licenses/LICENSE,sha256=PRzIKxZtpQcH7whTG6Egvzl1A0BvnSf30tmR2X2KrpA,1065
+qulab-2.11.15.dist-info/METADATA,sha256=wwaZbMBIXvh3I9Gc2xzjmwqK10kmhsFaTHVTs11Ypl0,3945
+qulab-2.11.15.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+qulab-2.11.15.dist-info/entry_points.txt,sha256=b0v1GXOwmxY-nCCsPN_rHZZvY9CtTbWqrGj8u1m8yHo,45
+qulab-2.11.15.dist-info/top_level.txt,sha256=3T886LbAsbvjonu_TDdmgxKYUn939BVTRPxPl9r4cEg,6
+qulab-2.11.15.dist-info/RECORD,,