invoice_macos_dialog.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
发票提取器 - macOS 版本
纯 AppleScript 对话框实现（不依赖 tkinter）
"""

import sys
import os
import subprocess
import importlib.util


def get_resource_path(relative_path):
    """获取资源文件的绝对路径（兼容 PyInstaller 打包后的路径）"""
    if hasattr(sys, '_MEIPASS'):
        return os.path.join(sys._MEIPASS, relative_path)
    return os.path.join(os.path.dirname(__file__), relative_path)


def run_osascript(script, timeout=300):
    """运行 AppleScript 并返回结果"""
    try:
        result = subprocess.run(
            ['osascript'],
            input=script,
            capture_output=True,
            text=True,
            timeout=timeout
        )
        if result.returncode == 0:
            return result.stdout.strip()
        return None
    except subprocess.TimeoutExpired:
        return None
    except Exception:
        return None


def osascript_dialog(text, buttons, default_button=1):
    """显示 macOS 原生对话框，返回用户点击的按钮"""
    button_list = buttons.split(',')
    button_script = ','.join([f'"{b}"' for b in button_list])
    escaped_text = text.replace('"', '\\"').replace('\\', '\\\\')

    script = f'''display dialog "{escaped_text}" buttons {{{button_script}}} default button {default_button}
return button returned of result
'''
    return run_osascript(script)


def osascript_input_dialog(text, default_answer=""):
    """显示输入对话框"""
    escaped_text = text.replace('"', '\\"').replace('\\', '\\\\')
    escaped_default = default_answer.replace('"', '\\"').replace('\\', '\\\\')

    script = f'''display dialog "{escaped_text}" default answer "{escaped_default}" buttons {{"确定", "取消"}} default button 1
return text returned of result
'''
    return run_osascript(script)


def osascript_choose_folder(prompt="选择文件夹:"):
    """显示文件夹选择对话框"""
    escaped_prompt = prompt.replace('"', '\\"').replace('\\', '\\\\')

    script = f'''set dialogResult to choose folder with prompt "{escaped_prompt}"
return POSIX path of dialogResult
'''
    return run_osascript(script)


def osascript_choose_file(prompt="选择文件", default_name=""):
    """显示文件保存对话框"""
    escaped_prompt = prompt.replace('"', '\\"').replace('\\', '\\\\')
    escaped_name = default_name.replace('"', '\\"').replace('\\', '\\\\')

    script = f'''set dialogResult to choose file name with prompt "{escaped_prompt}" default name "{escaped_name}"
return POSIX path of dialogResult
'''
    return run_osascript(script)


def show_welcome():
    """显示欢迎对话框（while 循环避免递归调用栈溢出）"""
    while True:
        script = '''display alert "发票提取器 v1.0.0" ¬
            message "智能识别PDF发票，自动提取发票信息\\n支持普通发票和高速费发票，一键生成Excel清单\\n\\n开发者: 阿凯" ¬
            buttons {"访问主页", "开始使用", "退出"} default button 2
return button returned of result'''

        result = run_osascript(script)
        if result == "访问主页":
            # 打开开发者 GitHub 主页，然后再次显示欢迎对话框
            subprocess.run(['open', 'https://github.com/MaydayV'])
        elif result == "开始使用":
            return True
        else:
            return False


def run_extraction():
    """执行发票提取流程"""
    print("=" * 50)
    print("       发票提取器")
    print("=" * 50)

    # 步骤1：选择发票目录
    base_path = osascript_choose_folder("请选择发票所在目录:")

    if not base_path:
        print("未选择目录，程序退出")
        return

    # 清理路径
    base_path = os.path.expanduser(base_path)
    if os.name != 'nt':
        base_path = base_path.replace('\\ ', ' ')

    if not base_path:
        print("未选择目录，程序退出")
        return

    # 步骤2：输入购买方关键词
    buyer_keyword = osascript_input_dialog("请输入购买方公司名称关键词:")

    if buyer_keyword is None:
        buyer_keyword = ""

    if not buyer_keyword:
        print("未输入关键词，程序退出")
        return

    # 步骤3：选择输出文件（可选）
    output_path = osascript_choose_file("选择Excel保存位置", "发票清单.xlsx")

    if not output_path:
        output_path = os.path.join(base_path, "发票清单.xlsx")

    # 显示确认信息
    confirm_msg = f'''请确认：

发票目录: {base_path}
购买方关键词: {buyer_keyword}
输出文件: {output_path}

开始处理？'''
    confirm = osascript_dialog(confirm_msg, "开始处理,取消", 1)

    if confirm != "开始处理":
        print("已取消")
        return

    # 执行处理
    print()
    print("-" * 50)
    print(f"发票目录: {base_path}")
    print(f"购买方关键词: {buyer_keyword}")
    print(f"输出文件: {output_path}")
    print("-" * 50)
    print()

    # 导入并运行主程序
    extractor_path = get_resource_path("invoice_extractor.py")
    spec = importlib.util.spec_from_file_location("invoice_extractor", extractor_path)
    extractor = importlib.util.module_from_spec(spec)
    spec.loader.exec_module(extractor)

    def log_callback(msg):
        print(msg)

    extractor.process_invoices(base_path, buyer_keyword, output_path, log_callback)

    # 完成提示
    complete_msg = f'''处理完成！

输出文件已保存到:
{output_path}'''
    osascript_dialog(complete_msg, "确定")


def main():
    # 依赖已在打包时内置，仅在开发调试模式下才尝试安装
    # （打包版中 sys.executable 指向 bootloader 而非 Python 解释器）
    try:
        import fitz  # noqa: F401
        import openpyxl  # noqa: F401
    except ImportError:
        # 仅在非打包环境（开发模式）下尝试自动安装
        if not getattr(sys, 'frozen', False):
            try:
                subprocess.check_call(
                    [sys.executable, '-m', 'pip', 'install', '-q', 'PyMuPDF', 'openpyxl']
                )
            except Exception:
                print("警告：自动安装依赖失败，请手动运行: pip install PyMuPDF openpyxl")

    # 显示欢迎界面
    if show_welcome():
        run_extraction()


if __name__ == "__main__":
    try:
        main()
    except KeyboardInterrupt:
        print("\n程序已取消")
    except Exception as e:
        print(f"错误: {e}")
        import traceback
        traceback.print_exc()