QuLab 2.4.13__cp310-cp310-macosx_10_9_universal2.whl → 2.4.15__cp310-cp310-macosx_10_9_universal2.whl

{QuLab-2.4.13.dist-info → QuLab-2.4.15.dist-info}/METADATA

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

{QuLab-2.4.13.dist-info → QuLab-2.4.15.dist-info}/RECORD

@@ -1,15 +1,17 @@
 qulab/__init__.py,sha256=P-Mx2p4TVmL91SoxoeXcj8Qm0x4xUf5Q_FLk0Yc_gIQ,138
-qulab/__main__.py,sha256=ThCRooT22_qpwFtbAvVOyHzpBCuGlfDVkXBXvApzgps,597
+qulab/__main__.py,sha256=JgErYkiskih8Y6oRwowALtR-rwQhAAdqOYWjQraRIPI,59
 qulab/dicttree.py,sha256=tRRMpGZYVOLw0TEByE3_2Ss8FdOmzuGL9e1DWbs8qoY,13684
-qulab/fun.cpython-310-darwin.so,sha256=E7G2FLvF6TCJtlPXbd6ShtFPFUi5PKNfPpQrGDj6mn8,126864
-qulab/version.py,sha256=_e118vufispiK25vPHegUtgNTXX7-QuWeU66JWddGAE,22
+qulab/fun.cpython-310-darwin.so,sha256=cTaVtCuQtSu53XooP6NYgZR_6zBKaLaSZDSpPWHw3wk,126864
+qulab/version.py,sha256=QW2MBz1JeuFtoyMh-qAewXiVGQGFzZAiBx69V3wIX5U,22
+qulab/cli/__init__.py,sha256=tgDIkkeIoasQXAifJZ6NU8jDgpNgb2a-B0C4nF0evrE,559
+qulab/cli/config.py,sha256=QksTYD3-RBYBG8xnKLAh7EXrNOwxr128XXr30ZTFW6A,2983
 qulab/executor/__init__.py,sha256=LosPzOMaljSZY1thy_Fxtbrgq7uubJszMABEB7oM7tU,101
-qulab/executor/__main__.py,sha256=qgZcdmx1wBZInHf6ionpuuZytIJnaDje9WRv-a8o2OM,5870
-qulab/executor/load.py,sha256=2gpBUcVdrZcbVI-YqTAfyKhW9br4rO8gnQtoYv7AO6c,9167
-qulab/executor/schedule.py,sha256=Ww0HHRSJxk-V3FM-Qgwqn-l6qPyWjPNoCz9Yy8MVG4o,11663
+qulab/executor/cli.py,sha256=LRSb5-J8Tn7dj-fxwCp3OP_6YUgPRoY-f8O18sbWG8I,5164
+qulab/executor/load.py,sha256=4qQrw9O5dmWYYwxWt15z7uc6B3YuRB7xKgKi0g8-YtA,10418
+qulab/executor/schedule.py,sha256=SfMZys8WdmIvfW5fPM42g-F6XwewTdpC0-puhf5TA54,11721
 qulab/executor/storage.py,sha256=M66Q5_Uc5MMfc_QAuuaaexwAz7wxBPMkeleB5nRpQmI,4621
 qulab/executor/transform.py,sha256=inaOn6eqCs22ZZ0xAQl8s8YCoEACaXSwFNNu7jqdwAk,2148
-qulab/executor/utils.py,sha256=n3uCSKh-qdDFFeNvOpj7_es2_B4AaC-ASAlV9gPmSO0,3086
+qulab/executor/utils.py,sha256=jp3F0swO9gpKhmTauxbgEK9WVorjI9FAL6zT60cYTdk,3106
 qulab/monitor/__init__.py,sha256=nTHelnDpxRS_fl_B38TsN0njgq8eVTEz9IAnN3NbDlM,42
 qulab/monitor/__main__.py,sha256=w3yUcqq195LzSnXTkQcuC1RSFRhy4oQ_PEBmucXguME,97
 qulab/monitor/config.py,sha256=fQ5JcsMApKc1UwANEnIvbDQZl8uYW0tle92SaYtX9lI,744
