139 lines
4.2 KiB
Python
139 lines
4.2 KiB
Python
#!/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()
|