优化导出

app/batch_export_dialog.py

@@ -0,0 +1,82 @@
+"""
+批量导出选项对话框
+"""
+
+from PyQt5.QtWidgets import QDialog, QVBoxLayout, QHBoxLayout, QButtonGroup, QRadioButton
+from PyQt5.QtCore import Qt
+from qfluentwidgets import BodyLabel, PushButton, PrimaryPushButton, SubtitleLabel
+
+
+class BatchExportDialog(QDialog):
+    """批量导出选项对话框"""
+    
+    EXPORT_NORMAL = 0  # 普通文件夹导出
+    EXPORT_MROBOT = 1  # MRobot 格式导出
+    
+    def __init__(self, parent=None):
+        super().__init__(parent)
+        self.setWindowTitle("导出选项")
+        self.setGeometry(200, 200, 400, 250)
+        self.export_type = self.EXPORT_NORMAL
+        self.init_ui()
+    
+    def init_ui(self):
+        """初始化UI"""
+        layout = QVBoxLayout(self)
+        layout.setContentsMargins(20, 20, 20, 20)
+        layout.setSpacing(20)
+        
+        # 标题
+        title_label = SubtitleLabel("选择导出方式")
+        layout.addWidget(title_label)
+        
+        # 选项组
+        self.button_group = QButtonGroup()
+        
+        # 普通导出选项
+        normal_radio = QRadioButton("普通导出")
+        normal_radio.setChecked(True)
+        normal_radio.setToolTip("将每个交易的图片导出到单独的文件夹（文件夹名：日期_金额）")
+        self.button_group.addButton(normal_radio, self.EXPORT_NORMAL)
+        layout.addWidget(normal_radio)
+        
+        normal_desc = BodyLabel("每个交易的图片保存在独立文件夹中，便于查看和管理")
+        layout.addWidget(normal_desc)
+        
+        layout.addSpacing(15)
+        
+        # MRobot 格式导出选项
+        mrobot_radio = QRadioButton("MRobot 专用格式")
+        mrobot_radio.setToolTip("导出为 .mrobot 文件（专用格式，用于数据转交）")
+        self.button_group.addButton(mrobot_radio, self.EXPORT_MROBOT)
+        layout.addWidget(mrobot_radio)
+        
+        mrobot_desc = BodyLabel("导出为 .mrobot 文件（ZIP 格式），包含完整的交易数据和图片，用于转交给他人")
+        layout.addWidget(mrobot_desc)
+        
+        layout.addStretch()
+        
+        # 按钮
+        btn_layout = QHBoxLayout()
+        btn_layout.addStretch()
+        
+        cancel_btn = PushButton("取消")
+        cancel_btn.clicked.connect(self.reject)
+        btn_layout.addWidget(cancel_btn)
+        
+        ok_btn = PrimaryPushButton("确定")
+        ok_btn.clicked.connect(self.on_ok)
+        btn_layout.addWidget(ok_btn)
+        
+        layout.addLayout(btn_layout)
+    
+    def on_ok(self):
+        """确定按钮点击"""
+        checked_button = self.button_group.checkedButton()
+        if checked_button:
+            self.export_type = self.button_group.id(checked_button)
+        self.accept()
+    
+    def get_export_type(self):
+        """获取选择的导出方式"""
+        return self.export_type

app/finance_interface.py

@@ -19,9 +19,11 @@ from pathlib import Path
 from datetime import datetime, timedelta
 import json
 import os
+import shutil
 
 from .tools.finance_manager import FinanceManager, TransactionType, Transaction, Account
 from .category_management_dialog import CategoryManagementDialog
+from .batch_export_dialog import BatchExportDialog
 
 
 class CreateTransactionDialog(QDialog):
@@ -405,6 +407,10 @@ class RecordViewDialog(QDialog):
         btn_layout = QHBoxLayout()
         btn_layout.addStretch()
         
+        export_images_btn = PushButton("导出图片")
+        export_images_btn.clicked.connect(self.on_export_images)
+        btn_layout.addWidget(export_images_btn)
+        
         close_btn = PushButton("关闭")
         close_btn.clicked.connect(self.reject)
         btn_layout.addWidget(close_btn)