@@ -89,9 +91,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.4.13.dist-info/LICENSE,sha256=PRzIKxZtpQcH7whTG6Egvzl1A0BvnSf30tmR2X2KrpA,1065
-QuLab-2.4.13.dist-info/METADATA,sha256=oq0Nmb1rdu0TDEPYZWZ9JbWT7-qrfRIWu0osA1W3eq0,3699
-QuLab-2.4.13.dist-info/WHEEL,sha256=Yd3eJSBM2hj8W-ouaiMfFUwQYAS-D6P73Ob9yN5MZd0,114
-QuLab-2.4.13.dist-info/entry_points.txt,sha256=b0v1GXOwmxY-nCCsPN_rHZZvY9CtTbWqrGj8u1m8yHo,45
-QuLab-2.4.13.dist-info/top_level.txt,sha256=3T886LbAsbvjonu_TDdmgxKYUn939BVTRPxPl9r4cEg,6
-QuLab-2.4.13.dist-info/RECORD,,
+QuLab-2.4.15.dist-info/LICENSE,sha256=PRzIKxZtpQcH7whTG6Egvzl1A0BvnSf30tmR2X2KrpA,1065
+QuLab-2.4.15.dist-info/METADATA,sha256=EgZbZvkhzL5UTZFxqJ2yHVir2RSvAkmSCNd_nUgitdE,3699
+QuLab-2.4.15.dist-info/WHEEL,sha256=Yd3eJSBM2hj8W-ouaiMfFUwQYAS-D6P73Ob9yN5MZd0,114
+QuLab-2.4.15.dist-info/entry_points.txt,sha256=b0v1GXOwmxY-nCCsPN_rHZZvY9CtTbWqrGj8u1m8yHo,45
+QuLab-2.4.15.dist-info/top_level.txt,sha256=3T886LbAsbvjonu_TDdmgxKYUn939BVTRPxPl9r4cEg,6
+QuLab-2.4.15.dist-info/RECORD,,

qulab/__main__.py

@@ -1,32 +1,4 @@
-import click
-
-from .executor.__main__ import create, get, maintain, run, set
-from .monitor.__main__ import main as monitor
-from .scan.server import server
-from .sys.net.cli import dht
-from .visualization.__main__ import plot
-
-
-@click.group()
-def cli():
-    pass
-
-
-@cli.command()
-def hello():
-    """Print hello world."""
-    click.echo('hello, world')
-
-
-cli.add_command(monitor)
-cli.add_command(plot)
-cli.add_command(dht)
-cli.add_command(server)
-cli.add_command(maintain)
-cli.add_command(run)
-cli.add_command(create)
-cli.add_command(set)
-cli.add_command(get)
+from .cli import cli
 
 if __name__ == '__main__':
     cli()

qulab/cli/__init__.py

@@ -0,0 +1,29 @@
+import click
+
+from ..executor.cli import create, get, maintain, run, set
+from ..monitor.__main__ import main as monitor
+from ..scan.server import server
+from ..sys.net.cli import dht
+from ..visualization.__main__ import plot
+
+
+@click.group()
+def cli():
+    pass
+
+
+@cli.command()
+def hello():
+    """Print hello world."""
+    click.echo('hello, world')
+
+
+cli.add_command(monitor)
+cli.add_command(plot)
+cli.add_command(dht)
+cli.add_command(server)
+cli.add_command(maintain)
+cli.add_command(run)
+cli.add_command(create)
+cli.add_command(set)
+cli.add_command(get)

qulab/cli/config.py

