from docflow import DocflowClient, AuthScope

client = DocflowClient.from_env()

workspace = client.workspace.create(
    name="経費精算処理スペース",
    auth_scope=AuthScope.PUBLIC,  # 公開権限
    description="社内の経費精算書類を処理"
)

print(f"ワークスペース ID: {workspace.workspace_id}")

# ページングで取得
workspaces = client.workspace.list(page=1, page_size=20)

for ws in workspaces.workspaces:
    print(f"{ws.workspace_id}: {ws.name}")

# イテレータで自動ページング
for workspace in client.workspace.iter():
    print(f"{workspace.workspace_id}: {workspace.name}")

# ワークスペースをバインド
ws = client.workspace("123")

# ワークスペース詳細を取得
detail = ws.get()

# ワークスペースを更新
ws.update(
    name="新しいワークスペース名",
    auth_scope=AuthScope.PRIVATE
)

# メソッドチェーン: カテゴリを操作
cat = ws.category("456")
cat.fields.batch_add(fields=[{"name": "請求書番号"}])
cat.tables.batch_add(tables=[{"name": "商品明細テーブル"}])

from docflow import ExtractModel, FieldType

# フィールド設定を定義
fields = [
    {
        "name": "請求書番号",
        "description": "請求書の一意な識別番号"
    },
    {
        "name": "発行日",
        "transform_settings": {
            "type": FieldType.DATETIME.value,
            "datetime_settings": {
                "format": "yyyy-MM-dd"
            }
        }
    },
    {
        "name": "請求書種別",
        "transform_settings": {
            "type": FieldType.ENUMERATE.value,
            "enumerate_settings": {
                "items": ["適格請求書", "通常請求書"]
            }
        }
    }
]

# カテゴリを作成
category = client.category.create(
    workspace_id="123",
    name="請求書",
    extract_model=ExtractModel.Model_1,  # Model 1 を使用
    sample_files=[
        "/path/to/invoice_sample1.pdf",
        "/path/to/invoice_sample2.pdf"
    ],
    fields=fields,
    category_prompt="請求書。請求書情報と商品明細を含む"
)

print(f"カテゴリ ID: {category.category_id}")

# カテゴリ作成後、テーブルを追加
cat = client.workspace("123").category(category.category_id)
tables = cat.tables.batch_add(tables=[{
    "name": "商品明細テーブル",
    "prompt": "商品名、仕様、数量、単価、金額を抽出"
}])
print(f"テーブル ID: {tables[0].table_id}")

# カテゴリコンテキストをバインド
cat = client.workspace("123").category("456")

# フィールドを一括追加
result = cat.fields.batch_add(fields=[
    {"name": "税率", "description": "税率の割合"}
])

# フィールド一覧を取得
fields = cat.fields.list()
for field in fields.fields:
    print(f"{field.id}: {field.name}")

# フィールドを一括更新
cat.fields.batch_update(fields=[
    {"field_id": "789", "name": "税率（%）", "required": True}
])

# フィールドを削除
cat.fields.delete(field_ids=["789"])

# テーブルを一括追加（列フィールドも同時に作成）
tables = cat.tables.batch_add(tables=[{
    "name": "商品明細テーブル",
    "prompt": "商品名、仕様、数量、単価、金額を抽出",
    "fields": [
        {"name": "商品名"},
        {"name": "数量"},
        {"name": "単価"}
    ]
}])

# テーブル一覧を取得
tables = cat.tables.list()
for table in tables.tables:
    print(f"{table.table_id}: {table.name}")

# サンプルファイルを一括アップロード
result = cat.samples.batch_upload(files=[
    "/path/to/invoice_sample1.pdf",
    "/path/to/invoice_sample2.pdf"
])

# サンプル一覧を取得
samples = cat.samples.list()

# サンプルをダウンロード（ZIP）
cat.samples.batch_download(
    sample_ids=["789"],
    save_path="/path/to/samples.zip"
)

# サンプルを削除
cat.samples.delete(sample_ids=["789"])

# ファイルをアップロード（自動的に認識を開始）
response = client.file.upload(
    workspace_id="123",
    category="請求書",  # カテゴリ名
    file_path="/path/to/invoice.pdf"
)

batch_number = response.batch_number
print(f"バッチ番号: {batch_number}")

import time

# 認識完了までポーリング
max_wait = 60
wait_interval = 3
elapsed = 0