@@ -436,6 +442,95 @@ class RecordViewDialog(QDialog):
             pixmap = QPixmap(str(full_path))
             scaled_pixmap = pixmap.scaledToWidth(150)
             label.setPixmap(scaled_pixmap)
+    
+    def on_export_images(self):
+        """导出交易的所有图片"""
+        if not self.transaction or not self.account_id:
+            InfoBar.warning(
+                title="提示",
+                content="没有可导出的图片",
+                isClosable=True,
+                position=InfoBarPosition.TOP,
+                duration=2000,
+                parent=self
+            )
+            return
+        
+        # 检查是否有图片
+        has_images = False
+        image_paths = {}
+        
+        if self.transaction.invoice_path:
+            full_path = self.finance_manager.get_transaction_image_path(self.account_id, self.transaction.invoice_path)
+            if full_path and full_path.exists():
+                image_paths['发票'] = full_path
+                has_images = True
+        
+        if self.transaction.payment_path:
+            full_path = self.finance_manager.get_transaction_image_path(self.account_id, self.transaction.payment_path)
+            if full_path and full_path.exists():
+                image_paths['支付记录'] = full_path
+                has_images = True
+        
+        if self.transaction.purchase_path:
+            full_path = self.finance_manager.get_transaction_image_path(self.account_id, self.transaction.purchase_path)
+            if full_path and full_path.exists():
+                image_paths['购买记录'] = full_path
+                has_images = True
+        
+        if not has_images:
+            InfoBar.warning(
+                title="提示",
+                content="该交易没有关联的图片",
+                isClosable=True,
+                position=InfoBarPosition.TOP,
+                duration=2000,
+                parent=self
+            )
+            return
+        
+        # 打开文件夹选择对话框
+        export_dir = QFileDialog.getExistingDirectory(
+            self,
+            "选择导出文件夹",
+            str(Path.home() / "Downloads")
+        )
+        
+        if not export_dir:
+            return
+        
+        export_path = Path(export_dir)
+        
+        # 创建文件夹，命名为 "日期_金额"
+        folder_name = f"{self.transaction.date}_{self.transaction.amount:.2f}"
+        folder_name = folder_name.replace(":", "-")  # 替换不允许的字符
+        transaction_folder = export_path / folder_name
+        transaction_folder.mkdir(parents=True, exist_ok=True)
+        
+        # 复制图片
+        try:
+            for img_name, img_path in image_paths.items():
+                ext = img_path.suffix
+                dest_file = transaction_folder / f"{img_name}{ext}"
+                shutil.copy(str(img_path), str(dest_file))
+            
+            InfoBar.success(
+                title="成功",
+                content=f"图片已导出到 {transaction_folder.name}",
+                isClosable=True,
+                position=InfoBarPosition.TOP,
+                duration=2000,
+                parent=self
+            )
+        except Exception as e:
+            InfoBar.warning(
+                title="错误",
+                content=f"导出失败: {str(e)}",
+                isClosable=True,
+                position=InfoBarPosition.TOP,
+                duration=2000,
+                parent=self
+            )
 
 
 class FinanceInterface(QWidget):
@@ -668,8 +763,8 @@ class FinanceInterface(QWidget):
         
         # 查询结果表格 - 使用 TreeWidget 风格
         self.query_result_table = TreeWidget()
-        self.query_result_table.setHeaderLabels(["日期", "交易人", "分类", "金额 (元)", "备注"])
-        self.query_result_table.setSelectionMode(self.query_result_table.SingleSelection)
+        self.query_result_table.setHeaderLabels(["", "日期", "交易人", "分类", "金额 (元)", "备注"])
+        self.query_result_table.setSelectionMode(self.query_result_table.MultiSelection)
         self.query_result_table.setIndentation(0)
         
         # 设置 TreeWidget 样式
@@ -677,11 +772,12 @@ class FinanceInterface(QWidget):
         if header:
             header.setStretchLastSection(False)
             # 列宽自适应，可拖动调整
