增加测试数据清理模块

2025-02-13 11:31:35 +08:00
parent bb1012a15d
commit 941b8c0be7
11 changed files with 330 additions and 4 deletions
--- a/jikimo_test_assistant/wizards/jikimo_data_clean_wizard.py
+++ b/jikimo_test_assistant/wizards/jikimo_data_clean_wizard.py
@@ -0,0 +1,99 @@
+from odoo import models, fields, api
+from datetime import datetime
+
+import logging
+
+_logger = logging.getLogger(__name__)
+
+class JikimoDataCleanWizard(models.TransientModel):
+    _name = 'jikimo.data.clean.wizard'
+    _description = '业务数据清理'
+
+    date = fields.Date(string='截止日期', required=True, default=fields.Date.context_today)
+    model_ids = fields.Many2many('ir.model', string='业务模型', domain=[
+        ('model', 'in', [
+            'sale.order',           # 销售订单
+            'purchase.order',       # 采购订单 
+            'mrp.production',       # 生产订单
+            'stock.picking',        # 库存调拨
+            'account.move',         # 会计凭证
+        ])
+    ])
+
+    def action_clean_data(self):
+        self.ensure_one()
+        model_list = self.model_ids.mapped('model')
+        
+        # 销售订单清理（排除已交付，已锁定，已取消）
+        if 'sale.order' in model_list:
+            self.model_cancel('sale.order', except_states=['delivered', 'done', 'cancel'])
+
+        # 采购订单清理（排除采购订单，已锁定，已取消）
+        if 'purchase.order' in model_list:
+            self.model_cancel('purchase.order', except_states=['purchase', 'done', 'cancel'])
+
+        # 生产订单清理（排除返工，报废，完成，已取消）
+        if 'mrp.production' in model_list:
+            self.model_cancel('mrp.production', except_states=['rework', 'scrap', 'done', 'cancel'])
+
+        # 工单清理 （排除返工，完成，已取消）
+        if 'mrp.workorder' in model_list:
+            self.model_cancel('mrp.production', except_states=['rework', 'done', 'cancel'])
+        
+        # 排程单清理 （排除已完成，已取消）
+        if 'mrp.workorder' in model_list:
+            self.model_cancel('mrp.production', except_states=['finished', 'cancel'])
+
+        # 工单库存移动 （排除完成，已取消）
+        if 'stock.move' in model_list:
+            self.model_cancel('stock.move')
+
+        # 库存调拨清理 （排除完成，已取消）
+        if 'stock.picking' in model_list:
+            self.model_cancel('stock.picking')
+
+        # 会计凭证清理 （排除已过账，已取消）
+        if 'account.move' in model_list:
+            self.model_cancel('account.move', except_states=['posted', 'cancel'])
+
+        return True
+    
+    def model_cancel(self, model_name, state_field='state', to_state='cancel',except_states=('done', 'cancel')):
+        table = self.env[model_name]._table
+        if isinstance(except_states, list):
+            except_states = tuple(except_states)
+        sql = """
+                UPDATE 
+                    %s SET %s = '%s'
+                WHERE 
+                    create_date < '%s'
+                AND state NOT IN %s
+""" % (table, state_field, to_state, self.date.strftime('%Y-%m-%d'), except_states)
+        self.env.cr.execute(sql)
+        self.env.cr.commit()
+
+    @api.model
+    def get_confirm_message(self):
+        date_str = self.date.strftime('%Y-%m-%d') if self.date else ''
+        model_names = ', '.join([model.name for model in self.model_ids])
+        return {
+            'date': date_str,
+            'model_names': model_names
+        }
+    
+    def action_clean_data_confirm(self):
+        model_names = self.model_ids.mapped('display_name')
+        return {
+            'type': 'ir.actions.client',
+            'tag': 'action_clean_data_confirm',
+            'params': {
+                'model_names': model_names,
+                'date': self.date,
+                'active_id': self.id,
+                'context': self.env.context
+            }
+        }
+
+            
+
+