增加测试数据清理模块

jikimo_test_assistant/wizards/__init__.py

@@ -0,0 +1,2 @@
+# -*- coding: utf-8 -*-
+from . import jikimo_data_clean_wizard

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
+            }
+        }
+
+            
+
+

jikimo_test_assistant/wizards/jikimo_data_clean_wizard.xml

@@ -0,0 +1,47 @@
+<?xml version="1.0" encoding="utf-8"?>
+<odoo>
+    <!-- Form View -->
+    <record id="view_jikimo_data_clean_form" model="ir.ui.view">
+        <field name="name">jikimo.data.clean.wizard.form</field>
+        <field name="model">jikimo.data.clean.wizard</field>
+        <field name="arch" type="xml">
+            <form string="业务数据清理">
+                <sheet>
+                    <group>
+                        <field name="date"/>
+                        <field name="model_ids" widget="many2many_tags"/>
+                    </group>
+                </sheet>
+                <footer>
+                    <button name="action_clean_data_confirm" 
+                            string="确认清理" 
+                            type="object" 
+                            class="btn-primary"/>
+                    <button special="cancel" 
+                            string="取消" 
+                            class="btn-secondary"/>
+                </footer>
+            </form>
+        </field>
+    </record>
+
+    <!-- Action -->
+    <record id="action_jikimo_data_clean" model="ir.actions.act_window">
+        <field name="name">业务数据清理</field>
+        <field name="res_model">jikimo.data.clean.wizard</field>
+        <field name="view_mode">form</field>
+        <field name="target">new</field>
+    </record>
+
+    <!-- Menu -->
+    <menuitem id="menu_test_root"
+              name="测试"
+              parent="base.menu_custom"
+              sequence="100"/>
+
+    <menuitem id="menu_jikimo_data_clean"
+              name="业务数据清理"
+              parent="menu_test_root"
+              action="action_jikimo_data_clean"
+              sequence="10"/>
+</odoo>