-            header.setSectionResizeMode(0, QHeaderView.ResizeToContents)
-            header.setSectionResizeMode(1, QHeaderView.ResizeToContents)
-            header.setSectionResizeMode(2, QHeaderView.ResizeToContents)
-            header.setSectionResizeMode(3, QHeaderView.ResizeToContents)
-            header.setSectionResizeMode(4, QHeaderView.Stretch)
+            header.setSectionResizeMode(0, QHeaderView.ResizeToContents)  # 复选框列
+            header.setSectionResizeMode(1, QHeaderView.ResizeToContents)  # 日期
+            header.setSectionResizeMode(2, QHeaderView.ResizeToContents)  # 交易人
+            header.setSectionResizeMode(3, QHeaderView.ResizeToContents)  # 分类
+            header.setSectionResizeMode(4, QHeaderView.ResizeToContents)  # 金额
+            header.setSectionResizeMode(5, QHeaderView.Stretch)          # 备注
             # 启用列宽可拖动
             header.setSectionsMovable(False)
             header.setStretchLastSection(False)
@@ -695,6 +791,14 @@ class FinanceInterface(QWidget):
         
         # 查询结果操作按钮
         query_btn_layout = QHBoxLayout()
+        
+        # 批量导出按钮
+        self.batch_export_btn = PushButton("批量导出")
+        self.batch_export_btn.setFixedWidth(90)
+        self.batch_export_btn.clicked.connect(self.on_batch_export)
+        self.batch_export_btn.setEnabled(False)
+        query_btn_layout.addWidget(self.batch_export_btn)
+        
         query_btn_layout.addStretch()
         
         self.query_view_btn = PushButton("查看详情")
@@ -1030,12 +1134,17 @@ class FinanceInterface(QWidget):
         for transaction in results:
             # 创建树形项
             item = QTreeWidgetItem()
-            item.setText(0, transaction.date)
-            item.setText(1, transaction.trader)
-            item.setText(2, transaction.category)
-            item.setText(3, f"¥ {transaction.amount:.2f}")
-            item.setText(4, transaction.notes or "")
-            # 存储交易ID
+            # 第一列添加复选框
+            # flags: ItemIsSelectable=1, ItemIsEnabled=2, ItemIsUserCheckable=32
+            item.setFlags(item.flags() | 32)  # 32 = Qt.ItemIsUserCheckable
+            item.setCheckState(0, 0)  # 0 = Qt.Unchecked
+            # 其他列是数据
+            item.setText(1, transaction.date)
+            item.setText(2, transaction.trader)
+            item.setText(3, transaction.category)
+            item.setText(4, f"¥ {transaction.amount:.2f}")
+            item.setText(5, transaction.notes or "")
+            # 存储交易ID（在第一列）
             item.setData(0, 32, transaction.id)
             
             self.query_result_table.addTopLevelItem(item)
@@ -1192,6 +1301,7 @@ class FinanceInterface(QWidget):
         selected_items = self.query_result_table.selectedItems()
         has_selection = len(selected_items) > 0
         self.query_view_btn.setEnabled(has_selection)
+        self.batch_export_btn.setEnabled(has_selection)
     
     def on_view_record_clicked(self):
         """查看记录按钮点击"""
@@ -1217,6 +1327,156 @@ class FinanceInterface(QWidget):
             if trans_id:
                 self.delete_record(trans_id)
     