@@ -0,0 +1,88 @@
+import configparser
+import functools
+import os
+import sys
+from pathlib import Path
+
+import click
+from loguru import logger
+
+CONFIG_PATH = os.path.expanduser("~/.qulab.ini")
+ENV_PREFIX = "QULAB_"
+
+
+def _get_config_value(option_name, type_cast=str, command_name=None):
+    """支持命令专属配置的优先级获取"""
+    # 构造环境变量名
+    if command_name:
+        env_var = f"{ENV_PREFIX}{command_name.upper()}_{option_name.upper()}"
+        config_section = command_name
+    else:
+        env_var = f"{ENV_PREFIX}{option_name.upper()}"
+        config_section = "common"
+
+    # 1. 检查环境变量
+    if env_value := os.getenv(env_var):
+        if type_cast is bool:
+            return env_value.lower() in ("true", "1", "yes")
+        if "path" in option_name:
+            return os.path.expanduser(env_value)
+        return type_cast(env_value)
+
+    # 2. 检查配置文件
+    config = configparser.ConfigParser()
+    # 先加载默认配置防止段不存在
+    config.read_dict({config_section: {}})
+    if Path(CONFIG_PATH).exists():
+        config.read(CONFIG_PATH)
+
+    # 从对应配置段读取
+    if config.has_section(config_section):
+        if config_value := config.get(config_section,
+                                      option_name,
+                                      fallback=None):
+            if type_cast is bool:
+                return config_value.lower() in ("true", "1", "yes")
+            if "path" in option_name:
+                return os.path.expanduser(config_value)
+            return type_cast(config_value)
+
+    return None  # 交给 Click 处理默认值
+
+
+def get_config_value(option_name, type_cast=str, command_name=None):
+    value = _get_config_value(option_name, type_cast, command_name)
+    if value is None and command_name is not None:
+        return _get_config_value(option_name, type_cast)
+    return value
+
+
+def log_options(func):
+    """通用配置装饰器（所有命令共用）"""
+
+    @click.option("--debug",
+                  is_flag=True,
+                  default=lambda: get_config_value("debug", bool) or False,
+                  help=f"Enable debug mode")
+    @click.option("--log",
+                  type=click.Path(),
+                  default=lambda: get_config_value("log"),
+                  help=f"Log file path")
+    @functools.wraps(func)
+    def wrapper(*args, log=None, debug=False, **kwargs):
+        if log is None and not debug:
+            logger.remove()
+            logger.add(sys.stderr, level='INFO')
+        elif log is None and debug:
+            logger.remove()
+            logger.add(sys.stderr, level='DEBUG')
+        elif log is not None and not debug:
+            logger.configure(handlers=[dict(sink=log, level='INFO')])
+        elif log is not None and debug:
+            logger.configure(handlers=[
+                dict(sink=log, level='DEBUG'),
+                dict(sink=sys.stderr, level='DEBUG')
+            ])
+        return func(*args, **kwargs)
+
+    return wrapper

qulab/executor/cli.py

