first commit

2026-09-02 11:44:52 +08:00
commit 0c8fa2653e
309 changed files with 57278 additions and 0 deletions
@@ -0,0 +1,138 @@
+#!/usr/bin/env python3
+"""
+完整报销流程：搜索供应商 → 搜索类别 → 创建付款单
+
+用法:
+    python finance_expense_flow.py \
+        --amount 5000 \
+        --supplier "华为" \
+        --category "差旅" \
+        --category-type expense
+
+    python finance_expense_flow.py --dry-run --amount 1000
+"""
+
+import sys
+import json
+import subprocess
+import argparse
+from typing import List, Any, Optional
+
+
+def run_dws(
+    args: List[str], dry_run: bool = False,
+) -> Optional[Any]:
+    cmd = ['dws'] + args
+    if dry_run:
+        print(f"[dry-run] {' '.join(cmd)}")
+        return {'dry_run': True}
+    try:
+        result = subprocess.run(
+            cmd, capture_output=True, text=True, timeout=60
+        )
+        if result.returncode != 0:
+            print(f"  ✗ 错误：{result.stderr.strip()}")
+            return None
+        return json.loads(result.stdout)
+    except (subprocess.TimeoutExpired, json.JSONDecodeError,
+            FileNotFoundError) as e:
+        print(f"  ✗ 错误：{e}")
+        return None
+
+
+def main():
+    parser = argparse.ArgumentParser(
+        description='完整报销流程'
+    )
+    parser.add_argument(
+        '--amount', required=True, help='报销金额'
+    )
+    parser.add_argument(
+        '--supplier', default='', help='供应商名称关键词'
+    )
+    parser.add_argument(
+        '--category', default='', help='费用类别关键词'
+    )
+    parser.add_argument(
+        '--category-type', default='expense',
+        choices=['income', 'expense'],
+    )
+    parser.add_argument('--tax', default='', help='税额')
+    parser.add_argument('--dry-run', action='store_true')
+    args = parser.parse_args()
+
+    supplier_code = ''
+    category_code = ''
+
+    if args.supplier:
+        print(f'🔍 搜索供应商: {args.supplier}')
+        data = run_dws([
+            'finance', 'supplier', 'search',
+            '--query', args.supplier,
+            '--format', 'json',
+        ], dry_run=args.dry_run)
+        if not args.dry_run and data:
+            if isinstance(data, list):
+                items = data
+            elif isinstance(data, dict):
+                inner = data.get('result', data)
+                items = inner if isinstance(inner, list) else []
+            else:
+                items = []
+            if items:
+                supplier_code = (items[0].get('code')
+                                 or items[0].get('supplierCode', ''))
+                name = items[0].get('name', '')
+                print(f"  ✓ 找到: {name} ({supplier_code})")
+            else:
+                print(f"  ⚠ 未找到供应商: {args.supplier}")
+
+    if args.category:
+        print(f'🔍 搜索费用类别: {args.category}')
+        data = run_dws([
+            'finance', 'category', 'search',
+            '--type', args.category_type,
+            '--query', args.category,
+            '--format', 'json',
+        ], dry_run=args.dry_run)
+        if not args.dry_run and data:
+            if isinstance(data, list):
+                items = data
+            elif isinstance(data, dict):
+                inner = data.get('result', data)
+                items = inner if isinstance(inner, list) else []
+            else:
+                items = []
+            if items:
+                category_code = (items[0].get('code')
+                                 or items[0].get('categoryCode', ''))
+                name = items[0].get('name', '')
+                print(f"  ✓ 找到: {name} ({category_code})")
+            else:
+                print(f"  ⚠ 未找到类别: {args.category}")
+
+    print(f'\n💰 创建付款单 (金额: {args.amount})')
+    cmd_args = [
+        'finance', 'receipt', 'create',
+        '--amount', args.amount,
+        '--format', 'json',
+    ]
+    if supplier_code:
+        cmd_args.extend(['--supplier-code', supplier_code])
+    if category_code:
+        cmd_args.extend(['--category-code', category_code])
+    if args.tax:
+        cmd_args.extend(['--tax', args.tax])
+
+    result = run_dws(cmd_args, dry_run=args.dry_run)
+    if result:
+        print(f"  ✓ 付款单已创建")
+    else:
+        print(f"  ✗ 创建失败")
+        sys.exit(1)
+
+    print('\n✅ 报销流程完成!')
+
+
+if __name__ == '__main__':
+    main()