while elapsed < max_wait:
    result = client.file.fetch(
        workspace_id="123",
        batch_number=batch_number
    )

    if result.files and result.files[0].recognition_status == 1:
        file_info = result.files[0]
        print(f"認識完了: {file_info.name}")

        # 抽出されたフィールドデータにアクセス
        if file_info.data and 'fields' in file_info.data:
            for field in file_info.data['fields']:
                print(f"{field['name']}: {field['value']}")

        # テーブルデータにアクセス
        if file_info.data and 'items' in file_info.data:
            for row in file_info.data['items']:
                print(row)

        break

    time.sleep(wait_interval)
    elapsed += wait_interval

# 1. 規則リポジトリを作成
repo = client.review.create_repo(
    workspace_id="123",
    name="経費精算レビュー規則リポジトリ"
)

# 2. 規則グループを作成
group = client.review.create_group(
    workspace_id="123",
    repo_id=repo.repo_id,
    name="請求書コンプライアンスチェック"
)

# 3. レビュー規則を作成
# フィールド ID のマッピングを取得
fields_response = client.category.fields.list(
    workspace_id="123",
    category_id="456"
)
field_map = {f.name: f.id for f in fields_response.fields}

# 規則を作成: 必須フィールドの入力チェック
client.review.create_rule(
    workspace_id="123",
    repo_id=int(repo.repo_id),
    group_id=group.group_id,
    name="必須フィールド入力チェック",
    prompt='「請求書番号」「発行日」「金額」がすべて入力されているか確認し、いずれかが空の場合はレビュー不合格にする',
    category_ids=["456"],
    risk_level=10,  # 高リスク
    referenced_fields=[
        {
            "category_id": "456",
            "category_name": "請求書",
            "fields": [
                {"field_id": field_map["請求書番号"], "field_name": "請求書番号"},
                {"field_id": field_map["発行日"], "field_name": "発行日"},
                {"field_id": field_map["金額"], "field_name": "金額"}
            ],
            "tables": []
        }
    ]
)

# 複数文書をまたぐ規則を作成: 請求金額と支払金額の一致を確認
client.review.create_rule(
    workspace_id="123",
    repo_id=int(repo.repo_id),
    group_id=group.group_id,
    name="複数文書の金額照合",
    prompt="請求書の金額と支払記録の取引金額が一致するか確認し、±0.1 の誤差を許容する",
    category_ids=["456", "789"],  # 複数カテゴリ
    risk_level=10,
    referenced_fields=[
        {
            "category_id": "456",
            "category_name": "請求書",
            "fields": [
                {"field_id": invoice_field_map["金額"], "field_name": "金額"}
            ],
            "tables": []
        },
        {
            "category_id": "789",
            "category_name": "支払記録",
            "fields": [
                {"field_id": payment_field_map["取引金額"], "field_name": "取引金額"}
            ],
            "tables": []
        }
    ]
)

# レビュータスクを送信
review_task = client.review.submit_task(
    workspace_id="123",
    name="2024年3月 経費精算レビュー",
    repo_id=repo.repo_id,
    extract_task_ids=["task_001", "task_002", "task_003"]
)

task_id = review_task['task_id']
print(f"レビュータスク ID: {task_id}")

import time

# レビュー完了までポーリング
max_wait = 120
wait_interval = 5
elapsed = 0

while elapsed < max_wait:
    result = client.review.get_task_result(
        workspace_id="123",
        task_id=task_id
    )

    status = result.get('status')
    # ステータス: 0=レビュー待ち, 1=合格, 2=失敗, 4=不合格, 7=認識失敗
    if status in (1, 2, 4, 7):
        print("レビュー完了")

        # レビュー結果を出力
        stats = result.get('statistics', {})
        print(f"合格した規則数: {stats.get('pass_count', 0)}")
        print(f"不合格の規則数: {stats.get('failure_count', 0)}")

        # 詳細レビュー結果
        for group in result.get('groups', []):
            print(f"\n【{group['group_name']}】")
            for task in group.get('review_tasks', []):
                result_icon = "✓" if task['review_result'] == 0 else "✗"
                print(f"  {result_icon} {task['rule_name']}")
                print(f"    {task['reasoning']}")

        break

    time.sleep(wait_interval)
    elapsed += wait_interval

