mirror of
https://github.com/bytedance/deer-flow.git
synced 2026-10-26 21:26:39 +00:00
3f3cd23965
* fix(config): avoid Unicode failure after generating configuration * fix(setup): handle legacy stdout encodings in setup commands --------- Co-authored-by: Willem Jiang <willem.jiang@gmail.com>
59 lines
1.6 KiB
Python
59 lines
1.6 KiB
Python
#!/usr/bin/env python3
|
|
"""Cross-platform config bootstrap script for DeerFlow."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import shutil
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
|
|
def copy_if_missing(src: Path, dst: Path) -> None:
|
|
if dst.exists():
|
|
return
|
|
if not src.exists():
|
|
raise FileNotFoundError(f"Missing template file: {src}")
|
|
dst.parent.mkdir(parents=True, exist_ok=True)
|
|
shutil.copyfile(src, dst)
|
|
|
|
|
|
def main() -> int:
|
|
project_root = Path(__file__).resolve().parent.parent
|
|
|
|
existing_config = [
|
|
project_root / "config.yaml",
|
|
project_root / "config.yml",
|
|
project_root / "configure.yml",
|
|
]
|
|
|
|
if any(path.exists() for path in existing_config):
|
|
print(
|
|
"Error: configuration file already exists "
|
|
"(config.yaml/config.yml/configure.yml). Aborting."
|
|
)
|
|
return 1
|
|
|
|
try:
|
|
copy_if_missing(project_root / "config.example.yaml", project_root / "config.yaml")
|
|
copy_if_missing(project_root / ".env.example", project_root / ".env")
|
|
copy_if_missing(
|
|
project_root / "frontend" / ".env.example",
|
|
project_root / "frontend" / ".env",
|
|
)
|
|
except (FileNotFoundError, OSError) as exc:
|
|
print("Error while generating configuration files:")
|
|
print(f" {exc}")
|
|
if isinstance(exc, PermissionError):
|
|
print(
|
|
"Hint: Check file permissions and ensure the files are not "
|
|
"read-only or locked by another process."
|
|
)
|
|
return 1
|
|
|
|
print("Configuration files generated")
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(main())
|