@@ -0,0 +1,178 @@
+import functools
+import importlib
+import os
+import sys
+from pathlib import Path
+
+import click
+from loguru import logger
+
+from ..cli.config import ENV_PREFIX, get_config_value, log_options
+from .load import find_unreferenced_workflows, get_entries, load_workflow
+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
+
+
+def command_option(command_name):
+    """命令专属配置装饰器工厂"""
+
+    def decorator(func):
+
+        @click.option(
+            '--code',
+            '-c',
+            default=lambda: get_config_value("code", str, command_name),
+            help='The path of the code.')
+        @click.option(
+            '--data',
+            '-d',
+            default=lambda: get_config_value("data", str, command_name),
+            help='The path of the data.')
+        @click.option(
+            '--api',
+            '-a',
+            default=lambda: get_config_value("api", str, command_name),
+            help='The modlule name of the api.')
+        @functools.wraps(func)
+        def wrapper(*args, **kwargs):
+            return func(*args, **kwargs)
+
+        return wrapper
+
+    return decorator
+
+
+@click.command()
+@click.argument('workflow')
+@click.option('--code',
+              '-c',
+              default=lambda: get_config_value("code", str, 'create'),
+              help='The path of the code.')
+def create(workflow, code):
+    """
+    Create a new workflow file.
+    """
+    if code is None:
+        code = Path.cwd()
+
+    fname = Path(code) / f'{workflow}'
+    fname = Path(os.path.expanduser(fname))
+    if fname.exists():
+        click.echo(f'{workflow} already exists')
+        return
+
+    fname.parent.mkdir(parents=True, exist_ok=True)
+    deps = find_unreferenced_workflows(code)
+
+    with open(fname, 'w') as f:
+        f.write(workflow_template(list(deps)))
+    click.echo(f'{workflow} created')
+
+
+@click.command()
+@click.argument('key')
+@click.argument('value', type=str)
+@click.option('--api',
+              '-a',
+              default=lambda: get_config_value("api", str, 'set'),
+              help='The modlule name of the api.')
+def set(key, value, api):
+    """
+    Set a config.
+    """
+    from . import transform
+    if api is not None:
+        api = importlib.import_module(api)
+        set_config_api(api.query_config, api.update_config)
+    try:
+        value = eval(value)
+    except:
+        pass
+    transform.update_config({key: value})
+
+
+@click.command()
+@click.argument('key')
+@click.option('--api',
+              '-a',
+              default=lambda: get_config_value("api", str, 'get'),
+              help='The modlule name of the api.')
+def get(key, api):
+    """
+    Get a config.
+    """
+    from . import transform
+    if api is not None:
+        api = importlib.import_module(api)
+        set_config_api(api.query_config, api.update_config)
+    click.echo(transform.query_config(key))
+
+
+@click.command()
+@click.argument('workflow')
+@click.option('--plot', '-p', is_flag=True, help='Plot the result.')
+@click.option('--no-dependents',
+              '-n',
+              is_flag=True,
+              help='Do not run dependents.')
+@log_options
+@command_option('run')
+def run(workflow, code, data, api, plot, no_dependents):
+    """
+    Run a workflow.
+    """
+    if api is not None:
+        api = importlib.import_module(api)
+        set_config_api(api.query_config, api.update_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))
+
+    wf = load_workflow(workflow, code)
+
+    if no_dependents:
+        if hasattr(wf, 'entries'):
+            for entry in get_entries(wf, code):
+                run_workflow(entry, code, data, plot=plot)
+        else:
+            run_workflow(wf, code, data, plot=plot)
+    else:
+        if hasattr(wf, 'entries'):
+            for entry in get_entries(wf, code):
+                maintain_workflow(entry, code, data, run=True, plot=plot)
+        else:
+            maintain_workflow(wf, code, data, run=True, plot=plot)
+
+
+@click.command()
+@click.argument('workflow')
+@click.option('--plot', '-p', is_flag=True, help='Plot the result.')
+@log_options
+@command_option('maintain')
+def maintain(workflow, code, data, api, plot):
+    """
+    Maintain a workflow.
+    """
+    if api is not None:
+        api = importlib.import_module(api)
+        set_config_api(api.query_config, api.update_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))
+
+    wf = load_workflow(workflow, code)
+    if hasattr(wf, 'entries'):
+        for entry in get_entries(wf, code):
+            maintain_workflow(entry, code, data, run=False, plot=plot)
+    else:
+        maintain_workflow(wf, code, data, run=False, plot=plot)

qulab/executor/load.py

@@ -119,6 +119,17 @@ def is_workflow(module: ModuleType) -> bool:
         return False
 
 
+def is_template(path: str | Path) -> bool:
+    path = Path(path)
+    if path.name == 'template.py':
+        return True
+    if path.name.endswith('_template.py'):
+        return True
+    if 'templates' in path.parts:
+        return True
+    return False
+
+
 def find_unreferenced_workflows(path: str) -> list[str]:
     root = Path(path).resolve()
     workflows = []
@@ -128,15 +139,14 @@ def find_unreferenced_workflows(path: str) -> list[str]:
     for file_path in root.rglob("*.py"):
         if file_path.name == "__init__.py":
             continue
-        if file_path.name == 'template.py' or file_path.name.endswith(
-                "_template.py"):
+        if is_template(file_path):
             continue
         try:
             rel_path = file_path.relative_to(root)
         except ValueError:
             continue
 
-        module = load_workflow_from_module(str(rel_path), root)
+        module = load_workflow_from_file(str(rel_path), root)
 
         if is_workflow(module):
             rel_str = str(rel_path)