+    def on_batch_export(self):
+        """批量导出选中的交易"""
+        account_id = self.get_current_account_id()
+        if not account_id:
+            InfoBar.warning(
+                title="提示",
+                content="请先选择账户",
+                isClosable=True,
+                position=InfoBarPosition.TOP,
+                duration=2000,
+                parent=self
+            )
+            return
+        
+        # 获取选中的项目
+        selected_items = self.query_result_table.selectedItems()
+        if not selected_items:
+            InfoBar.warning(
+                title="提示",
+                content="请先选择要导出的交易",
+                isClosable=True,
+                position=InfoBarPosition.TOP,
+                duration=2000,
+                parent=self
+            )
+            return
+        
+        # 提取交易 ID（从任何可用的列）
+        transaction_ids = set()
+        for item in selected_items:
+            trans_id = None
+            for col in range(6):  # 检查所有列
+                trans_id = item.data(col, 32)
+                if trans_id:
+                    break
+            if trans_id:
+                transaction_ids.add(trans_id)
+        
+        if not transaction_ids:
+            InfoBar.warning(
+                title="提示",
+                content="无法获取交易信息",
+                isClosable=True,
+                position=InfoBarPosition.TOP,
+                duration=2000,
+                parent=self
+            )
+            return
+        
+        # 打开导出方式选择对话框
+        export_dialog = BatchExportDialog(self)
+        if export_dialog.exec() != QFileDialog.Accepted:
+            return
+        
+        export_type = export_dialog.get_export_type()
+        
+        # 根据导出方式选择
+        if export_type == BatchExportDialog.EXPORT_MROBOT:
+            # MRobot 格式导出
+            file_dialog = QFileDialog()
+            file_path, _ = file_dialog.getSaveFileName(
+                self,
+                "保存为 MRobot 文件",
+                str(Path.home() / "Downloads" / "export.mrobot"),
+                "MRobot Files (*.mrobot)"
+            )
+            
+            if not file_path:
+                return
+            
+            if self.finance_manager.export_to_mrobot_format(account_id, list(transaction_ids), file_path):
+                InfoBar.success(
+                    title="导出成功",
+                    content=f"已导出 {len(transaction_ids)} 个交易到 {Path(file_path).name}",
+                    isClosable=True,
+                    position=InfoBarPosition.TOP,
+                    duration=2000,
+                    parent=self
+                )
+            else:
+                InfoBar.warning(
+                    title="导出失败",
+                    content="导出 MRobot 格式文件失败",
+                    isClosable=True,
+                    position=InfoBarPosition.TOP,
+                    duration=2000,
+                    parent=self
+                )
+        else:
+            # 普通文件夹导出
+            export_dir = QFileDialog.getExistingDirectory(
+                self,
+                "选择导出文件夹",
+                str(Path.home() / "Downloads")
+            )
+            
+            if not export_dir:
+                return
+            
+            export_path = Path(export_dir)
+            success_count = 0
+            
+            # 遍历选中的交易，导出它们的图片
+            for trans_id in transaction_ids:
+                transaction = self.finance_manager._load_transaction_data(account_id, trans_id)
+                if not transaction:
+                    continue
+                
+                # 创建文件夹，命名为 "日期_金额"
+                folder_name = f"{transaction.date}_{transaction.amount:.2f}"
+                folder_name = folder_name.replace(":", "-")  # 替换不允许的字符
+                transaction_folder = export_path / folder_name
+                transaction_folder.mkdir(parents=True, exist_ok=True)
+                
+                # 收集图片文件
+                images_found = False
+                try:
+                    if transaction.invoice_path:
+                        img_path = self.finance_manager.get_transaction_image_path(account_id, transaction.invoice_path)
+                        if img_path and img_path.exists():
+                            shutil.copy(str(img_path), str(transaction_folder / f"发票{img_path.suffix}"))
+                            images_found = True
+                    
+                    if transaction.payment_path:
+                        img_path = self.finance_manager.get_transaction_image_path(account_id, transaction.payment_path)
+                        if img_path and img_path.exists():
+                            shutil.copy(str(img_path), str(transaction_folder / f"支付记录{img_path.suffix}"))
+                            images_found = True
+                    
+                    if transaction.purchase_path:
+                        img_path = self.finance_manager.get_transaction_image_path(account_id, transaction.purchase_path)
+                        if img_path and img_path.exists():
+                            shutil.copy(str(img_path), str(transaction_folder / f"购买记录{img_path.suffix}"))
+                            images_found = True
+                    
+                    if images_found:
+                        success_count += 1
+                except Exception as e:
+                    print(f"导出交易 {trans_id} 失败: {e}")
+                    continue
+            
+            InfoBar.success(
+                title="导出完成",
+                content=f"成功导出 {success_count} 个交易",
+                isClosable=True,
+                position=InfoBarPosition.TOP,
+                duration=2000,
+                parent=self
+            )
+    
     def on_query_view_clicked(self):
         """查询结果查看详情按钮点击"""
         current_item = self.query_result_table.currentItem()

