qn-util 0.0.0__py3-none-any.whl

qn_util/__init__.py

@@ -0,0 +1,11 @@
+# from rmc_dft.rmc_pot.qn.config_error import ConfigError
+# from rmc_dft.rmc_pot.qn.qn_network_analysis import QnNetworkAnalysis
+# from rmc_dft.rmc_pot.qn.qn_network_builder import QnNetworkBuilder
+# from rmc_dft.rmc_pot.qn.rmc_constraint_writer import RmcConstraintWriter
+# from rmc_dft.rmc_pot.qn.vasp_structure import VaspStructure
+
+from qn.config_error import ConfigError
+from qn.qn_network_analysis import QnNetworkAnalysis
+from qn.qn_network_builder import QnNetworkBuilder
+from qn.rmc_constraint_writer import RmcConstraintWriter
+from qn.vasp_structure import VaspStructure

qn_util/config_error.py

@@ -0,0 +1,11 @@
+"""
+K.NAKADA, kengo.nakada@gmail.com
+"""
+
+class ConfigError(Exception):
+    """
+    設定ファイルまたは入力構造が不正な場合の例外。
+
+    JSON 設定の必須キー不足、POSCAR の形式不整合、未知の constraint source など、
+    ユーザー入力に起因するエラーをこの例外にまとめる。
+    """

qn_util/install.py

@@ -0,0 +1,49 @@
+"""qn_util に付属するサンプルスクリプトを作業ディレクトリへ配置する。"""
+
+import argparse
+import shutil
+from importlib.resources import files
+from pathlib import Path
+
+
+def install(force: bool = False, destination: Path | None = None) -> list[Path]:
+    """同梱 scripts の全ファイルを指定先（既定はカレント）へコピーする。"""
+    target = Path.cwd() if destination is None else Path(destination)
+    source = files("qn_util").joinpath("scripts")
+    if not source.is_dir():
+        raise FileNotFoundError("パッケージ内に scripts がありません。配布設定を確認してください。")
+    items = list(source.iterdir())
+    conflicts = [target / item.name for item in items if (target / item.name).exists()]
+    if conflicts and not force:
+        names = ", ".join(str(x) for x in conflicts)
+        raise FileExistsError(f"同名ファイルが存在します: {names}（上書きには --force を指定）")
+    copied = []
+    for item in items:
+        dst = target / item.name
+        if item.is_dir():
+            shutil.copytree(str(item), dst, dirs_exist_ok=force)
+        else:
+            with item.open("rb") as src, dst.open("wb") as out:
+                shutil.copyfileobj(src, out)
+        copied.append(dst)
+    return copied
+
+
+def main() -> None:
+    """qn_util コマンドのエントリーポイント。"""
+    parser = argparse.ArgumentParser(prog="qn_util")
+    sub = parser.add_subparsers(dest="command", required=True)
+    cmd = sub.add_parser("install", help="付属スクリプトをカレントディレクトリへコピー")
+    cmd.add_argument("--force", action="store_true", help="同名ファイルを上書き")
+    args = parser.parse_args()
+    if args.command == "install":
+        try:
+            copied = install(force=args.force)
+        except (FileExistsError, FileNotFoundError) as exc:
+            parser.exit(1, f"エラー: {exc}\n")
+        for path in copied:
+            print(f"コピー: {path}")
+
+
+if __name__ == "__main__":
+    main()