@@ -147,7 +157,7 @@ def find_unreferenced_workflows(path: str) -> list[str]:
 
     # Check dependencies for each workflow module
     for rel_str in workflows:
-        module = load_workflow_from_module(rel_str, root)
+        module = load_workflow_from_file(rel_str, root)
 
         depends_func = getattr(module, "depends", None)
         if depends_func and callable(depends_func):
@@ -157,7 +167,9 @@ def find_unreferenced_workflows(path: str) -> list[str]:
                 )
                 continue
             try:
-                depends_list = [n.__workflow_id__ for n in get_dependents(module, root)]
+                depends_list = [
+                    n.__workflow_id__ for n in get_dependents(module, root)
+                ]
             except Exception as e:
                 warnings.warn(f"Error calling depends() in {rel_str}: {e}")
                 continue
@@ -183,11 +195,9 @@ def find_unreferenced_workflows(path: str) -> list[str]:
     return unreferenced
 
 
-def load_workflow_from_module(file_name: str,
-                              base_path: str | Path,
-                              package='workflows') -> WorkflowType:
-    if file_name.startswith('cfg:'):
-        return SetConfigWorkflow(file_name[4:])
+def load_workflow_from_file(file_name: str,
+                            base_path: str | Path,
+                            package='workflows') -> WorkflowType:
     base_path = Path(base_path)
     path = Path(file_name)
     module_name = f"{package}.{'.'.join([*path.parts[:-1], path.stem])}"