app/tools/finance_manager.py

@@ -654,3 +654,85 @@ class FinanceManager:
         if not account:
             return []
         return account.categories
+    
+    def export_to_mrobot_format(self, account_id: str, transaction_ids: List[str], output_path: str) -> bool:
+        """导出交易为 .mrobot 专用格式（ZIP 文件）
+        
+        格式说明：
+        - 文件扩展名：.mrobot（实际上是 ZIP 文件）
+        - 结构：
+          - metadata.json（交易元数据）
+          - images/（所有相关图片）
+        """
+        try:
+            account = self.accounts.get(account_id)
+            if not account:
+                return False
+            
+            output_file = Path(output_path)
+            if not output_file.parent.exists():
+                output_file.parent.mkdir(parents=True, exist_ok=True)
+            
+            # 创建 ZIP 文件
+            with zipfile.ZipFile(output_file, 'w', zipfile.ZIP_DEFLATED) as zf:
+                # 收集交易数据和图片
+                transactions_data = []
+                image_index = 0
+                
+                for trans_id in transaction_ids:
+                    transaction = self._load_transaction_data(account_id, trans_id)
+                    if not transaction:
+                        continue
+                    
+                    # 转换交易为字典
+                    trans_dict = transaction.to_dict()
+                    
+                    # 处理图片，存储相对路径
+                    image_map = {}
+                    
+                    if transaction.invoice_path:
+                        img_path = self.get_transaction_image_path(account_id, transaction.invoice_path)
+                        if img_path and img_path.exists():
+                            archive_path = f"images/invoice_{image_index}{img_path.suffix}"
+                            zf.write(str(img_path), archive_path)
+                            image_map['invoice'] = archive_path
+                            image_index += 1
+                    
+                    if transaction.payment_path:
+                        img_path = self.get_transaction_image_path(account_id, transaction.payment_path)
+                        if img_path and img_path.exists():
+                            archive_path = f"images/payment_{image_index}{img_path.suffix}"
+                            zf.write(str(img_path), archive_path)
+                            image_map['payment'] = archive_path
+                            image_index += 1
+                    
+                    if transaction.purchase_path:
+                        img_path = self.get_transaction_image_path(account_id, transaction.purchase_path)
+                        if img_path and img_path.exists():
+                            archive_path = f"images/purchase_{image_index}{img_path.suffix}"
+                            zf.write(str(img_path), archive_path)
+                            image_map['purchase'] = archive_path
+                            image_index += 1
+                    
+                    trans_dict['image_map'] = image_map
+                    transactions_data.append(trans_dict)
+                
+                # 创建元数据 JSON
+                metadata = {
+                    'version': '1.0',
+                    'account_name': account.name,
+                    'account_description': account.description,
+                    'export_date': datetime.now().isoformat(),
+                    'transactions': transactions_data,
+                    'transaction_count': len(transactions_data)
+                }
+                
+                # 将元数据写入 ZIP
+                metadata_json = json.dumps(metadata, ensure_ascii=False, indent=2)
+                zf.writestr('metadata.json', metadata_json)
+            
+            return True
+        
+        except Exception as e:
+            print(f"导出失败: {e}")
+            return False

assets/Finance_Data/accounts/b87b752a-7fb6-40b6-a0cf-287df1d64e56/28ae4af8-67d9-431a-b2fe-312f587020b3/data.json

@@ -4,10 +4,10 @@
   "amount": 5000.0,
   "trader": "工作",
   "notes": "1月工资",
-  "invoice_path": null,
+  "invoice_path": "accounts/b87b752a-7fb6-40b6-a0cf-287df1d64e56/28ae4af8-67d9-431a-b2fe-312f587020b3/invoice/截屏2025-11-25 02.51.14.png",
   "payment_path": null,
   "purchase_path": null,
   "category": "NUC",
   "created_at": "2025-11-25T20:44:13.766681",
-  "updated_at": "2025-11-25T20:54:10.845738"
+  "updated_at": "2025-11-25T21:11:13.405339"
 }