from docflow import (
    ExtractModel,      # 抽出モデル種別
    EnabledStatus,     # 有効状態（検索用）
    EnabledFlag,       # 有効フラグ（更新用）
    AuthScope,         # 権限スコープ
    FieldType,         # フィールド種別
    MismatchAction,    # 不一致時の処理方法
    RecognitionStatus, # 認識ステータス
)

# ExtractModel - 抽出モデル
ExtractModel.Model_1  # 高速で安定した抽出結果
ExtractModel.Model_2  # 複雑な文書理解に適用
ExtractModel.Model_3  # マルチモーダル、シンプルな抽出に適用

# AuthScope - 権限スコープ
AuthScope.PRIVATE  # 0 - 非公開権限
AuthScope.PUBLIC   # 1 - 公開権限

# EnabledStatus - 有効状態（検索用）
EnabledStatus.ALL       # "all" - すべて
EnabledStatus.DISABLED  # "0"   - 無効
EnabledStatus.ENABLED   # "1"   - 有効

# EnabledFlag - 有効フラグ（更新用）
EnabledFlag.DISABLED  # 0 - 無効
EnabledFlag.ENABLED   # 1 - 有効

# FieldType - フィールド変換種別
FieldType.DATETIME   # "datetime"   - 日時
FieldType.ENUMERATE  # "enumerate"  - 列挙
FieldType.REGEX      # "regex"      - 正規表現

# RecognitionStatus - 認識ステータス
RecognitionStatus.PENDING    # 0 - 認識待ち
RecognitionStatus.SUCCESS    # 1 - 認識成功
RecognitionStatus.FAILED     # 2 - 認識失敗

# ワークスペースイテレータ
for workspace in client.workspace.iter():
    print(f"{workspace.workspace_id}: {workspace.name}")
    if some_condition:
        break  # いつでも中断可能

# カテゴリイテレータ
for category in client.category.iter(workspace_id="123"):
    print(f"{category.category_id}: {category.name}")

# 最大ページ数を制限
for category in client.category.iter(workspace_id="123", max_pages=5):
    print(category.name)

# リストへ変換（すべてのデータを取得）
all_workspaces = list(client.workspace.iter())

from docflow.exceptions import (
    DocflowException,
    AuthenticationError,
    ValidationError,
    ResourceNotFoundError,
    APIError,
)

try:
    workspace = client.workspace.get(workspace_id="123")
except AuthenticationError as e:
    print(f"認証に失敗しました: {e.message}")
except ResourceNotFoundError as e:
    print(f"ワークスペースが見つかりません: {e.message}")
except ValidationError as e:
    print(f"パラメータ検証に失敗しました: {e.message}")
except APIError as e:
    print(f"API エラー [HTTP {e.status_code}]: {e.message}")
except DocflowException as e:
    print(f"SDK エラー: {e.message}")

from docflow import DocflowClient, set_language

# 英語を使用
set_language('en_US')
client = DocflowClient.from_env()

# 中国語を使用（デフォルト）
set_language('zh_CN')

# 言語を動的に切り替え
client.set_language('en_US')

# 現在の言語を取得
current_lang = client.get_language()  # 'en_US' または 'zh_CN'

client = DocflowClient(
    app_id="your-app-id",
    secret_code="your-secret-code",
    timeout=60,           # リクエストタイムアウト（秒）、デフォルト 30
    max_retries=5,        # 最大リトライ回数、デフォルト 3
    retry_backoff_factor=1.0,  # バックオフ係数、デフォルト 1.0
)

# リトライ対象ステータスコードを指定
client = DocflowClient(
    app_id="your-app-id",
    secret_code="your-secret-code",
    retry_status_codes=[429, 503],  # 429 と 503 のみリトライ
)

# リトライ対象メソッドを指定
client = DocflowClient(
    app_id="your-app-id",
    secret_code="your-secret-code",
    retry_methods=["GET"],  # GET リクエストのみリトライ
)

# リトライを無効化
client = DocflowClient(
    app_id="your-app-id",
    secret_code="your-secret-code",
    max_retries=0  # リトライを無効化
)

client = DocflowClient(
    app_id="your-app-id",
    secret_code="your-secret-code",
    base_url="https://custom-api.example.com"
)

with DocflowClient.from_env() as client:
    workspaces = client.workspace.list()
    # 終了時に接続を自動でクローズ

import logging
logging.getLogger("docflow").setLevel(logging.DEBUG)