@@ -195,6 +205,9 @@ def load_workflow_from_module(file_name: str,
     module = module_from_spec(spec)
     spec.loader.exec_module(module)
 
+    if hasattr(module, 'entries'):
+        return module
+
     if not hasattr(module, '__timeout__'):
         module.__timeout__ = None
 
@@ -207,37 +220,63 @@ def load_workflow_from_module(file_name: str,
     return module
 
 
-def load_workflow_from_template(file_name: str,
+def load_workflow_from_template(template_path: str,
                                 mappping: dict[str, str],
                                 base_path: str | Path,
-                                subtitle: str | None = None,
+                                target_path: str | None = None,
                                 package='workflows') -> WorkflowType:
     base_path = Path(base_path)
-    path = Path(file_name)
+    path = Path(template_path)
 
     with open(base_path / path) as f:
         content = f.read()
-    template = string.Template(content)
+
+    def replace(text):
+        """
+        将给定文本中的所有 _D_("var") 替换为 ${var}。
+        
+        Args:
+            text (str): 包含 _D_ 调用的字符串。
+        
+        Returns:
+            str: 已经替换的新字符串。
+        """
+        pattern = re.compile(r'_D_\s*\(\s*(["\'])(\w+)\1\s*\)')
+        replacement = r'${\2}'
+        new_text = re.sub(pattern, replacement, text)
+        return new_text
+
+    template = string.Template(replace(content))
     content = template.substitute(mappping)
 
     hash_str = hashlib.md5(pickle.dumps(mappping)).hexdigest()[:8]
-    if subtitle is None:
+    if target_path is None:
         if path.stem == 'template':
             path = path.parent / f'tmp{hash_str}.py'
-        else:
+        elif path.stem.endswith('_template'):
             path = path.parent / path.stem.replace('_template',
                                                    f'_tmp{hash_str}.py')
-    else:
-        if path.stem == 'template':
-            path = path.parent / f'{subtitle}.py'
         else:
-            path = path.parent / path.stem.replace('_template',
-                                                   f'_{subtitle}.py')
+            path = path.parent / f'{path.stem}_tmp{hash_str}.py'
+    else:
+        path = target_path
 
-    with open(base_path / path, 'w') as f:
-        f.write(content)
+    file = base_path / path
+    if not file.exists():
+        file.parent.mkdir(parents=True, exist_ok=True)
+        with open(file, 'w') as f:
+            f.write(content)
+    else:
+        if file.stat().st_mtime < Path(template_path).stat().st_mtime:
+            with open(file, 'w') as f:
+                f.write(content)
+        else:
+            if file.read_text() != content:
+                logger.warning(
+                    f"`{file}` already exists and is different from the new one generated from template `{template_path}`"
+                )
 
-    module = load_workflow_from_module(str(path), base_path, package)
+    module = load_workflow_from_file(str(path), base_path, package)
 
     return module
 
@@ -251,9 +290,9 @@ def load_workflow(workflow: str | tuple[str, dict],
             w = load_workflow_from_template(file_name, mapping, base_path,
                                             None, package)
         elif len(workflow) == 3:
-            file_name, subtitle, mapping = workflow
-            w = load_workflow_from_template(file_name, mapping, base_path,
-                                            subtitle, package)
+            template_path, target_path, mapping = workflow
+            w = load_workflow_from_template(template_path, mapping, base_path,
+                                            target_path, package)
         else:
             raise ValueError(f"Invalid workflow: {workflow}")
         w.__workflow_id__ = str(Path(w.__file__).relative_to(base_path))
@@ -263,7 +302,7 @@ def load_workflow(workflow: str | tuple[str, dict],
             w = SetConfigWorkflow(key)
             w.__workflow_id__ = workflow
         else:
-            w = load_workflow_from_module(workflow, base_path, package)
+            w = load_workflow_from_file(workflow, base_path, package)
             w.__workflow_id__ = str(Path(w.__file__).relative_to(base_path))
     else:
         raise TypeError(f"Invalid workflow: {workflow}")
@@ -274,3 +313,7 @@ def load_workflow(workflow: str | tuple[str, dict],
 def get_dependents(workflow: WorkflowType,
                    code_path: str | Path) -> list[WorkflowType]:
     return [load_workflow(n, code_path) for n in workflow.depends()[0]]
+
+
+def get_entries(workflow: WorkflowType, code_path: str | Path) -> WorkflowType:
+    return [load_workflow(n, code_path) for n in workflow.entries()]

qulab/executor/schedule.py

@@ -289,8 +289,9 @@ def run(workflow: WorkflowType,
         code_path: str | Path,
         state_path: str | Path,
         plot: bool = False):
+    session_id = uuid.uuid4().hex
     logger.debug(f'run "{workflow.__workflow_id__}" without dependences.')
-    result = calibrate(workflow, code_path, state_path, plot)
+    result = calibrate(workflow, code_path, state_path, plot, session_id=session_id)
     if result.bad_data or not result.in_spec:
         raise CalibrationFailedError(
             f'"{workflow.__workflow_id__}": All dependents passed, but calibration failed!'

qulab/executor/utils.py

@@ -34,6 +34,8 @@ def workflow_template(deps: list[str]) -> str:
     return f"""
 from loguru import logger
 
+def _D_(s): pass
+
 import numpy as np
 
 

qulab/version.py

@@ -1 +1 @@
-__version__ = "2.4.13"
+__version__ = "2.4.15"

qulab/executor/__main__.py

@@ -1,200 +0,0 @@
-import functools
-import importlib
-import os
-import sys
-from pathlib import Path
-
-import click
-from loguru import logger
-
-from .load import find_unreferenced_workflows, load_workflow
-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
-
-
-def load_config():
-    import yaml
-
-    config_paths = [
-        Path.home() / ".myapp/config.yaml",  # 用户主目录
-        Path("config.yaml")  # 当前目录
-    ]
-    for path in config_paths:
-        if path.exists():
-            with open(path) as f:
-                return yaml.safe_load(f)
-    return {"defaults": {"log": "default.log", "debug": False}}
-
-
-def get_config_value(option_name):
-    # 1. 尝试从环境变量读取
-    env_value = os.environ.get(f"MYAPP_{option_name.upper()}")
-    if env_value:
-        return env_value
-
-    # 2. 尝试从配置文件读取
-    config = load_config()
-    return config["defaults"].get(option_name)
-
-
-def log_options(func):
-
-    @click.option("--debug", is_flag=True, help="Enable debug mode.")
-    @click.option("--log", type=str, help="Log file path.")
-    @functools.wraps(func)  # 保持函数元信息
-    def wrapper(*args, log=None, debug=False, **kwargs):
-        if log is None and not debug:
-            logger.remove()
-            logger.add(sys.stderr, level='INFO')
-        elif log is None and debug:
-            logger.remove()
-            logger.add(sys.stderr, level='DEBUG')
-        elif log is not None and not debug:
-            logger.configure(handlers=[dict(sink=log, level='INFO')])
-        elif log is not None and debug:
-            logger.configure(handlers=[
-                dict(sink=log, level='DEBUG'),
-                dict(sink=sys.stderr, level='DEBUG')
-            ])
-        return func(*args, **kwargs)
-
-    return wrapper
-
-
-@click.group()
-def cli():
-    pass
-
-
-@click.command()
-@click.argument('workflow')
-@click.option('--code', '-c', default=None, help='The path of the code.')
-def create(workflow, code):
-    """
-    Create a new workflow file.
-    """
-    if code is None:
-        code = Path.cwd()
-
-    fname = Path(code) / f'{workflow}'
-    fname = Path(os.path.expanduser(fname))
-    if fname.exists():
-        click.echo(f'{workflow} already exists')
-        return
-
-    fname.parent.mkdir(parents=True, exist_ok=True)
-    deps = find_unreferenced_workflows(code)
-
-    with open(fname, 'w') as f:
-        f.write(workflow_template(list(deps)))
-    click.echo(f'{workflow} created')
-
-
-@click.command()
-@click.argument('key')
-@click.argument('value', type=str)
-@click.option('--api', '-a', default=None, help='The modlule name of the api.')
-def set(key, value, api):
-    """
-    Set a config.
-    """
-    from . import transform
-    if api is not None:
-        api = importlib.import_module(api)
-        set_config_api(api.query_config, api.update_config)
-    try:
-        value = eval(value)
-    except:
-        pass
-    transform.update_config({key: value})
-
-
-@click.command()
-@click.argument('key')
-@click.option('--api', '-a', default=None, help='The modlule name of the api.')
-def get(key, api):
-    """
-    Get a config.
-    """
-    from . import transform
-    if api is not None:
-        api = importlib.import_module(api)
-        set_config_api(api.query_config, api.update_config)
-    click.echo(transform.query_config(key))
-
-
-@click.command()
-@click.argument('workflow')
-@click.option('--code', '-c', default=None, help='The path of the code.')
-@click.option('--data', '-d', default=None, help='The path of the data.')
-@click.option('--api', '-a', default=None, help='The modlule name of the api.')
-@click.option('--plot', '-p', is_flag=True, help='Plot the result.')
-@click.option('--no-dependents',
-              '-n',
-              is_flag=True,
-              help='Do not run dependents.')
-@log_options
-def run(workflow, code, data, api, plot, no_dependents):
-    """
-    Run a workflow.
-    """
-    if api is not None:
-        api = importlib.import_module(api)
-        set_config_api(api.query_config, api.update_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))
-
-    if no_dependents:
-        run_workflow(load_workflow(workflow, code), code, data, plot=plot)
-    else:
-        maintain_workflow(load_workflow(workflow, code),
-                          code,
-                          data,
-                          run=True,
-                          plot=plot)
-
-
-@click.command()
-@click.argument('workflow')
-@click.option('--code', '-c', default=None, help='The path of the code.')
-@click.option('--data', '-d', default=None, help='The path of the data.')
-@click.option('--api', '-a', default=None, help='The modlule name of the api.')
-@click.option('--plot', '-p', is_flag=True, help='Plot the result.')
-@log_options
-def maintain(workflow, code, data, api, plot):
-    """
-    Maintain a workflow.
-    """
-    if api is not None:
-        api = importlib.import_module(api)
-        set_config_api(api.query_config, api.update_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))
-
-    maintain_workflow(load_workflow(workflow, code),
-                      code,
-                      data,
-                      run=False,
-                      plot=plot)
-
-
-cli.add_command(maintain)
-cli.add_command(run)
-cli.add_command(create)
-cli.add_command(set)
-cli.add_command(get)
-
-if __name__ == '__main__':
-    cli()