import yaml

# 方法1：顯式指定 SafeLoader（推薦）
with open('config.yaml', 'r', encoding='utf-8') as f:
    data = yaml.load(f, Loader=yaml.SafeLoader)

# 方法2：使用安全加載快捷函數（更簡潔）
with open('config.yaml', 'r', encoding='utf-8') as f:
    data = yaml.safe_load(f)  # 內部使用 SafeLoader

# 字符串加載
yaml_text = """
name: 張三
age: 28
skills:
  - Python
  - YAML
"""
data = yaml.safe_load(yaml_text)
print(data)

# {'name': '張三', 'age': 28, 'skills': ['Python', 'YAML']}

# 處理多個 YAML 文檔（--- 分隔）
multi_doc = """
---
name: 文檔1
value: 100
---
name: 文檔2
value: 200
"""
# 生成器方式遍歷所有文檔
for doc in yaml.safe_load_all(multi_doc):
    print(doc)

import yaml
from yaml import SafeLoader

# 添加自定義標簽處理（保持安全邊界）
class MySafeLoader(SafeLoader):
    pass

# 注冊自定義構造器
def constructor_env_variable(loader, node):
    value = loader.construct_scalar(node)
    return os.path.expandvars(value)  # 展開環(huán)境變量

# 注冊 !!env 標簽
MySafeLoader.add_constructor('!env', constructor_env_variable)

# 使用自定義 Loader
yaml_text = "path: !env $HOME/config"
data = yaml.load(yaml_text, Loader=MySafeLoader)

from pydantic import BaseModel
import yaml

class Config(BaseModel):
    name: str
    port: int
    debug: bool = False

# 安全加載 + 模式驗證
with open('config.yaml') as f:
    raw_data = yaml.safe_load(f)
    config = Config(**raw_data)  # 驗證類型安全

# 危險：version 會被解析為字符串 "1.10" 還是數字？
version: 1.10      # 實際解析為浮點數 1.1
date: 2024-01-15  # 解析為日期對象，不是字符串！

# 強制字符串類型
version: !!str 1.10
date: !!str 2024-01-15

# 無限遞歸定義（SafeLoader 可安全處理，但需注意）
person: &anchor
  name: 張三
  friend: *anchor  # 循環(huán)引用

# SafeLoader 會正確解析為自引用對象
# 使用 copy.deepcopy 時需設置最大深度防棧溢出
import copy
data = yaml.safe_load(yaml_text)
safe_copy = copy.deepcopy(data)  # 可能拋出 RecursionError

import yaml
from yaml import YAMLError
try:
    with open('config.yaml', 'r', encoding='utf-8') as f:
        config = yaml.safe_load(f)
except YAMLError as e:
    print(f"YAML 解析錯誤: {e}")
except FileNotFoundError:
    print("配置文件不存在")